nerf-lightning

Unofficial lightning implementation version of nerf.

DataSet

Sample Dataset Download

How to use

Using the Conda environment

• conda create -n nerf_lightning python=3.8
• conda activate nerf_lightning
• pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

Configuring packages

• pip install -r requirements.txt

Train

Adjust the parameters in the opt.py file.Enter the pre-configured nerf_lightning virtual environment.Utilize the following command for training.

python train.py

Test

Adjust the parameters in the eval.py file to load the corresponding model. You can download our pretrained models psnr=29.52.ckpt. Use the following command for testing.

python eval.py