Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
xmeng525 authored Jun 26, 2020
1 parent d091870 commit 67101d8
Showing 1 changed file with 45 additions and 0 deletions.
45 changes: 45 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -41,3 +41,48 @@ If you find this implementation useful in your research, please consider citing:
publisher = {The Eurographics Association},
}
```
### Prerequisite Installation
* Python3
* TensorFlow 1.13.1
* Pillow 6.1.0 (or newer)
* scikit-image 0.16.1 (or newer)
* OpenEXR 1.3.2 (or newer)

### Test with the Pre-trained Models
1. Clone this repo, and we'll call the directory `${MCDNBG_ROOT}`.
2. Download pre-trained models ["classroom"](https://www.dropbox.com/sh/8o7yijfc6rvba16/AADVi0wNoLrRbSgPBIvgcftsa?dl=0) and put the pretrained model to `${MCDNBG_ROOT}/classroom/model`.
3. Download the [1-spp dataset (19GB)](https://etsin.fairdata.fi/dataset/0ab24b68-4658-4259-9f1d-3150be898c63/data) or the [packed testdata for scene "classroom" (1.4GB)](https://www.dropbox.com/s/i8lqh6ezzeymwr9/bw_data_128x128_1scenes_60ips_50ppi_test.tfrecords?dl=0).
If you are using
4. Recompile the bilateral kernels by running
```
cd bilateral_kernels
./bilateral_kernel.sh
cd ..
```
5. Apply the denoiser by running
```
python network_test.py
```
- Input
- If you use the [1-spp dataset (19GB)](https://etsin.fairdata.fi/dataset/0ab24b68-4658-4259-9f1d-3150be898c63/data), please change the data-path in the argument list:
```
python network_test.py -d ${your-data-path}
```
- if you use the [packed testdata for scene "classroom" (1.4GB)](https://www.dropbox.com/s/i8lqh6ezzeymwr9/bw_data_128x128_1scenes_60ips_50ppi_test.tfrecords?dl=0), please put the tfrecords file in `${MCDNBG_ROOT}`.
- There are a few options in the arguments:
```
--export_exr ## export the exr file of the 1-spp radiance, denoised image, and ground truth
--export_all ## export all the denoised images from the lower-resolution layers
```
6. Evaluate the outputs by running:
```
python evaluation.py -d "classroom"
```
- The per-frame PSNR, SSIM, RMSE, SMAPE, and relative-MSE are saved in `${MCDNBG_ROOT}/classroom/result/evaluations`
- The SSIM errormaps and relative-MSE errormaps are saved in `${MCDNBG_ROOT}/classroom/result/evaluations`

### Retrain Your Own Model
Run "python network_train.py"

### Comparison with Neural Bilateral Grid Denoiser (MR-KP)
Please visit [our implementation of MR-KP](https://github.com/xmeng525/RealTimeDenoisingNeuralBilateralGrid) for more information.

0 comments on commit 67101d8

Please sign in to comment.