DenseAV-Lowell

Running

lorocksUMD commited on Mar 26

Commit

32f1889

verified ·

1 Parent(s): ecbe2cb

Update DenseAV/denseav/plotting.py

Files changed (1) hide show

DenseAV/denseav/plotting.py CHANGED Viewed

@@ -46,7 +46,7 @@ def write_video_with_audio(video_frames, audio_array, video_fps, audio_fps, outp
         wavfile.write(temp_audio_path, audio_fps, audio_array.cpu().to(torch.float64).permute(1, 0).numpy())
         video_clip = VideoFileClip(temp_video_path)
         audio_clip = AudioFileClip(temp_audio_path)
-        final_clip = video_clip.set_audio(audio_clip)
         final_clip.write_videofile(output_path, codec='libx264', verbose=False)
         os.remove(temp_video_path)
         os.remove(temp_audio_path)

         wavfile.write(temp_audio_path, audio_fps, audio_array.cpu().to(torch.float64).permute(1, 0).numpy())
         video_clip = VideoFileClip(temp_video_path)
         audio_clip = AudioFileClip(temp_audio_path)
+        final_clip = video_clip.with_audio(audio_clip)
         final_clip.write_videofile(output_path, codec='libx264', verbose=False)
         os.remove(temp_video_path)
         os.remove(temp_audio_path)