瀏覽代碼

change multi scale range to [0.5, 1.0] -> [320, 640]

yjh0410 2 年之前
父節點
當前提交
26b436e8c5
共有 1 個文件被更改,包括 1 次插入3 次删除
  1. 1 3
      dataset/data_augment/yolov5_augment.py

+ 1 - 3
dataset/data_augment/yolov5_augment.py

@@ -306,9 +306,7 @@ class YOLOv5Augmentation(object):
         else:
             img = image
 
-        # rescale bboxes
-        if target is not None:
-            img_h, img_w = img.shape[:2]
+        img_h, img_w = img.shape[:2]
 
         # hsv augment
         augment_hsv(img, hgain=self.trans_config['hsv_h'],