Do not create last job shorter than overlap size

retailnext · Oct 25, 2023 · c9f41ea · c9f41ea
1 parent c19ea38
commit c9f41ea
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/cvat/apps/engine/task.py b/cvat/apps/engine/task.py
@@ -141,12 +141,14 @@ def _segments():
 
         segment_step -= overlap
 
-        segments = (
+        segments = [
             SegmentParams(start_frame, min(start_frame + segment_size - 1, data_size - 1))
             for start_frame in range(0, data_size, segment_step)
-        )
+        ]
+        if segments[-1].stop_frame - segments[-1].start_frame <= overlap:
+            segments.pop(-1)
 
-    return SegmentsParams(segments, segment_size, overlap)
+    return SegmentsParams(iter(segments), segment_size, overlap)
 
 def _save_task_to_db(db_task: models.Task, *, job_file_mapping: Optional[JobFileMapping] = None):
     job = rq.get_current_job()