diff --git a/albumentations/augmentations/transforms.py b/albumentations/augmentations/transforms.py index 513f42c14..fd9b039c7 100644 --- a/albumentations/augmentations/transforms.py +++ b/albumentations/augmentations/transforms.py @@ -1389,7 +1389,7 @@ def apply(self, img, gamma=1, **params): return F.gamma_transform(img, gamma=gamma) def get_params(self): - return {"gamma": random.randint(self.gamma_limit[0], self.gamma_limit[1]) / 100.0} + return {"gamma": random.uniform(self.gamma_limit[0], self.gamma_limit[1]) / 100.0} def get_transform_init_args_names(self): return ("gamma_limit", "eps")