Fix numpy >= 1.20 np.bool deprecation

imgaug/augmentables/polys.py

@@ -950,7 +950,7 @@ def extract_from_image(self, image):
         rr_face, cc_face = skimage.draw.polygon(
             yy_mask, xx_mask, shape=(height_mask, width_mask))
 
-        mask = np.zeros((height_mask, width_mask), dtype=np.bool)
+        mask = np.zeros((height_mask, width_mask), dtype=bool)
         mask[rr_face, cc_face] = True
 
         if image.ndim == 3:

imgaug/augmenters/meta.py

@@ -3384,7 +3384,7 @@ def _get_augmenter_active(self, nb_rows, random_state):
         # pylint: disable=invalid-name
         nn = self._get_n(nb_rows, random_state)
         nn = [min(n, len(self)) for n in nn]
-        augmenter_active = np.zeros((nb_rows, len(self)), dtype=np.bool)
+        augmenter_active = np.zeros((nb_rows, len(self)), dtype=bool)
         for row_idx, n_true in enumerate(nn):
             if n_true > 0:
                 augmenter_active[row_idx, 0:n_true] = 1

test/augmentables/test_bbs.py

@@ -749,7 +749,7 @@ def test_draw_label_on_image(self):
     def _get_standard_draw_box_on_image_vars(cls):
         image = np.zeros((10, 10, 3), dtype=np.uint8)
         bb = ia.BoundingBox(y1=1, x1=1, y2=3, x2=3)
-        bb_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        bb_mask = np.zeros(image.shape[0:2], dtype=bool)
         bb_mask[1:3+1, 1] = True
         bb_mask[1:3+1, 3] = True
         bb_mask[1, 1:3+1] = True
@@ -823,7 +823,7 @@ def test_draw_box_on_image_no_copy(self):
     def test_draw_box_on_image_bb_outside_of_image(self):
         image = np.zeros((10, 10, 3), dtype=np.uint8)
         bb = ia.BoundingBox(y1=-1, x1=-1, y2=2, x2=2)
-        bb_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        bb_mask = np.zeros(image.shape[0:2], dtype=bool)
         bb_mask[2, 0:3] = True
         bb_mask[0:3, 2] = True
 
@@ -837,7 +837,7 @@ def test_draw_box_on_image_bb_outside_of_image(self):
     def test_draw_box_on_image_bb_outside_of_image_and_very_small(self):
         image, bb, bb_mask = self._get_standard_draw_box_on_image_vars()
         bb = ia.BoundingBox(y1=-1, x1=-1, y2=1, x2=1)
-        bb_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        bb_mask = np.zeros(image.shape[0:2], dtype=bool)
         bb_mask[0:1+1, 1] = True
         bb_mask[1, 0:1+1] = True
 
@@ -850,7 +850,7 @@ def test_draw_box_on_image_bb_outside_of_image_and_very_small(self):
 
     def test_draw_box_on_image_size_2(self):
         image, bb, _ = self._get_standard_draw_box_on_image_vars()
-        bb_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        bb_mask = np.zeros(image.shape[0:2], dtype=bool)
         bb_mask[0:5, 0:5] = True
         bb_mask[2, 2] = False
 

test/augmentables/test_kps.py

@@ -624,7 +624,7 @@ def test_draw_on_image(self):
         kpi = ia.KeypointsOnImage(keypoints=kps, shape=(5, 5, 3))
         image = np.zeros((5, 5, 3), dtype=np.uint8) + 10
 
-        kps_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        kps_mask = np.zeros(image.shape[0:2], dtype=bool)
         kps_mask[2, 1] = 1
         kps_mask[4, 3] = 1
 
@@ -640,7 +640,7 @@ def test_draw_on_image_alpha_is_50_percent(self):
         kpi = ia.KeypointsOnImage(keypoints=kps, shape=(5, 5, 3))
         image = np.zeros((5, 5, 3), dtype=np.uint8) + 10
 
-        kps_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        kps_mask = np.zeros(image.shape[0:2], dtype=bool)
         kps_mask[2, 1] = 1
         kps_mask[4, 3] = 1
 
@@ -659,7 +659,7 @@ def test_draw_on_image_size_3(self):
         kpi = ia.KeypointsOnImage(keypoints=kps, shape=(5, 5, 3))
         image = np.zeros((5, 5, 3), dtype=np.uint8) + 10
 
-        kps_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        kps_mask = np.zeros(image.shape[0:2], dtype=bool)
         kps_mask[2, 1] = 1
         kps_mask[4, 3] = 1
 
@@ -678,7 +678,7 @@ def test_draw_on_image_blue_color(self):
         kpi = ia.KeypointsOnImage(keypoints=kps, shape=(5, 5, 3))
         image = np.zeros((5, 5, 3), dtype=np.uint8) + 10
 
