fix($data/transforms): change augmix augmentation pool

change augmentation_all from augmentation_reid in augmix
because we found AutoAugmentation using ImageNet Policy will not harm performance
pull/44/head
liaoxingyu 2020-04-27 15:06:27 +08:00
parent 4d3e5fd378
commit 9e3f2c1e7a
1 changed files with 1 additions and 1 deletions

View File

@ -179,7 +179,7 @@ class AugMix(object):
self.mixture_width = mixture_width
self.mixture_depth = mixture_depth
self.severity = severity
self.aug_list = augmentations_reid
self.aug_list = augmentations_all
def __call__(self, image):
"""Perform AugMix augmentations and compute mixture.