Skip to content

Commit

Permalink
Add installation instructions
Browse files Browse the repository at this point in the history
  • Loading branch information
adityac8 authored Mar 28, 2020
1 parent 740c09b commit 5a42b86
Showing 1 changed file with 16 additions and 4 deletions.
20 changes: 16 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -40,34 +40,46 @@ Given a synthetic RAW noisy image as input, the RAW2RGB network branch maps it t
<br/>
</p>

## Installation
The model is built in PyTorch 1.1.0 and tested on Ubuntu 16.04 environment (Python3.7, CUDA9.0, cuDNN7.5).

For installing, follow these intructions
```
sudo apt-get install cmake build-essential libjpeg-dev libpng-dev
conda create -n pytorch1 python=3.7
conda activate pytorch1
conda install pytorch=1.1 torchvision=0.3 cudatoolkit=9.0 -c pytorch
pip install matplotlib scikit-image yacs lycon natsort h5py tqdm
```

## Evaluation
#### Denoising RAW images of DND
- Download the [model](https://drive.google.com/file/d/1yjI3JtfC1IluGB0LRlnpPTAY_UyY9mG8/view?usp=sharing) and place it in ./pretrained_models/denoising/
- Download RAW [images](https://drive.google.com/drive/folders/15Bay1UJURlbP7kpS10_fJ96MxQ6B03Xv?usp=sharing) of DND and place them in ./datasets/dnd/dnd_raw/
- Run
```
$ python test_dnd_raw.py --save_images
python test_dnd_raw.py --save_images
```
#### Denoising RAW images of SIDD
- Download the [model](https://drive.google.com/file/d/1m2A4goZENg_kKV1-FJoeg1rBqLj6WRIm/view?usp=sharing) and place it in ./pretrained_models/denoising/
- Download RAW [images](https://drive.google.com/drive/folders/1invur2uE-QXHh-btHTZQetFcgjUAfbwc?usp=sharing) of SIDD and place them in ./datasets/sidd/sidd_raw/
- Run
```
$ python test_sidd_raw.py --save_images
python test_sidd_raw.py --save_images
```
#### Denoising sRGB images of DND
- Download the [model](https://drive.google.com/file/d/1740sYH7bG-c-jL5wc3e1_NOpxwGTXS9c/view?usp=sharing) and place it in ./pretrained_models/denoising/
- Download sRGB [images](https://drive.google.com/drive/folders/101AfVtkfizl20-XQ3leNt2cE_rI5ypeu?usp=sharing) of DND and place them in ./datasets/dnd/dnd_rgb/noisy/
- Run
```
$ python test_dnd_rgb.py --save_images
python test_dnd_rgb.py --save_images
```
#### Denoising sRGB images of SIDD
- Download the [model](https://drive.google.com/file/d/1sraG9JKmp0ieLjntRL7Jj2FXBrPr-YVp/view?usp=sharing) and place it in ./pretrained_models/denoising/
- Download sRGB [images](https://drive.google.com/drive/folders/1JGoCXqHESBIocpDabk74mnee6RA3oCoP?usp=sharing) of SIDD and place them in ./datasets/sidd/sidd_rgb/
- Run
```
$ python test_sidd_rgb.py --save_images
python test_sidd_rgb.py --save_images
```
## Results on Real Image Datasets
Experiments are performed for denoising images in RAW and sRGB spaces.
Expand Down

0 comments on commit 5a42b86

Please sign in to comment.