| |
| |
|
|
| 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) |
|
|
