|
|
|
|
|
|
|
import os |
|
''' |
|
pip install youtube-dl==2020.12.12 |
|
''' |
|
import subprocess |
|
import pandas as pd |
|
|
|
def download_youtube_video(video_url, output_path): |
|
""" |
|
:param video_url: youtube video url |
|
:param output_dir: file path to save |
|
""" |
|
|
|
try: |
|
if not os.path.exists(os.path.dirname(output_path)): |
|
os.makedirs(os.path.dirname(output_path), exist_ok=True) |
|
|
|
command = ['yt-dlp', '-f', 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]', '--merge-output-format', |
|
'mp4', '--output', output_path , video_url] |
|
|
|
result = subprocess.run(command, capture_output=True, text=True, encoding='utf-8') |
|
|
|
if result.returncode == 0: |
|
print('Download {:s} successfully!'.format(video_url)) |
|
else: |
|
print("Fail to download {:s}, error info:\n{:s}".format(video_url, result.stderr)) |
|
except Exception as e: |
|
print(f"error: {e}") |
|
|
|
if __name__ == '__main__': |
|
df = pd.read_csv("./echomimicv2_benchmark_url+start_timecode+end_timecode.txt") |
|
save_dir = "ori_video_dir" |
|
urls = list(set(df['URL'])) |
|
video_output_paths = [os.path.join(save_dir, url.split('v=')[1]+".mp4") for url in urls] |
|
|
|
for video_url, output_path in zip(urls, video_output_paths): |
|
download_youtube_video(video_url, output_path) |
|
|
