git clone https://github.com/bmild/nerf.git
cd nerf
conda虚拟环境里实现
conda create -n nerf python=3.7
conda activate nerf
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
进入nerf项目中,将requirements.txt文件中修改为(注释前两行)
然后执行:
pip install -r requirements.txt
bash download_example_data.sh
https://drive.google.com/drive/folders/128yBriW1IG_3NJ5Rp7APSTZsJqdJdfc1
下载好后解压放在nerf/data里面
根据自己的数据路径相应的修改一下configs/fern.txt文件里的文件路径:
修改run_nerf.py里面的数据路径:
默认为200K次
在命令行添加config的参数
python run_nerf.py --config configs/fern.txt
在run_nerf.py里面修改config路径:
parser.add_argument('--config', is_config_file=True, default='./configs/fern.txt',
help='config file path')
然后执行:
python run_nerf.py
开始训练时运行截图:
dbq有点错位了
除了自己训练数据集之外,你还可以下载预训练模型
https://drive.google.com/drive/folders/1jIr8dkvefrQmv737fFm2isiT6tqpbTbv
我这里下载了flower_test,放在logs文件夹下
然后执行:
python run_nerf.py --config configs/flower.txt --render_only
开始训练时运行截图:
训练完成:
查看渲染结果 在 ./logs/flower_test/renderonly_path_100000里面: