An open API service indexing awesome lists of open source software.

https://github.com/lwwu2/diver-rt

Real-time application of DIVeR
https://github.com/lwwu2/diver-rt

Last synced: 29 days ago
JSON representation

Real-time application of DIVeR

Awesome Lists containing this project

README

        

# DIVeR: Real-time CUDA Application
This repo contains the code for the real-time application of DIVeR implemented in Python+CUDA.

## Setup

- python 3.8
- pytorch 1.9.0
- moderngl
- moderngl-window

### Pre-trained model

We have two versions of pre-trained models trained on nerf_synthetic dataset.

- Our [256 model](https://drive.google.com/file/d/1dEpMamHreZVtKV9BZA9uGFUJKQbBdJFq/view?usp=sharing) (256x256x256 voxel grid).
- Our [128 model](https://drive.google.com/file/d/11p0XdSNQrp_9HDbvQZaS7s9LZDi_v3QH/view?usp=sharing) (128x128x128 voxel grid).

The 128 model runs much faster with smaller storage cost, but the 256 model has better rendering quality.

## Usage

To launch the real-time application, run:

```shell
python run.py --weight_path=PATH_TO_WEIGHT_FILE \
--voxel_num=VOXEL_GRID_SIZE \
--device=GPU_DEVICE
```

## Resources

- [Project page](https://lwwu2.github.io/diver)
- [Paper](https://arxiv.org/abs/2111.10427)
- [Training code](https://github.com/lwwu2/diver)

## Citation
```
@misc{wu2021diver,
title={DIVeR: Real-time and Accurate Neural Radiance Fields with Deterministic Integration for Volume Rendering},
author={Liwen Wu and Jae Yong Lee and Anand Bhattad and Yuxiong Wang and David Forsyth},
year={2021},
eprint={2111.10427},
archivePrefix={arXiv},
primaryClass={cs.CV}
}
```