Back to Waifu2x

Benchmarks

appendix/benchmark.md

0.13.24.3 KB
Original Source

Benchmarks

Photo

Note: waifu2x's photo models was trained on the blending dataset of kou's photo collection and ukbench.

Note: PSNR in this benchmark uses a MATLAB's rgb2ycbcr 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.

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

Datasets

BSD100: https://www2.eecs.berkeley.edu/Research/Projects/CS/vision/grouping/segbench/ (100 test images in BSDS300) Urban100: https://github.com/jbhuang0604/SelfExSR

2x - PSNR

Dataset/ModelBicubicvgg_7/photoupconv_7/photoupconv_7l/photoresnet_14l/photo
BSD10029.55831.42731.64031.74931.847
Urban10026.85230.05730.47730.75931.016

2x - benchmark time (sec)

Dataset/Modelvgg_7/photoupconv_7/photoupconv_7l/photoresnet_14l/photo
BSD1004.0572.5094.9476.86
Urban10016.3497.08314.17827.87

2x with TTA - PSNR

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).

Dataset/ModelBicubicvgg_7/photoupconv_7/photoupconv_7l/photoresnet_14l/photo
BSD10029.55831.47431.70531.81231.915
Urban10026.85230.14030.59930.86831.162

2x with TTA - benchmark time (sec)

Dataset/Modelvgg_7/photoupconv_7/photoupconv_7l/photoresnet_14l/photo
BSD10036.61120.21942.48660.38
Urban100132.41665.125129.916255.20

Art

command: See appendix/benchmark.sh

Dataset

art_test: This dataset contains 84 various fan-arts. Sorry, This dataset is private.

2x - PSNR

Filter/ModelBicubicvgg_7/artupconv_7/artcunet/art
Lanczos31.02237.49538.33039.886
Sinc30.94737.72238.53840.312
Catrom(Bicubic)30.66337.27837.18940.184
Box30.89137.70938.41039.672

2x - benchmark time (sec)

Dataset/Modelvgg_7/artupconv_7/artcunet/art
art_test24.15310.79424.222

2x with TTA - PSNR

Filter/ModelBicubicvgg_7/artupconv_7/artcunet/art
Lanczos31.02237.77738.67740.289
Sinc30.94738.00538.88340.707
Catrom(Bicubic)30.66337.49837.41740.592
Box30.89138.03238.76840.032

2x with TTA - benchmark time (sec)

Dataset/Modelvgg_7/artupconv_7/artcunet/art
art_test207.21799.151211.520