Skip to content

Commit a060e5b

Browse files
committed
fix: ensure video is closed
1 parent 21b0c46 commit a060e5b

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

tracklab/visualization/visualization_engine.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,9 @@ def visualize(self, tracker_state: TrackerState, video_id, detections, image_pre
128128
video_writer.write(output_image)
129129
progress.on_module_step_end(None, "vis", None, None)
130130

131+
if self.save_videos:
132+
video_writer.release()
133+
131134
def draw_frame(self, image_metadata, detections_pred, detections_gt,
132135
image_pred, image_gt, nframes):
133136
image = cv2_load_image(image_metadata.file_path)

0 commit comments

Comments
 (0)