1
0
Fork 0
mirror of synced 2024-06-23 08:41:01 +12:00
waifu2x/appendix/benchmark.md

84 lines
4.3 KiB
Markdown
Raw Normal View History

2016-10-08 23:36:26 +13:00
# Benchmarks
2016-03-17 21:40:41 +13:00
2016-10-08 23:36:26 +13:00
## Photo
2016-10-20 14:17:19 +13:00
Note: waifu2x's photo models was trained on the blending dataset of [kou's photo collection](http://photosku.com/photo/category/%E6%92%AE%E5%BD%B1%E8%80%85/kou/) and [ukbench](http://vis.uky.edu/~stewe/ukbench/).
2016-10-09 00:01:15 +13:00
2016-10-08 23:54:02 +13:00
Note: PSNR in this benchmark uses a [MATLAB's rgb2ycbcr](https://jp.mathworks.com/help/images/ref/rgb2ycbcr.html?lang=en) compatible function (dynamic range [16 235], not [0 255]) for converting grayscale image. I think it's not correct PSNR. But many paper used this metric.
2016-03-17 21:40:41 +13:00
2016-10-08 23:36:26 +13:00
command:
`th tools/benchmark.lua -dir <dataset_dir> -model1_dir <model_dir> -method scale -filter Catrom -color y -range_bug 1 -tta <0|1> -force_cudnn 1`
2016-05-06 05:40:15 +12:00
2016-10-08 23:36:26 +13:00
### Datasets
2016-03-17 21:40:41 +13:00
2016-10-09 00:01:15 +13:00
BSD100: https://www2.eecs.berkeley.edu/Research/Projects/CS/vision/grouping/segbench/ (100 test images in BSDS300)
2016-10-08 23:36:26 +13:00
Urban100: https://github.com/jbhuang0604/SelfExSR
2016-03-17 21:40:41 +13:00
2016-10-08 23:36:26 +13:00
### 2x - PSNR
2016-03-17 21:40:41 +13:00
| Dataset/Model | Bicubic | vgg\_7/photo | upconv\_7/photo | upconv\_7l/photo | resnet_14l/photo |
|---------------|---------------|---------------|------------------|------------------|--------------------|
| BSD100 | 29.558 | 31.427 | 31.640 | 31.749 | 31.847 |
| Urban100 | 26.852 | 30.057 | 30.477 | 30.759 | 31.016 |
2016-03-17 21:40:41 +13:00
2018-11-15 00:30:04 +13:00
### 2x - benchmark time (sec)
2018-11-15 00:21:44 +13:00
| Dataset/Model | vgg\_7/photo | upconv\_7/photo | upconv\_7l/photo | resnet_14l/photo |
|---------------|---------------|------------------|------------------|--------------------|
| BSD100 | 4.057 | 2.509 | 4.947 | 6.86 |
| Urban100 | 16.349 | 7.083 | 14.178 | 27.87 |
2016-10-08 23:36:26 +13:00
### 2x with TTA - PSNR
2016-03-17 21:40:41 +13:00
2016-10-08 23:54:02 +13:00
Note: TTA is an ensemble technique that is supported by waifu2x. TTA method is 8x slower than non TTA method but it improves PSNR (~+0.1 on photo, ~+0.4 on art).
2016-03-17 21:40:41 +13:00
| Dataset/Model | Bicubic | vgg\_7/photo | upconv\_7/photo | upconv\_7l/photo | resnet_14l/photo |
|---------------|---------------|---------------|------------------|------------------|--------------------|
| BSD100 | 29.558 | 31.474 | 31.705 | 31.812 | 31.915 |
| Urban100 | 26.852 | 30.140 | 30.599 | 30.868 | 31.162 |
2016-03-17 21:40:41 +13:00
2018-11-15 00:30:04 +13:00
### 2x with TTA - benchmark time (sec)
2016-10-08 23:36:26 +13:00
| Dataset/Model | vgg\_7/photo | upconv\_7/photo | upconv\_7l/photo | resnet_14l/photo |
|---------------|---------------|------------------|------------------|--------------------|
| BSD100 | 36.611 | 20.219 | 42.486 | 60.38 |
| Urban100 | 132.416 | 65.125 | 129.916 | 255.20 |
2016-10-08 23:36:26 +13:00
## Art
2018-11-14 23:03:10 +13:00
command: See `appendix/benchmark.sh`
2016-10-08 23:36:26 +13:00
### Dataset
2018-11-14 23:03:10 +13:00
art_test: This dataset contains 84 various fan-arts. Sorry, This dataset is private.
2016-10-08 23:36:26 +13:00
### 2x - PSNR
2018-11-14 23:03:10 +13:00
| Filter/Model | Bicubic | vgg\_7/art | upconv\_7/art | cunet/art |
|----------------|---------------|-------------|----------------|----------------|
| Lanczos | 31.022 | 37.495 | 38.330 | 39.886 |
| Sinc | 30.947 | 37.722 | 38.538 | 40.312 |
| Catrom(Bicubic)| 30.663 | 37.278 | 37.189 | 40.184 |
| Box | 30.891 | 37.709 | 38.410 | 39.672 |
2016-10-08 23:36:26 +13:00
2018-11-15 00:30:04 +13:00
### 2x - benchmark time (sec)
2018-11-15 00:21:44 +13:00
| Dataset/Model | vgg\_7/art | upconv\_7/art | cunet/art |
|---------------|-------------|----------------|----------------|
| art_test | 24.153 | 10.794 | 24.222 |
2016-10-08 23:36:26 +13:00
### 2x with TTA - PSNR
2018-11-14 23:03:10 +13:00
| Filter/Model | Bicubic | vgg\_7/art | upconv\_7/art | cunet/art |
|----------------|---------------|-------------|----------------|----------------|
| Lanczos | 31.022 | 37.777 | 38.677 | 40.289 |
| Sinc | 30.947 | 38.005 | 38.883 | 40.707 |
| Catrom(Bicubic)| 30.663 | 37.498 | 37.417 | 40.592 |
| Box | 30.891 | 38.032 | 38.768 | 40.032 |
2018-11-15 00:30:04 +13:00
### 2x with TTA - benchmark time (sec)
2016-10-08 23:36:26 +13:00
2018-11-15 00:21:44 +13:00
| Dataset/Model | vgg\_7/art | upconv\_7/art | cunet/art |
2016-10-08 23:36:26 +13:00
|---------------|-------------|----------------|----------------|
2018-11-14 23:03:10 +13:00
| art_test | 207.217 | 99.151 | 211.520 |