Tensorflow implementation of optimizing a neural representation for a single scene and rendering new views.
NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis
Ben Mildenhall*1,
Pratul P. Srinivasan*1,
Matthew Tancik*1,
Jonathan T. Barron2,
Ravi Ramamoorthi3,
Ren Ng1
1UC Berkeley, 2Google Research, 3UC San Diego
*denotes equal contribution
Python 3 dependencies:
- Tensorflow 1.15
- matplotlib
- numpy
- imageio
- configargparse
The LLFF data loader requires ImageMagick.
You will also need the LLFF code (and COLMAP) set up to compute poses if you want to run on your own real data.
A neural radiance field is a simple fully connected network (weights are ~5MB) trained to reproduce input views of a single scene using a rendering loss. The network directly maps from spatial location and viewing direction (5D input) to color and opacity (4D output), acting as the "volume" so we can use volume rendering to differentiably render new views.
Optimizing a NeRF takes between a few hours and a day or two (depending on resolution) and only requires a single GPU. Rendering an image from an optimized NeRF takes somewhere between less than a second and ~30 seconds, again depending on resolution.
Run
bash download_example_data.sh
to get the our synthetic Lego dataset and the LLFF Fern dataset. To optimize a low-res Fern NeRF:
python run_nerf.py --config config_fern.txt
To optimize a low-res Lego NeRF:
python run_nerf.py --config config_lego.txt
Run
bash download_example_weights.sh
to get a pretrained high-res NeRF for the Fern dataset. Now you can use the render_demo.ipynb to render new views.
We recommend using the imgs2poses.py script from the LLFF code. Then you can pass the base scene directory into our code using --datadir <myscene> along with -dataset_type llff. You can take a look at the config_fern.txt config file for example settings to use for a forward facing scene.