Araxeus / PNG-Upscale

AI Super - Resolution
MIT License
214 stars 24 forks source link
java javacv machine-learning png resampling resize resizing-images software super-resolution upscaling

:fire: PNG Upscale - AI Super Resolution :fire:

Small tool using pretrained models to upscale images

Download is available from the Releases Page or Google Drive or MediaFire

Models

all of the model download links below are already included in the MediaFire folder.

There are four trained models integrated into the program :

EDSR

[Best Quality]+[Slowest]

Trained models can be downloaded from here.

ESPCN

[Fast]

Trained models can be downloaded from here.

FSRCNN

[Fast]

Trained models can be downloaded from here.

LapSRN

[Has x8]

Trained models can be downloaded from here.


Benchmarks

Comparing different algorithms. Scale x4 on monarch.png

Inference time in seconds (CPU) PSNR SSIM
ESPCN 0.01159 26.5471 0.88116
EDSR 3.26758 29.2404 0.92112
FSRCNN 0.01298 26.5646 0.88064
LapSRN 0.28257 26.7330 0.88622
Bicubic 0.00031 26.0635 0.87537
Nearest neighbor 0.00014 23.5628 0.81741
Lanczos 0.00101 25.9115 0.87057

As a Demo this image was resized from 256x256 to 85x85, and then upscaled using this program

Original

x2 Demo (85x85 -> 170x170)

Original Bicubic Interpolation EDSR
Original Bicubic EDSR
ESPCN FSRCNN LapSRN
ESPCN FSRCNN LapSRN

Bicubic Interpolation is the standart resizing technique used by most editing tools like photoship etc..

x4 Demo (85x85 -> 340x340)

Original Bicubic Interpolation EDSR
Original Bicubic EDSR
ESPCN FSRCNN LapSRN
ESPCN FSRCNN LapSRN

References

[1] Bee Lim, Sanghyun Son, Heewon Kim, Seungjun Nah, and Kyoung Mu Lee, "Enhanced Deep Residual Networks for Single Image Super-Resolution", 2nd NTIRE: New Trends in Image Restoration and Enhancement workshop and challenge on image super-resolution in conjunction with CVPR 2017. [PDF] [arXiv] [Slide]

[2] Shi, W., Caballero, J., Huszár, F., Totz, J., Aitken, A., Bishop, R., Rueckert, D. and Wang, Z., "Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network", Proceedings of the IEEE conference on computer vision and pattern recognition CVPR 2016. [PDF] [arXiv]

[3] Chao Dong, Chen Change Loy, Xiaoou Tang. "Accelerating the Super-Resolution Convolutional Neural Network", in Proceedings of European Conference on Computer Vision ECCV 2016. [PDF] [arXiv] [Project Page]

[4] Lai, W. S., Huang, J. B., Ahuja, N., and Yang, M. H., "Deep laplacian pyramid networks for fast and accurate super-resolution", In Proceedings of the IEEE conference on computer vision and pattern recognition CVPR 2017. [PDF] [arXiv] [Project Page]