Merge pull request #7 from cxy1997/master

Fixed typos in kitti augmentation
hm-ysjiang · May 23, 2020 · dd91321 · dd91321
2 parents 559176d + 13aa363
commit dd91321
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/datasets.py b/core/datasets.py
@@ -221,7 +221,7 @@ def __init__(self, args, image_size=None, do_augument=True, is_test=False, is_va
         self.do_pad = do_pad
 
         if self.do_augument:
-            self.augumentor = FlowAugumentorKITTI(self.image_size, args.eraser_aug, min_scale=-0.2, max_scale=0.5)
+            self.augumentor = FlowAugmentorKITTI(self.image_size, min_scale=-0.2, max_scale=0.5)
 
         if self.is_test:
             images1 = sorted(glob(os.path.join(root, 'testing', 'image_2/*_10.png')))