From ba3c4f763c809036a1e8b61be0971019efce1843 Mon Sep 17 00:00:00 2001
From: HTensor <27774381+HTensor@users.noreply.github.com>
Date: Wed, 3 May 2023 17:22:15 +0800
Subject: [PATCH] Update augmentor.py

---
 IGEV-Stereo/core/utils/augmentor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)