Commit 4419a2c4 authored by Ruilong Li's avatar Ruilong Li
Browse files

every 5k steps; examples doc

parent 15f1c2a5
Instant-NGP Instant-NGP
==================== ====================
See code at our github repository: https://github.com/KAIR-BAIR/nerfacc/tree/master/examples/ See code `examples/train_ngp_nerf.py` at our `github repository`_ for details.
Benchmarks Benchmarks
------------ ------------
We trained on NeRF-Synthetic trainval set using TITAN RTX, and evaluated on testset. Here we trained a NGP Nerf model on the NeRF-Synthetic dataset. We follow the same
Note Instant-NGP's results are taken from the paper, which is trained on a Nvidia 3090, settings with the paper, which uses trainval split for training and test split for
with random background trick using alpha channel. evaluation. Our experiments are conducted on a single NVIDIA TITAN RTX GPU.
+----------------------+----------+----------+------------+-------+--------+ .. note::
| | Lego | Mic | Materials |Chair |Hotdog |
| | | | | | | The paper makes use of the alpha channel in the images to apply random background
+======================+==========+==========+============+=======+========+ augmentation during training. Yet we only uses a constant white background.
| Paper (PSNR: 5min) | 36.39 | 36.22 | 29.78 | 35.00| 37.40 |
+----------------------+----------+----------+------------+-------+--------+ +----------------------+----------+----------+------------+-------+--------+--------+--------+--------+
| Ours (PSNR) | 36.61 | 37.45 | 30.15 | 36.06 | 38.17 | | | Lego | Mic | Materials |Chair |Hotdog | Ficus | Drums | Ship |
+----------------------+----------+----------+------------+-------+--------+ | | | | | | | | | |
| Ours (Training time)| 300s | 272s | 258s | 331s | 287s | +======================+==========+==========+============+=======+========+========+========+========+
+----------------------+----------+----------+------------+-------+--------+ | Paper (PSNR: 5min) | 36.39 | 36.22 | 29.78 | 35.00 | 37.40 | 33.51 | 26.02 | 31.10 |
+----------------------+----------+----------+------------+-------+--------+--------+--------+--------+
| Ours (PSNR) | 36.61 | 37.45 | 30.15 | 36.10 | 38.17 | | 25.83 | |
+----------------------+----------+----------+------------+-------+--------+--------+--------+--------+
| Ours (Training time)| 300s | 272s | 258s | 311s | 287s | | 249s | |
+----------------------+----------+----------+------------+-------+--------+--------+--------+--------+
.. _`github repository`: : https://github.com/KAIR-BAIR/nerfacc/
\ No newline at end of file
Vanilla Nerf
====================
...@@ -170,7 +170,7 @@ if __name__ == "__main__": ...@@ -170,7 +170,7 @@ if __name__ == "__main__":
optimizer.step() optimizer.step()
scheduler.step() scheduler.step()
if step % 100 == 0: if step % 5000 == 0:
elapsed_time = time.time() - tic elapsed_time = time.time() - tic
loss = F.mse_loss(rgb[alive_ray_mask], pixels[alive_ray_mask]) loss = F.mse_loss(rgb[alive_ray_mask], pixels[alive_ray_mask])
print( print(
......
...@@ -196,7 +196,7 @@ if __name__ == "__main__": ...@@ -196,7 +196,7 @@ if __name__ == "__main__":
optimizer.step() optimizer.step()
scheduler.step() scheduler.step()
if step % 100 == 0: if step % 5000 == 0:
elapsed_time = time.time() - tic elapsed_time = time.time() - tic
loss = F.mse_loss(rgb[alive_ray_mask], pixels[alive_ray_mask]) loss = F.mse_loss(rgb[alive_ray_mask], pixels[alive_ray_mask])
print( print(
......
...@@ -201,7 +201,7 @@ if __name__ == "__main__": ...@@ -201,7 +201,7 @@ if __name__ == "__main__":
optimizer.step() optimizer.step()
scheduler.step() scheduler.step()
if step % 100 == 0: if step % 5000 == 0:
elapsed_time = time.time() - tic elapsed_time = time.time() - tic
loss = F.mse_loss(rgb[alive_ray_mask], pixels[alive_ray_mask]) loss = F.mse_loss(rgb[alive_ray_mask], pixels[alive_ray_mask])
print( print(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment