FIX ffmpeg codec err

app.py

@@ -309,16 +309,16 @@ def tracking_objects(Seg_Tracker, frame_num, input_video):
     # output_frames = int(total_frames * scale_slider)
     output_frames = len([name for name in os.listdir(output_combined_dir) if os.path.isfile(os.path.join(output_combined_dir, name)) and name.endswith('.png')])
     out_fps = fps * output_frames / total_frames
-    ffmpeg.input(os.path.join(output_combined_dir, '%07d.png'), framerate=out_fps).output(output_video_path, vcodec='h264_nvenc', pix_fmt='yuv420p').run()
-    # fourcc = cv2.VideoWriter_fourcc(*"mp4v")
-    # out = cv2.VideoWriter(output_video_path, fourcc, out_fps, (frame_width, frame_height))
+    # ffmpeg.input(os.path.join(output_combined_dir, '%07d.png'), framerate=out_fps).output(output_video_path, vcodec='h264_nvenc', pix_fmt='yuv420p').run()
+    fourcc = cv2.VideoWriter_fourcc(*"mp4v")
+    out = cv2.VideoWriter(output_video_path, fourcc, out_fps, (frame_width, frame_height))
 
-    # for i in range(output_frames):
-    #     frame_path = os.path.join(output_combined_dir, f'{i:07d}.png')
-    #     frame = cv2.imread(frame_path)
-    #     out.write(frame)
+    for i in range(output_frames):
+        frame_path = os.path.join(output_combined_dir, f'{i:07d}.png')
+        frame = cv2.imread(frame_path)
+        out.write(frame)
 
-    # out.release()
+    out.release()
 
     zip_folder(output_masks_dir, output_zip_path)
     print("done")