In this technical report, we improve the DVGO framework (called DVGOv2), which is based on Pytorch and uses the simplest dense grid representation. First, we re-implement part of the Pytorch operations with cuda, achieving 2-3x speedup. The cuda extension is automatically compiled just in time. Second, we extend DVGO to support Forward-facing and Unbounded Inward-facing capturing. Third, we improve the space time complexity of the distortion loss proposed by mip-NeRF 360 from O(N^2) to O(N). The distortion loss improves our quality and training speed. Our efficient implementation could allow more future works to benefit from the loss.
Cite
@article{arxiv.2206.05085,
title = {Improved Direct Voxel Grid Optimization for Radiance Fields Reconstruction},
author = {Cheng Sun and Min Sun and Hwann-Tzong Chen},
journal= {arXiv preprint arXiv:2206.05085},
year = {2022}
}