nerf.rst 1.84 KB
Newer Older
1
.. _`Vanilla NeRF Example`:
Ruilong Li(李瑞龙)'s avatar
Ruilong Li(李瑞龙) committed
2

3
4
Vanilla NeRF
====================
Ruilong Li(李瑞龙)'s avatar
Ruilong Li(李瑞龙) committed
5
6
7
See code `examples/train_mlp_nerf.py` at our `github repository`_ for details.


8
9
10
11
Radiance Field
--------------
We follow the original `NeRF`_ paper to implement a 8-layer-MLP radiance field (`examples/radiance_fields/mlp.py`)
with positional encoding. 
Ruilong Li(李瑞龙)'s avatar
Ruilong Li(李瑞龙) committed
12
13
14
15
16
17
18

.. note:: 
    The vanilla Nerf paper uses two MLPs for course-to-fine sampling. Instead here we only use a 
    single MLP with more samples (1024). Both ways share the same spirit to do dense sampling 
    around the surface. Our fast rendering inheritly skip samples away from the surface 
    so we can simplly increase the number of samples with a single MLP, to achieve the same goal 
    with the coarse-to-fine sampling, without runtime or memory issue.
Ruilong Li(李瑞龙)'s avatar
Ruilong Li(李瑞龙) committed
19

20
21
22
23
24
25
26
27

Benchmark: Nerf-Synthetic Dataset
---------------------------------
*updated on 2022-10-08*

Our experiments are conducted on a single NVIDIA TITAN RTX GPU. 
The training memory footprint is about 10GB.

28
29
30
31
32
33
+----------------------+-------+-------+---------+-------+-------+-------+-------+-------+-------+
| PSNR                 | Lego  | Mic   |Materials| Chair |Hotdog | Ficus | Drums | Ship  | MEAN  |
|                      |       |       |         |       |       |       |       |       |       |
+======================+=======+=======+=========+=======+=======+=======+=======+=======+=======+
| NeRF  (~ days)       | 32.54 | 32.91 | 29.62   | 33.00 | 36.18 | 30.13 | 25.01 | 28.65 | 31.00 |
+----------------------+-------+-------+---------+-------+-------+-------+-------+-------+-------+
34
| Ours  (~ 1 hr)       | 33.69 | 33.76 | 29.73   | 33.32 | 35.80 | 32.52 | 25.39 | 28.18 | 31.55 |
35
+----------------------+-------+-------+---------+-------+-------+-------+-------+-------+-------+
Ruilong Li(李瑞龙)'s avatar
Ruilong Li(李瑞龙) committed
36

37
38
.. _`github repository`: https://github.com/KAIR-BAIR/nerfacc/
.. _`NeRF`: https://arxiv.org/abs/2003.08934