iNeRF

PyTorch implementation of iNeRF, an RGB-only method that inverts neural radiance fields (NeRFs) for 6DoF pose estimation.

iNeRF Inverting Neural Radiance Fields for Pose Estimation
Lin Yen-Chen1, Pete Florence2, Jonathan T. Barron2, Alberto Rodriguez1, Phillip Isola1, Tsung-Yi Lin2
1MIT, 2Google
IROS 2021

Overview

This preliminary codebase currently only shows how to apply iNeRF with pixelNeRF. However, iNeRF can work with the original NeRF as well.

Environment setup

To start, create the environment using conda:

cd pixel-nerf
conda env create -f environment.yml
conda activate pixelnerf
pip install mediapy
pip install jupyter

Please make sure you have up-to-date NVIDIA drivers supporting CUDA 10.2 at least.

Quick start

  1. Download all pixelNeRF's pretrained weight files from here. Extract this to ./pixel-nerf/checkpoints/, so that ./pixel-nerf/checkpoints/srn_car/pixel_nerf_latest exists.

  2. Launch the Jupyter notebook.

cd pixel-nerf
jupyter notebook
  1. Open pose_estimation.ipynb and run through it. You can preview the results here. In the following, we show the overlay of images rendered with our predicted poses and the target image.

BibTeX

@inproceedings{yen2020inerf,
  title={{iNeRF}: Inverting Neural Radiance Fields for Pose Estimation},
  author={Lin Yen-Chen and Pete Florence and Jonathan T. Barron and Alberto Rodriguez and Phillip Isola and Tsung-Yi Lin},
  booktitle={IEEE/RSJ International Conference on Intelligent Robots and Systems ({IROS})},
  year={2021}
}

Acknowledgements

This implementation is based on Alex Yu's pixel-nerf.