Spaces:

GabrielML
/

Explain-Animal-CNN

Sleeping

GabrielML commited on Oct 3, 2023

Commit

ec26c06

1 Parent(s): 0fdea39

fix bug with video output

Files changed (1) hide show

app.py CHANGED Viewed

@@ -146,7 +146,7 @@ def gradcam_video(video, alpha=0.5, cam_method=GradCAM, layer=None, specific_cla
     cam.remove_hooks()
     # save video
-    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     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:

     cam.remove_hooks()
     # save video
+    fourcc = cv2.VideoWriter_fourcc(*'H264')
     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: