Merge pull request #1 from felipe-parodi/fp-fix-coco-import-bug

Fix COCO Dataset Loading for Invisible Keypoints
talmolab · Dec 5, 2024 · 8345ea5 · 8345ea5
2 parents fff8761 + 23aba1e
commit 8345ea5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sleap/io/format/coco.py b/sleap/io/format/coco.py
@@ -180,6 +180,9 @@ def read(
 
                 if flag == 0:
                     # node not labeled for this instance
+                    if (x, y) != (0, 0): 
+                        # If labeled but invisible, place the node at the coord
+                        points[node] = Point(x, y, False)
                     continue
 
                 is_visible = flag == 2