forked from yxuhan/AdaMPI
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathadd_audio_to_video.py
37 lines (27 loc) · 1 KB
/
add_audio_to_video.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import warnings
warnings.filterwarnings("ignore")
from pathlib import Path
import argparse
from PIL import Image
import cv2
import glob
import os
import numpy as np
import gc
import moviepy
from moviepy.editor import ImageSequenceClip
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--video_path', type=str, default="")
parser.add_argument('--audio_path', type=str, default="")
opt, _ = parser.parse_known_args()
# python add_audio_to_video.py --video_path /AdaMPI/data/ipmanVR_LR.mp4 --audio_path /AdaMPI/data/audio.mp3
print("video_path:", opt.video_path)
print("audio_path:", opt.audio_path)
audio_clip = moviepy.editor.AudioFileClip(opt.audio_path)
if audio_clip is not None:
print("add audio")
video_clip = moviepy.editor.VideoFileClip(opt.video_path)
audio_clip = audio_clip.subclip(0, video_clip.end)
final_clip = video_clip.set_audio(audio_clip)
final_clip.write_videofile(opt.video_path[:-4] + '_with_audio.mp4')
print("Done")