Add checking of type image_fill_value

albumentations-team · May 2, 2020 · efc5367 · efc5367
1 parent 834d50d
commit efc5367
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/albumentations/augmentations/transforms.py b/albumentations/augmentations/transforms.py
@@ -3161,7 +3161,7 @@ def apply(self, img, dropout_mask=None, image_fill_value=0, **params):
         if dropout_mask is None:
             return img
 
-        if image_fill_value == "inpaint":
+        if isinstance(image_fill_value, str) and image_fill_value == "inpaint":
             dropout_mask = dropout_mask.astype(np.uint8)
             _, _, w, h = cv2.boundingRect(dropout_mask)
             radius = min(3, max(w, h) // 2)