debug YOLOv5BaseTransform

dataset/data_augment/yolov5_augment.py

@@ -392,7 +392,7 @@ class YOLOv5BaseTransform(object):
         # r = min(r, 1.0) # only scale down, do not scale up (for better val mAP)
         if r != 1: 
             new_shape = (int(round(img_w0 * r)), int(round(img_h0 * r)))
-            img = cv2.resize(image, new_shape, interpolation=cv2.INTER_LINEAR)
+            img = cv2.resize(image, new_shape)
         else:
             img = image
 

test.py

@@ -33,7 +33,7 @@ def parse_args():
                         help='use cuda.')
     parser.add_argument('--save_folder', default='det_results/', type=str,
                         help='Dir to save results')
-    parser.add_argument('-vs', '--visual_threshold', default=0.3, type=float,
+    parser.add_argument('-vs', '--visual_threshold', default=0.4, type=float,
                         help='Final confidence threshold')
     parser.add_argument('-ws', '--window_scale', default=1.0, type=float,
                         help='resize window of cv2 for visualization.')