前言：目前互联网上能搜到下载RealEstate10K数据集原始视频的方法都已经不能用了，这篇博客介绍一种目前可用的下载RealEstate10K数据集原始视频的方法，并给出自动化的脚本代码。
目录
RealEstate10K简介
RealEstate10K标注文本下载
RealEstate10K原始视频下载
环境安装
完整代码
RealEstate10K简介
RealEstate10K是一个庞大的相机姿势数据集，从大约1万个YouTube视频中收集到大约8万个视频片段，对应1000万帧。对于每个剪辑，姿态形成一个轨迹，其中每个姿态指定摄像机沿轨迹的位置和方向。这些姿势是通过在大量视频上运行SLAM和捆绑调整算法得到的。
该数据集为谷歌2018年的SIGGRAPH论文《立体放大:使用多平面图像学习视图合成》提供了支持，该论文学习使用RealEstate10K等训练数据将窄基线立体对转换为迷你光场。该数据集旨在帮助研究人员在视图合成，3D计算机视觉等方面的工作。
RealEstate10K标注文本下载
进入网址：https://google.github.io/realestate10k/index.html
点击下载：
RealEstate10K原始视频下载
环境安装
pip install yt-dlp
pip install tqdm完整代码
import os
from tqdm import tqdm
import subprocess
def video_is_exit(title, video_folder):
for filename in os.listdir(video_folder):
if title in filename:
return True
return False
input_folder = "RealEstate10K/train"
save_folder = "RealEstate10K_videos/train"
os.makedirs(save_folder, exist_ok=True)
if __name__ == "__main__":
# 遍历文件夹中的所有文件
for filename in tqdm(os.listdir(input_folder)):
# 检查文件扩展名是否为.txt
if filename.endswith('.txt'):
save_name = filename.split('.')[0] + ".mp4"
save_path = os.path.join(save_folder, save_name)
file_path = os.path.join(input_folder, filename)
# 打开文件
with open(file_path, 'r', encoding='utf-8') as file:
# 这里可以添加读取文件内容的代码
url = file.readline().strip()
if not video_is_exit(save_name, save_folder):
bash_command = f"yt-dlp -o {save_path} {url}"
result = subprocess.run(bash_command, shell=True, text=True, capture_output=True)
else:
print(f"{save_path} is exist! ")
