Merge pull request #372 from albu/equalize_true_divide

fix _equalize_cv true divide
albumentations-team · Sep 23, 2019 · f38e153 · f38e153
2 parents db20c0b + 166262b
commit f38e153
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/albumentations/augmentations/functional.py b/albumentations/augmentations/functional.py
@@ -472,7 +472,7 @@ def _equalize_cv(img, mask=None):
     if histogram[i] == total:
         return np.full_like(img, i)
 
-    scale = 255 / (total - histogram[i])
+    scale = 255.0 / (total - histogram[i])
     _sum = 0
 
     lut = np.zeros(256, dtype=np.uint8)