diff --git a/IGEV-Stereo/core/utils/augmentor.py b/IGEV-Stereo/core/utils/augmentor.py index fa81ba9..3641677 100644 --- a/IGEV-Stereo/core/utils/augmentor.py +++ b/IGEV-Stereo/core/utils/augmentor.py @@ -200,7 +200,7 @@ class SparseFlowAugmentor: # photometric augmentation params self.photo_aug = Compose([ColorJitter(brightness=0.3, contrast=0.3, saturation=saturation_range, hue=0.3/3.14), AdjustGamma(*gamma)]) self.asymmetric_color_aug_prob = 0.2 - self.eraser_aug_prob = 0.5 + self.eraser_aug_prob = 1.0 def color_transform(self, img1, img2): image_stack = np.concatenate([img1, img2], axis=0)