소스 검색

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'],