CF-3DGS

> 编程语言

开源

COLMAP-Free 3D Gaussian Splatting 杨付 · 刘思飞 · 古尔卡尼 · 扬·考茨阿列克谢·阿弗罗斯 · 王晓龙论文 | 视频 | 项目 P…

785 stars0 点赞0 次浏览

访问官网 GitHub

工具介绍

COLMAP-Free 3D Gaussian Splatting 杨付 · 刘思飞 · 古尔卡尼 · 扬·考茨阿列克谢·阿弗罗斯 · 王晓龙论文 | 视频 | 项目 P…

COLMAP-Free 3D 高斯分散杨福 · 刘思飞 · Amey Kulkarni · Jan Kautz Alexei A. Efros · 王小龙论文 | 视频 | 项目页面

安装

(推荐)

这些代码已在 Python 3.10 和 CUDA >= 11.6 上进行了测试。要安装所有依赖项，最简单的方法是使用 anaconda 和 pip，按照以下步骤操作：

bash

conda create -n cf3dgs python=3.10
conda activate cf3dgs
conda install conda-forge::cudatoolkit-dev=11.7.0
conda install pytorch==2.0.0 torchvision==0.15.0 pytorch-cuda=11.7 -c pytorch -c nvidia
git clone --recursive [email protected]:NVlabs/CF-3DGS.git
pip install -r requirements.txt

数据集准备

默认情况下,DATAROOT 为 ./data。请先使用 mkdir data 创建数据文件夹。

战车和庙宇

下载 Nope-NeRF 预处理的数据,如下所示,数据保存在 ./data/Tanks 文件夹中。

bash

wget https://www.robots.ox.ac.uk/~wenjing/Tanks.zip

CO3D

下载我们预处理的数据,并将其保存到 ./data/co3d 文件夹中。

运行

训练

bash

python run_cf3dgs.py -s data/Tanks/Francis \ # change the scene path
                     --mode train \
                     --data_type tanks

评估

bash

# pose estimation
python run_cf3dgs.py --source data/Tanks/Francis \
                     --mode eval_pose \
                     --data_type tanks \
                     --model_path ${CKPT_PATH} 
# by default the checkpoint should be store in "./output/progressive/Tanks_Francis/chkpnt/ep00_init.pth"
# novel view synthesis
python run_cf3dgs.py --source data/Tanks/Francis \
                     --mode eval_nvs \
                     --data_type tanks \
                     --model_path ${CKPT_PATH}

我们发布了一些新的视图合成结果 ([gdrive](https://drive.google.com/drive/folders/1p3WljCN90zrm1N5lO-24OLHmUFmFWntt?usp