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

Tests influence nb TF

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-11-08 16:50:02 -05:00
parent 0066da2e4d
commit 0d1a684aed
5 changed files with 46 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
higher/dataug.py
View file

@ -316,6 +316,8 @@ class Data_augV3(nn.Module): #Echantillonage uniforme/Mixte
class Data_augV4(nn.Module): #Transformations avec mask
def __init__(self, TF_dict=TF.TF_dict, N_TF=1, mix_dist=0.0):
super(Data_augV4, self).__init__()
assert len(TF_dict)>0
self._data_augmentation = True
#self._TF_matrix={}