Debugging: Ensure that mask is integer after scaling

hajicj · Oct 15, 2018 · 82f9c7b · 82f9c7b
1 parent 069c7d8
commit 82f9c7b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/muscima/cropobject.py b/muscima/cropobject.py
@@ -1093,6 +1093,7 @@ def scale(self, zoom=1.0):
                                             output_shape=new_mask_shape)
         new_mask[new_mask >= 0.5] = 1
         new_mask[new_mask < 0.5] = 0
+        new_mask = new_mask.astype('uint8')
 
         new_height, new_width = new_mask.shape
         new_top = int(self.top * zoom)