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

Supression res mauvais test brutus

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-12-04 10:36:34 -05:00
parent 6b7ed4836e
commit c4e2e30151
19 changed files with 11 additions and 182711 deletions
Download patch file Download diff file Expand all files Collapse all files

3
higher/dataug.py
View file

@ -5,7 +5,7 @@ from torch.distributions import *
#import kornia
#import random
#import numpy as np
import numpy as np
import copy
import transformations as TF
@ -692,7 +692,6 @@ class Data_augV5(nn.Module): #Optimisation jointe (mag, proba)
else:
return "Data_augV5(Mix%.1f%s-%dTFx%d-%s)" % (self._mix_factor,dist_param, self._nb_tf, self._N_seqTF, mag_param)
import numpy as np
class Data_augV6(nn.Module): #Optimisation sequentielle
def __init__(self, TF_dict=TF.TF_dict, N_TF=1, mix_dist=0.0, fixed_prob=False, prob_set_size=None, fixed_mag=True, shared_mag=True):
super(Data_augV6, self).__init__()