AntoineH/smart_augmentation
1
0
Fork 0
mirror of https://github.com/AntoineHX/smart_augmentation.git synced 2025-05-04 12:10:45 +02:00
Code Issues Projects Releases Packages Wiki Activity

Test WRN Brutus

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-12-04 16:34:02 -05:00
parent b26fbcd2a2
commit fa5bc72616
6 changed files with 72 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

26
higher/datasets.py
View file

@ -65,21 +65,26 @@ class AugmentedDataset(VisionDataset):
self._TF = [
'Invert',
'Cutout',
'Sharpness',
'AutoContrast',
'Posterize',
'ShearX',
## Geometric TF ##
'Rotate',
'TranslateX',
'TranslateY',
'ShearX',
'ShearY',
'Rotate',
'Equalize',
'Cutout',
## Color TF ##
'Contrast',
'Color',
'Solarize',
'Brightness'
'Brightness',
'Sharpness',
#'Posterize',
#'Solarize',
'Invert',
'AutoContrast',
'Equalize',
]
self._op_list =[]
self.prob=0.5
@ -119,6 +124,7 @@ class AugmentedDataset(VisionDataset):
for idx, image in enumerate(self.sup_data):
if (idx/self.dataset_info['sup'])%0.2==0: print("Augmenting data... ", idx,"/", self.dataset_info['sup'])
#if idx==10000:break
for _ in range(aug_copy):
chosen_policy = policies[np.random.choice(len(policies))]