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

+controle mag reg + Tests fonction mag reg

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-02-25 14:05:17 -05:00
parent fc0fb25148
commit 534e244307
6 changed files with 45 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

7
higher/smart_aug/transformations.py
View file

@ -31,6 +31,13 @@ PARAMETER_MAX = 1
# What is the min 'level' a transform could be predicted
PARAMETER_MIN = 0.1
#Dict containing the value for wich TF are closer to identity
#TF_identity={
# PARAMETER_MAX:{'Solarize', 'Posterize'},
# PARAMETER_MAX/2:{'Contrast','Color','Brightness','Sharpness'},
# PARAMETER_MIN:{'Rotate','TranslateX','TranslateY','ShearX','ShearY'},
#}
class TF_loader(object):
""" Transformations builder.