Răsfoiți Sursa

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

yjh0410 2 ani în urmă
părinte
comite
26b436e8c5
1 a modificat fișierele cu 1 adăugiri și 3 ștergeri
  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'],