训练DUSt3R模型

DUSt3R的介绍可参考本博客DUSt3R模型》

安装依赖

根据DUSt3R的github的说明，安装依赖

下载数据集

数据集分为很多个，需要分别下载

CO3D数据集

mkdir -p data/co3d_subset
cd data/co3d_subset
git clone https://github.com/facebookresearch/co3d
cd co3d
python3 ./co3d/download_dataset.py --download_folder ../ --single_sequence_subset
rm ../*.zip
cd ../../..

python3 datasets_preprocess/preprocess_co3d.py --co3d_dir data/co3d_subset --output_dir data/co3d_subset_processed  --single_sequence_subset

ARKitScenes数据集

1
2
3

cd data/
git clone https://github.com/apple/ARKitScenes.git
python3 download_data.py raw --video_id_csv raw/raw_train_val_splits.csv --download_dir arkit_data/ --raw_dataset_assets lowres_depth vga_wide vga_wide_intrinsics lowres_wide.traj

主要需要提前安装zip和unzip，否则会报错