fix bug with video output

app.py

@@ -27,8 +27,8 @@ IMAGES_PER_ROW = 7
 
 MAXIMAL_FRAMES = 1000
 BATCHES_TO_PROCESS = 10
-OUTPUT_FPS = 15
-MAX_OUT_FRAMES = 60
+OUTPUT_FPS = 10
+MAX_OUT_FRAMES = 40
 
 CAM_METHODS = {
     "GradCAM": GradCAM,
@@ -146,7 +146,9 @@ def gradcam_video(video, alpha=0.5, cam_method=GradCAM, layer=None, specific_cla
     cam.remove_hooks()
         
     # save video
-    fourcc = cv2.VideoWriter_fourcc(*'H264')
+    # 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:
@@ -255,7 +257,7 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped', css=css) as demo:
         # -------------------------------------------
         #                EXPLAIN
         # -------------------------------------------
-        with gr.Tab("Explain"):
+        with gr.Tab("Explain Image"):
             with gr.Row():
                 with gr.Column():
                     cam_method = gr.Radio(
@@ -314,7 +316,7 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped', css=css) as demo:
         # -------------------------------------------
         #                GIF CAM
         # -------------------------------------------
-        with gr.Tab("Gif Cam"):
+        with gr.Tab("Explain Video"):
             build_video_to_camvideo(CAM_METHODS, ALL_CLASSES, gradcam_video)
         
         # -------------------------------------------

src/gradio_blocks.py

@@ -52,8 +52,8 @@ def build_video_to_camvideo(CAM_METHODS, ALL_CLASSES, gradcam_video):
                 
         with gr.Column():
             with gr.Column():
-                video_in = gr.Video(autoplay=True, include_audio=False)
-                video_out = gr.Video(autoplay=True, include_audio=False)
+                video_in = gr.Video(autoplay=False, include_audio=False)
+                video_out = gr.Video(autoplay=False, include_audio=False)
                 
             gif_cam_mode_button = gr.Button(value="Show GradCAM-Video", label="GradCAM", scale=1)
             gif_cam_mode_button.click(fn=gradcam_video, inputs=[video_in, video_alpha, video_cam_method, video_layer, video_animal_to_explain], outputs=[video_out], queue=True)

src/results/gradcam_video.mp4

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