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

leger chgt mag regularisation

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-11-19 21:46:14 -05:00
parent 64282bda3a
commit cc737b7997
5 changed files with 21 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

5
higher/dataug.py
View file

@ -552,6 +552,8 @@ class Data_augV5(nn.Module): #Optimisation jointe (mag, proba)
else torch.tensor(0.5).expand(self._nb_tf)), #[0, PARAMETER_MAX]
})
#for t in TF.TF_no_mag: self._params['mag'][self._TF.index(t)].data-=self._params['mag'][self._TF.index(t)].data #Mag inutile pour les TF ignore_mag
#Distribution
self._samples = []
self._mix_dist = False
@ -561,8 +563,7 @@ class Data_augV5(nn.Module): #Optimisation jointe (mag, proba)
#Mag regularisation
if not self._fixed_mag:
ignore={'Identity', 'FlipUD', 'FlipLR', 'Solarize', 'Posterize'}
self._reg_mask=[self._TF.index(t) for t in self._TF if t not in ignore]
self._reg_mask=[self._TF.index(t) for t in self._TF if t not in TF.TF_ignore_mag]
self._reg_tgt = torch.full(size=(len(self._reg_mask),), fill_value=TF.PARAMETER_MAX) #Encourage amplitude max
def forward(self, x):