Fix cvat format (#2071)

6 years ago · 8cbe3956a3
parent 0982ea3f57
commit 8cbe3956a3
1 changed files with 6 additions and 3 deletions
--- a/cvat/apps/dataset_manager/formats/cvat.py
+++ b/cvat/apps/dataset_manager/formats/cvat.py
@ -398,8 +398,9 @@ def dump_as_cvat_interpolation(file_object, annotations):
                z_order=shape.z_order,
                frame=shape.frame,
                attributes=shape.attributes,
-            ),
-            annotations.TrackedShape(
+            )] +
+            ( # add a finishing frame if it does not hop over the last frame
+            [annotations.TrackedShape(
                type=shape.type,
                points=shape.points,
                occluded=shape.occluded,
@ -408,8 +409,10 @@ def dump_as_cvat_interpolation(file_object, annotations):
                z_order=shape.z_order,
                frame=shape.frame + annotations.frame_step,
                attributes=shape.attributes,
+            )] if shape.frame + annotations.frame_step < \
+                    int(annotations.meta['task']['stop_frame']) \
+               else []
            ),
-            ],
        ))
        counter += 1