Fix Affine deprecation warning (aleju#685)

This patch fixes a deprecation warning in `Affine` that would be caused when providing boolean images and `order != 0`.
erjel · Jun 1, 2020 · 0101108 · 0101108
1 parent 6c73eaa
commit 0101108
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/changelogs/master/fixed/20200601_fix_affine_skimage_order_0.md b/changelogs/master/fixed/20200601_fix_affine_skimage_order_0.md
@@ -0,0 +1,3 @@
+- Fixed a deprecation warning in `Affine` that would
+  be caused when providing boolean images and
+  `order != 0`. #685
diff --git a/imgaug/augmenters/geometric.py b/imgaug/augmenters/geometric.py
@@ -231,6 +231,12 @@ def _warp_affine_arr_skimage(arr, matrix, cval, mode, order, output_shape):
 
     input_dtype = arr.dtype
 
+    # tf.warp() produces a deprecation warning for bool images with
+    # order!=0. We either need to convert them to float or use NN
+    # interpolation.
+    if input_dtype == iadt._BOOL_DTYPE and order != 0:
+        arr = arr.astype(np.float32)
+
     image_warped = tf.warp(
         arr,
         np.linalg.inv(matrix),