HTensor/IGEV
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update augmentor.py

Browse Source
This commit is contained in:
HTensor 2023-05-03 17:22:15 +08:00
parent 9df896bb70
commit ba3c4f763c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
IGEV-Stereo/core/utils/augmentor.py
View File

@ -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)