Update get_dr_txt.py

get_dr_txt.py

@@ -39,9 +39,8 @@ class mAP_Yolo(YOLO):
             images = torch.from_numpy(images)
             if self.cuda:
                 images = images.cuda()
-                
+            outputs = self.net(images)
             
-        outputs = self.net(images)
         output_list = []
         for i in range(3):
             output_list.append(self.yolo_decodes[i](outputs[i]))
@@ -94,4 +93,4 @@ for image_id in image_ids:
     print(image_id," done!")
     
 
-print("Conversion completed!")
\ No newline at end of file
+print("Conversion completed!")