Merge pull request #7 from cxy1997/master

Fixed typos in kitti augmentation
This commit is contained in:
Zach Teed 2020-05-23 10:02:19 -04:00 committed by GitHub
commit dd91321527
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -221,7 +221,7 @@ class KITTI(FlowDataset):
self.do_pad = do_pad self.do_pad = do_pad
if self.do_augument: 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: if self.is_test:
images1 = sorted(glob(os.path.join(root, 'testing', 'image_2/*_10.png'))) images1 = sorted(glob(os.path.join(root, 'testing', 'image_2/*_10.png')))