Save mask as uint8 for better compatibility with other tools

wkentaro · Jun 13, 2024 · 9085123 · 9085123
1 parent 906a217
commit 9085123
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/labelme/app.py b/labelme/app.py
@@ -10,6 +10,7 @@
 
 import imgviz
 import natsort
+import numpy as np
 from qtpy import QtCore
 from qtpy import QtGui
 from qtpy import QtWidgets
@@ -1303,7 +1304,9 @@ def format_shape(s):
                     description=s.description,
                     shape_type=s.shape_type,
                     flags=s.flags,
-                    mask=None if s.mask is None else utils.img_arr_to_b64(s.mask),
+                    mask=None
+                    if s.mask is None
+                    else utils.img_arr_to_b64(s.mask.astype(np.uint8)),
                 )
             )
             return data

diff --git a/labelme/label_file.py b/labelme/label_file.py
@@ -111,7 +111,9 @@ def load(self, filename):
                     flags=s.get("flags", {}),
                     description=s.get("description"),
                     group_id=s.get("group_id"),
-                    mask=utils.img_b64_to_arr(s["mask"]) if s.get("mask") else None,
+                    mask=utils.img_b64_to_arr(s["mask"]).astype(bool)
+                    if s.get("mask")
+                    else None,
                     other_data={k: v for k, v in s.items() if k not in shape_keys},
                 )
                 for s in data["shapes"]