Official implementation of 'Text2NeRF: Text-Driven 3D Scene Generation with Neural Radiance Fields'. Note: This code is forked from TensoRF.
Install environment:
conda env create -f environment.yml
conda activate text2nerf
pip install -r requirements.txt
The training script is in text2nerf_main.py
:
CUDA_VISIBLE_DEVICES=0 python text2nerf_main.py --config 'configs/text2nerf_scenes.txt' --expname 'text000' --prompt 'a beautiful garden' --datadir 'data_example/text000' --pose_traj 'local_fixed' --regen_pose
CUDA_VISIBLE_DEVICES=0 python text2nerf_main.py --config 'configs/text2nerf_scenes.txt' --expname 'text000_360' --prompt 'a beautiful garden' --datadir 'data_example/text000' --pose_traj 'circle' --regen_pose
CUDA_VISIBLE_DEVICES=0 python text2nerf_main.py --config 'configs/text2nerf_scenes.txt' --expname 'text000' --prompt 'a beautiful garden' --datadir 'data_example/text000' --pose_traj 'local_fixed' --regen_pose --render_only 1 --render_test 1 --ckpt '[path of ckpt]'
If you find our code or paper helps, please consider citing:
@article{zhang2023text2nerf,
title={Text2NeRF: Text-Driven 3D Scene Generation with Neural Radiance Fields},
author={Zhang, Jingbo and Li, Xiaoyu and Wan, Ziyu and Wang, Can and Liao, Jing},
journal={arXiv preprint arXiv:2305.11588},
year={2023}
}