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

+ Controle mag Augmented Dataset

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-12-09 13:49:24 -05:00
parent 6c0597e7ea
commit d1ee0c632e
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
higher/datasets.py
View file

@ -197,8 +197,8 @@ class AugmentedDatasetV2(VisionDataset):
'Color', 'Color',
'Brightness', 'Brightness',
'Sharpness', 'Sharpness',
#'Posterize', 'Posterize',
#'Solarize', 'Solarize',
'Invert', 'Invert',
'AutoContrast', 'AutoContrast',
@ -206,8 +206,9 @@ class AugmentedDatasetV2(VisionDataset):
] ]
self._op_list =[] self._op_list =[]
self.prob=0.5 self.prob=0.5
self.mag_range=(1, 10)
for tf in self._TF: for tf in self._TF:
for mag in range(1, 10): for mag in range(self.mag_range[0], self.mag_range[1]):
self._op_list+=[(tf, self.prob, mag)] self._op_list+=[(tf, self.prob, mag)]
self._nb_op = len(self._op_list) self._nb_op = len(self._op_list)
@ -267,7 +268,7 @@ class AugmentedDatasetV2(VisionDataset):
return self.dataset_info['unsup']#self.dataset_info['length'] return self.dataset_info['unsup']#self.dataset_info['length']
def __str__(self): def __str__(self):
return "CIFAR10(Sup:{}-Unsup:{}-{}TF)".format(self.dataset_info['sup'], self.dataset_info['unsup'], len(self._TF)) return "CIFAR10(Sup:{}-Unsup:{}-{}TF(Mag{}-{}))".format(self.dataset_info['sup'], self.dataset_info['unsup'], len(self._TF), self.mag_range[0], self.mag_range[1])
### Classic Dataset ### ### Classic Dataset ###