@@ -700,22 +700,22 @@ def get_params(brightness, contrast, saturation, hue):
700
700
"""
701
701
transforms = []
702
702
if brightness > 0 :
703
- brightness_factor = np . random .uniform (max (0 , 1 - brightness ), 1 + brightness )
703
+ brightness_factor = random .uniform (max (0 , 1 - brightness ), 1 + brightness )
704
704
transforms .append (Lambda (lambda img : F .adjust_brightness (img , brightness_factor )))
705
705
706
706
if contrast > 0 :
707
- contrast_factor = np . random .uniform (max (0 , 1 - contrast ), 1 + contrast )
707
+ contrast_factor = random .uniform (max (0 , 1 - contrast ), 1 + contrast )
708
708
transforms .append (Lambda (lambda img : F .adjust_contrast (img , contrast_factor )))
709
709
710
710
if saturation > 0 :
711
- saturation_factor = np . random .uniform (max (0 , 1 - saturation ), 1 + saturation )
711
+ saturation_factor = random .uniform (max (0 , 1 - saturation ), 1 + saturation )
712
712
transforms .append (Lambda (lambda img : F .adjust_saturation (img , saturation_factor )))
713
713
714
714
if hue > 0 :
715
- hue_factor = np . random .uniform (- hue , hue )
715
+ hue_factor = random .uniform (- hue , hue )
716
716
transforms .append (Lambda (lambda img : F .adjust_hue (img , hue_factor )))
717
717
718
- np . random .shuffle (transforms )
718
+ random .shuffle (transforms )
719
719
transform = Compose (transforms )
720
720
721
721
return transform
@@ -782,7 +782,7 @@ def get_params(degrees):
782
782
Returns:
783
783
sequence: params to be passed to ``rotate`` for random rotation.
784
784
"""
785
- angle = np . random .uniform (degrees [0 ], degrees [1 ])
785
+ angle = random .uniform (degrees [0 ], degrees [1 ])
786
786
787
787
return angle
788
788
0 commit comments