fix bug with video output

app.py

@@ -20,7 +20,11 @@ from util import transform
 from gradio_blocks import build_video_to_camvideo
 import cv2
 import ffmpeg
+import shutil
+import mediapy
 
+ffmpeg_path = shutil.which('ffmpeg')
+mediapy.set_ffmpeg(ffmpeg_path)
 
 IMAGE_PATH = os.path.join(os.getcwd(), 'src/examples')
 IMAGES_PER_ROW = 7
@@ -148,11 +152,12 @@ def gradcam_video(video, alpha=0.5, cam_method=GradCAM, layer=None, specific_cla
     # save video
     # fourcc = cv2.VideoWriter_fourcc(*'AVC1')
     # fourcc = cv2.VideoWriter_fourcc(*'MP4V')
-    fourcc = cv2.VideoWriter_fourcc(*'XVID')
-    size = (results[0].shape[1], results[0].shape[0])
-    video = cv2.VideoWriter('src/results/gradcam_video.mp4', fourcc, OUTPUT_FPS, size)
-    for frame in results:
-        video.write(frame)
+    # fourcc = cv2.VideoWriter_fourcc(*'XVID')
+    # size = (results[0].shape[1], results[0].shape[0])
+    # video = cv2.VideoWriter('src/results/gradcam_video.mp4', fourcc, OUTPUT_FPS, size)
+    # for frame in results:
+    #     video.write(frame)
+    mediapy.write_video('src/results/gradcam_video.mp4', results, fps=OUTPUT_FPS)
     video.release()
     return 'src/results/gradcam_video.mp4'
 

packages.txt

@@ -0,0 +1 @@
+ffmpeg

src/results/gradcam_video.mp4

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:23fecbd3a23e2abfa91532fe87ebeb754a69897d583232de98a1a07a2819f0d1
-size 1049141
+oid sha256:0b5c5763b98e7c7c4e0191e120a415cf50580aeb3d15f8c6fdee8e0a3a50fe05
+size 648305