Fix an issue with occluded field

cvat-ai · Feb 14, 2023 · 7b3ff12 · 7b3ff12
1 parent 5f68cfa
commit 7b3ff12
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cvat/apps/dataset_manager/formats/cvat.py b/cvat/apps/dataset_manager/formats/cvat.py
@@ -742,6 +742,11 @@ def dump_labeled_shapes(shapes, is_skeleton=False):
                         ("outside", str(int(shape.outside)))
                     ]))
 
+                if shape.type != 'skeleton':
+                    dump_data.update(OrderedDict([
+                        ("occluded", str(int(shape.occluded)))
+                    ]))
+
                 if shape.type == "rectangle":
                     dump_data.update(OrderedDict([
                         ("xtl", "{:.2f}".format(shape.points[0])),
@@ -795,7 +800,6 @@ def dump_labeled_shapes(shapes, is_skeleton=False):
                     ]))
                 elif shape.type != 'skeleton':
                     dump_data.update(OrderedDict([
-                        ("occluded", str(int(shape.occluded))),
                         ("points", ';'.join((
                             ','.join((
                                 "{:.2f}".format(x),