fix interpolation

SIPEC-Animal-Data-Analysis · Aug 5, 2024 · 8b3360e · 8b3360e
1 parent db78ca0
commit 8b3360e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/SwissKnife/mrcnn/utils.py b/SwissKnife/mrcnn/utils.py
@@ -529,10 +529,10 @@ def minimize_mask(bbox, mask, mini_shape):
 
     See inspect_data.ipynb notebook for more details.
     """
-    mini_mask = np.zeros(mini_shape + (mask.shape[-1],), dtype=bool)
+    mini_mask = np.zeros((mini_shape[0], mini_shape[1], mask.shape[-1]), dtype=np.uint8)
     for i in range(mask.shape[-1]):
         # Pick slice and cast to bool in case load_mask() returned wrong dtype
-        m = mask[:, :, i].astype(bool)
+        m = mask[:, :, i]
         y1, x1, y2, x2 = bbox[i][:4]
         m = m[y1:y2, x1:x2]
         if m.size == 0: