Home

Awesome

Versatile-NP

This repository contains the official implementation for the following paper:

Versatile Neural Processes for Learning Implicit Neural Representations, ICLR 2023

<img src="imgs/framework.png" width="600"/>

Reproducing 3D Experiments

The code for 3D experiments follows the logistics of Trans-INR.

Environment

Data

mkdir data and put different dataset folders in it.

Training

cd exp3D
CUDA_VISIBLE_DEVICES=[GPU] python run_trainer.py --cfg [CONFIG] --load-root [DATADIR]

Configs are in cfgs/. Four 3090Ti or four 32GB V100 GPUs are suggested for training.

Evaluation

For view synthesis, run in a single GPU with configs in cfgs/nvs_eval.

Checkpoint models

The pretrained checkpoint models can be found in Google Drive.

CarsLampsChairs
PSNR (dB)24.2124.1019.54

Since the code is reorganized with unified training settings, the performances of lamps and chairs are slightly better than our initial submission in openreview, and the performance of cars is slightly lower than our initial submission. By adjusting the annealing strategy of beta coefficient, the performance of cars could be further improved.

Reproducing 1D Experiments

The code for 1D toy examples can be found in the supplementary material of our Openreview submission.