-
Notifications
You must be signed in to change notification settings - Fork 1
/
README.txt
50 lines (36 loc) · 1.92 KB
/
README.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
This repository has been built over this pytorch implementation of NeRF: https://github.com/yenchenlin/nerf-pytorch
CONTRIBUTION:
Tanmay - Implemented inerf_sampling.py
Videsh - Adapted and implemented so3_helpers.py
Both - pose_estimation function in run_inerf.py, results.py
DEPENDENCIES:
- pytorch
- torchvision
- numpy
- imageio
- imageio-ffmpeg
- matplotlib
- configargparse
- opencv-python
- pyquaternion
INSTALLATION:
Run the following commands.
cd inerf
pip install -r requirements.txt
IMPLEMENTATION DETAILS:
Currently, this implementation supports 2 datasets i.e. nerf-synthetic and nerf-llff. Performing pose estimation
for any query image of a scene also requires its pretrained NeRF representation. This repository may not contain
all the details to train NeRF. We have evaluated results for scenes lego (synthetic) and fern (llff).
run_inerf.py script performs pose estimation for a random query image from the validation/test dataset.
The pose is randomly initialized in some vicinity of the ground truth camera pose.
Please check out the report for more details.
It usually takes less than 5 minutes to optimize pose for 1 query image on a single GPU.
Data:
- lego (synthetic): pretrained NeRF, images and corresponding camera poses
- fern (llff): pretrained NeRF, images and corresponding camera poses
How to optimize camera pose:
python run_inerf.py --config inerf_configs/{SCENE}.txt (replace {SCENE} with lego | fern)
Also, for more data, pretrained models and other implementation details on NeRF, checkout the file nerf_README.md
CODE CITATIONS:
- Pytorch NeRF implementation: https://github.com/yenchenlin/nerf-pytorch
- Modern Robotics for exponential parameterization: https://github.com/NxRLab/ModernRobotics/blob/master/packages/Python/modern_robotics/core.py