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