Merge pull request #26 from ksugar/dev

v0.6.0 -> v0.6.1
ksugar · Sep 3, 2024 · 914703a · 914703a
2 parents 3bad8a4 + 890bba8
commit 914703a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 12 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "samapi"
-version = "0.6.0"
+version = "0.6.1"
 description = "Segment Anything Model (SAM) API"
 authors = ["Ko Sugawara <[email protected]>"]
 readme = "README.md"

diff --git a/src/samapi/main.py b/src/samapi/main.py
@@ -758,18 +758,19 @@ async def video_predictor(body: SAMVideoBody):
                 "t": frame_idx,
             }
         for obj_id, mask in masks.items():
-            geometry = mask_to_geometry(mask[0])
-            geometry["plane"] = plane
-            features.append(
-                Feature(
-                    geometry=geometry,
-                    properties={
-                        "object_idx": obj_id,
-                        "label": "object",
-                        "sam_model": body.type,
-                    },
+            if np.any(mask):
+                geometry = mask_to_geometry(mask[0])
+                geometry["plane"] = plane
+                features.append(
+                    Feature(
+                        geometry=geometry,
+                        properties={
+                            "object_idx": obj_id,
+                            "label": "object",
+                            "sam_model": body.type,
+                        },
+                    )
                 )
-            )
     return features