-        kps_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        kps_mask = np.zeros(image.shape[0:2], dtype=bool)
         kps_mask[2, 1] = 1
         kps_mask[4, 3] = 1
 
@@ -694,7 +694,7 @@ def test_draw_on_image_single_int_as_color(self):
         kpi = ia.KeypointsOnImage(keypoints=kps, shape=(5, 5, 3))
         image = np.zeros((5, 5, 3), dtype=np.uint8) + 10
 
-        kps_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        kps_mask = np.zeros(image.shape[0:2], dtype=bool)
         kps_mask[2, 1] = 1
         kps_mask[4, 3] = 1
 
@@ -710,7 +710,7 @@ def test_draw_on_image_copy_is_false(self):
         kpi = ia.KeypointsOnImage(keypoints=kps, shape=(5, 5, 3))
         image = np.zeros((5, 5, 3), dtype=np.uint8) + 10
 
-        kps_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        kps_mask = np.zeros(image.shape[0:2], dtype=bool)
         kps_mask[2, 1] = 1
         kps_mask[4, 3] = 1
 
@@ -733,7 +733,7 @@ def test_draw_on_image_keypoint_is_outside_of_image(self):
         )
         image = np.zeros((5, 5, 3), dtype=np.uint8) + 10
 
-        kps_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        kps_mask = np.zeros(image.shape[0:2], dtype=bool)
         kps_mask[2, 1] = 1
         kps_mask[4, 3] = 1
 
@@ -766,7 +766,7 @@ def test_draw_on_image_one_kp_at_bottom_right_corner(self):
             shape=(5, 5, 3))
         image = np.zeros((5, 5, 3), dtype=np.uint8) + 10
 
-        kps_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        kps_mask = np.zeros(image.shape[0:2], dtype=bool)
         kps_mask[2, 1] = 1
         kps_mask[4, 3] = 1
 
@@ -784,7 +784,7 @@ def test_draw_on_image_one_kp_at_bottom_right_corner_and_raise_true(self):
             shape=(5, 5, 3))
         image = np.zeros((5, 5, 3), dtype=np.uint8) + 10
 
-        kps_mask = np.zeros(image.shape[0:2], dtype=np.bool)
+        kps_mask = np.zeros(image.shape[0:2], dtype=bool)
         kps_mask[2, 1] = 1
         kps_mask[4, 3] = 1
 
@@ -908,7 +908,7 @@ def test_to_keypoint_image_size_1(self):
 
         image = kpi.to_keypoint_image(size=1)
 
-        kps_mask = np.zeros((5, 5, 2), dtype=np.bool)
+        kps_mask = np.zeros((5, 5, 2), dtype=bool)
         kps_mask[2, 1, 0] = 1
         kps_mask[4, 3, 1] = 1
         assert np.all(image[kps_mask] == 255)
@@ -920,7 +920,7 @@ def test_to_keypoint_image_size_3(self):
 
         image = kpi.to_keypoint_image(size=3)
 
-        kps_mask = np.zeros((5, 5, 2), dtype=np.bool)
+        kps_mask = np.zeros((5, 5, 2), dtype=bool)
         kps_mask[2-1:2+1+1, 1-1:1+1+1, 0] = 1
         kps_mask[4-1:4+1+1, 3-1:3+1+1, 1] = 1
         assert np.all(image[kps_mask] >= 128)

test/augmentables/test_segmaps.py

@@ -107,7 +107,7 @@ def test_bool_arr_3d(self):
     def test_boolean_masks(self):
         # Test for #189 (boolean mask inputs into SegmentationMapsOnImage not
         # working)
-        for dt in [bool, np.bool]:
+        for dt in [bool, bool]:
             arr = np.array([
                 [0, 0, 0],
                 [0, 1, 0],

test/test_imgaug.py

@@ -225,7 +225,7 @@ class _Dummy(object):
         np.zeros((1, 2), dtype=np.float16),
         np.zeros((100,), dtype=np.float32),
         np.zeros((1, 2), dtype=np.float64),
-        np.zeros((1, 2), dtype=np.bool)
+        np.zeros((1, 2), dtype=bool)
     ]
     for value in values_true:
         assert ia.is_integer_array(value) is True
@@ -250,7 +250,7 @@ class _Dummy(object):
         np.zeros((1, 2), dtype=np.uint16),
         np.zeros((1, 2), dtype=np.int32),
         np.zeros((1, 2), dtype=np.int64),
-        np.zeros((1, 2), dtype=np.bool)
+        np.zeros((1, 2), dtype=bool)
     ]
     for value in values_true:
         assert ia.is_float_array(value) is True