DReg-NeRF: Deep Registration for Neural Radiance Fields
This addresses the need for automated registration of NeRFs in computer vision, offering a novel method that eliminates the requirement for human-annotated keypoints, though it is incremental in improving upon existing techniques.
The paper tackles the problem of registering multiple Neural Radiance Fields (NeRFs) for object-centric scenes without human intervention, achieving a mean RPE of 9.67° and a mean RTE of 0.038, which beats state-of-the-art point cloud registration methods by a large margin.
Although Neural Radiance Fields (NeRF) is popular in the computer vision community recently, registering multiple NeRFs has yet to gain much attention. Unlike the existing work, NeRF2NeRF, which is based on traditional optimization methods and needs human annotated keypoints, we propose DReg-NeRF to solve the NeRF registration problem on object-centric scenes without human intervention. After training NeRF models, our DReg-NeRF first extracts features from the occupancy grid in NeRF. Subsequently, our DReg-NeRF utilizes a transformer architecture with self-attention and cross-attention layers to learn the relations between pairwise NeRF blocks. In contrast to state-of-the-art (SOTA) point cloud registration methods, the decoupled correspondences are supervised by surface fields without any ground truth overlapping labels. We construct a novel view synthesis dataset with 1,700+ 3D objects obtained from Objaverse to train our network. When evaluated on the test set, our proposed method beats the SOTA point cloud registration methods by a large margin, with a mean $\text{RPE}=9.67^{\circ}$ and a mean $\text{RTE}=0.038$. Our code is available at https://github.com/AIBluefisher/DReg-NeRF.