diff --git a/higher/transformations.py b/higher/transformations.py index 5c82b48..d9b1306 100755 --- a/higher/transformations.py +++ b/higher/transformations.py @@ -102,8 +102,8 @@ TF_dict={ #Dataugv5 #'Equalize': (lambda mag: None), } -TF_no_mag={'Identity', 'FlipUD', 'FlipLR'} -TF_ignore_mag= TF_no_mag | {'Solarize', 'Posterize'} +TF_no_mag={'Identity', 'FlipUD', 'FlipLR', 'Random', 'RandBlend'} +TF_ignore_mag= TF_no_mag | {'Solarize', 'Posterize', '=Solarize', '=Posterize'} def int_image(float_image): #ATTENTION : legere perte d'info (granularite : 1/256 = 0.0039) return (float_image*255.).type(torch.uint8)