About PyTorch 1.1.0
Clone this repository into any place you want.
git clone https://github.com/lhjthp/HNCT.git
## Quickstart (Demo)
Run the script in ``src`` folder. Before you run the demo, please uncomment the appropriate line in ```demo.sh``` that you want to execute.
```bash
cd src # You are now in */HNCT/src
sh demo.sh
You can find the result images from experiment/test/
folder.
| Model | Scale | File name (.pt) | Parameters | **PSNR/SSIM | | --- | --- | --- | --- | Set5 Set14 BSD100 Urban100 Manga109 | | --- | 2 | x2.pt | 356K | 38.08/0.9608 33.65/0.9182 32.22/0.9001 32.22/0.9294 38.87/0.9774 | | --- | 3 | x3.pt | 363K | 34.47/0.9275 30.44/0.8439 29.15/0.8067 28.28/0.8557 33.81/0.9459 | | --- | 4 | x4.pt | 372K | 32.31/0.8957 28.71/0.7834 27.63/0.7381 26.20/0.7896 30.70/0.9112 | You can evaluate your models with widely-used benchmark datasets:
We used DIV2K dataset to train our model.
Unpack the tar file to any place you want. Then, change the dir_data
argument in src/option.py
to the place where DIV2K images are located.
We recommend you to pre-process the images before training. This step will decode all png files and save them as binaries. Use --ext sep_reset
argument on your first run. You can skip the decoding part and use saved binaries with --ext sep
argument.
You can train HNCT by yourself. All scripts are provided in the src/demo.sh
.
cd src # You are now in */HNCT/src
sh demo.sh
Update log