Researchers From UC Berkeley Develop NerfAcc, A PyTorch Nerf Acceleration Toolbox For Both Training And Inference

Neural Radiance Fields (NeRFs) is a revolutionary approach for 3D representation that uses a multi-layer perceptron to describe the geometry and view-dependent appearance of the scene (MLP). During the last two years, they have been solid in several downstream 3D applications, such as static/dynamic scene reconstruction, relighting, and content production. A radiance field may be optimized/trained from calibrated RGB photos alone with differentiable volumetric rendering. However, with the vanilla MLP NeRF, such an optimization procedure typically takes days to converge on a single scene and tens of seconds to display a single picture.

Many recent studies have shown that integrating explicit 3D voxels into the radiance field representation may significantly expedite the NeRF optimization and inference process. However, because they are only relevant to a single, static scene, voxel-based radiance fields are less adaptable than MLP-based representations. Those approaches have yet to benefit applications such as dynamic scene reconstruction and generalization across many locations. Regardless of the variety in radiance field representations and uses, most use the same volumetric rendering process: cast a ray into space and gather colors along the beam.

Until now, there has been little attention towards building a workable volumetric rendering technique that can be used for any downstream applications. Several efforts, notably Plenoctrees and Instant-NGP, use efficient volumetric rendering techniques such as bypassing empty and occluded regions. They are, however, heavily tuned for constrained static scenes and employ particular CUDA implementations, making them challenging to expand to other applications. This study presents NerfAcc, an acronym for “NeRF Acceleration,” a toolkit focusing on the appealing volumetric display of radiance fields.

👉 Read our latest Newsletter: Microsoft’s FLAME for spreadsheets; Dreamix creates and edit video from image and text prompts......

They expand the approaches suggested in Instant-NGP to cover not just limited static scenes but also dynamic and unbounded scenes. They also provide a simple Python API developed in PyTorch that is ready for the plug-and-play acceleration of most NeRFs. Several examples demonstrate how to utilize the NerfAcc toolbox to speed up current NeRF procedures.

  • The vanilla NeRF model with an 8-layer MLP can be trained to superior quality (+0.5 PSNR) in 1 hour using the NerfAcc toolbox, rather than 1 2 days as described in the study. 
  • When compared to the official pure CUDA implementation, the Instant-NGP NeRF model can be trained to higher quality (+0.7 PSNR) in Python in less time (4.5 vs. 5 minutes).
  • The D-NeRF model for dynamic situations may be trained in 1 hour instead of 2 days as described in the study, and with higher quality (+2.0 PSNR).

NerfAcc is a PyTorch Nerf acceleration toolbox for both training and inference, and the Python code implementation is available on GitHub for free.

This Article is written as a research summary article by Marktechpost Staff based on the research paper 'NERFACC: A GENERAL NERF ACCELERATION TOOLBOX'. All Credit For This Research Goes To Researchers on This Project. Check out the paper and github link.

Please Don't Forget To Join Our ML Subreddit

Aneesh Tickoo is a consulting intern at MarktechPost. He is currently pursuing his undergraduate degree in Data Science and Artificial Intelligence from the Indian Institute of Technology(IIT), Bhilai. He spends most of his time working on projects aimed at harnessing the power of machine learning. His research interest is image processing and is passionate about building solutions around it. He loves to connect with people and collaborate on interesting projects.