load visiblity from annotation file in datumaro format

openvinotoolkit · Oct 15, 2024 · 80ad1dd · 80ad1dd
1 parent 94360b2
commit 80ad1dd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/datumaro/plugins/data_formats/datumaro/base.py b/src/datumaro/plugins/data_formats/datumaro/base.py
@@ -339,6 +339,7 @@ def _load_annotations(self, item: Dict):
                             points,
                             label=label_id,
                             id=ann_id,
+                            visibility=ann.get("visibility"),
                             attributes=attributes,
                             group=group,
                             object_id=object_id,

diff --git a/tests/unit/data_formats/datumaro/conftest.py b/tests/unit/data_formats/datumaro/conftest.py
@@ -91,14 +91,15 @@ def fxt_test_datumaro_format_dataset():
                         },
                     ),
                     Points(
-                        [1, 2, 2, 0, 1, 1],
+                        [1, 2, 0, 0, 1, 1],
                         label=0,
                         id=5,
                         z_order=4,
                         attributes={
                             "x": 1,
                             "y": "2",
                         },
+                        visibility=[1, 0, 2],
                     ),
                     Mask(
                         label=3,