yjh0410 1 년 전
부모
커밋
a9fc8d6863
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      yolo/dataset/voc.py

+ 1 - 0
yolo/dataset/voc.py

@@ -250,6 +250,7 @@ if __name__ == "__main__":
             self.train_img_size = 640
             self.test_img_size  = 640
             self.aug_type = 'ssd'
+            self.affine_params = None
 
     if args.aug_type == "yolo":
         cfg = YoloBaseConfig()