Use tight bounding box to represent bbox for shape_type="mask"

wkentaro · Dec 30, 2023 · f52c879 · f52c879
1 parent 68f6b3c
commit f52c879
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/labelme/widgets/canvas.py b/labelme/widgets/canvas.py
@@ -768,12 +768,12 @@ def paintEvent(self, event):
                 points=[[point.x(), point.y()] for point in drawing_shape.points],
                 point_labels=drawing_shape.point_labels,
             )
-            y1, x1, y2, x2 = imgviz.instances.mask_to_bbox([mask])[0].astype(int)
+            y1, x1, y2, x2 = imgviz.instances.masks_to_bboxes([mask])[0].astype(int)
             drawing_shape.setShapeRefined(
                 shape_type="mask",
                 points=[QtCore.QPointF(x1, y1), QtCore.QPointF(x2, y2)],
                 point_labels=[1, 1],
-                mask=mask[y1:y2, x1:x2],
+                mask=mask[y1 : y2 + 1, x1 : x2 + 1],
             )
             drawing_shape.selected = True
             drawing_shape.paint(p)
@@ -818,12 +818,12 @@ def finalise(self):
                 points=[[point.x(), point.y()] for point in self.current.points],
                 point_labels=self.current.point_labels,
             )
-            y1, x1, y2, x2 = imgviz.instances.mask_to_bbox([mask])[0].astype(int)
+            y1, x1, y2, x2 = imgviz.instances.masks_to_bboxes([mask])[0].astype(int)
             self.current.setShapeRefined(
                 shape_type="mask",
                 points=[QtCore.QPointF(x1, y1), QtCore.QPointF(x2, y2)],
                 point_labels=[1, 1],
-                mask=mask[y1:y2, x1:x2],
+                mask=mask[y1 : y2 + 1, x1 : x2 + 1],
             )
         self.current.close()
 

diff --git a/setup.py b/setup.py
@@ -22,7 +22,7 @@ def get_version():
 def get_install_requires():
     install_requires = [
         "gdown",
-        "imgviz>=0.11",
+        "imgviz>=1.7.5",
         "matplotlib",
         "natsort>=7.1.0",
         "numpy",