HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Accelerating the Super-Resolution Convolutional Neural Network

Chao Dong; Chen Change Loy; Xiaoou Tang

Accelerating the Super-Resolution Convolutional Neural Network

Abstract

As a successful deep model applied in image super-resolution (SR), the Super-Resolution Convolutional Neural Network (SRCNN) has demonstrated superior performance to the previous hand-crafted models either in speed and restoration quality. However, the high computational cost still hinders it from practical usage that demands real-time performance (24 fps). In this paper, we aim at accelerating the current SRCNN, and propose a compact hourglass-shape CNN structure for faster and better SR. We re-design the SRCNN structure mainly in three aspects. First, we introduce a deconvolution layer at the end of the network, then the mapping is learned directly from the original low-resolution image (without interpolation) to the high-resolution one. Second, we reformulate the mapping layer by shrinking the input feature dimension before mapping and expanding back afterwards. Third, we adopt smaller filter sizes but more mapping layers. The proposed model achieves a speed up of more than 40 times with even superior restoration quality. Further, we present the parameter settings that can achieve real-time performance on a generic CPU while still maintaining good performance. A corresponding transfer strategy is also proposed for fast training and testing across different upscaling factors.

Code Repositories

MohammedAlkhrashi/TMA
Mentioned in GitHub
Nhat-Thanh/FSRCNN-Pytorch
pytorch
Mentioned in GitHub
yjn870/FSRCNN-pytorch
pytorch
Mentioned in GitHub
poikilos/pyrotocanvas
tf
Mentioned in GitHub
MohammedAlkhrashi/4D-Dream
Mentioned in GitHub
Araxeus/PNG-Upscale
tf
Mentioned in GitHub
Lornatang/FSRCNN-PyTorch
pytorch
Mentioned in GitHub
xanderex-sid/FSRCNN
pytorch
Mentioned in GitHub
GatorSense/SRrootimaging
pytorch
Mentioned in GitHub
Nhat-Thanh/FSRCNN-TF
tf
Mentioned in GitHub
NicoCeresa/FSRCNN-2016
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-super-resolution-on-bsd100-2x-upscalingFSRCNN [[Dong et al.2016]]
PSNR: 31.53
image-super-resolution-on-ffhq-1024-x-1024-4xFSRCNN
FID: 23.97
MS-SSIM: 0.951
PSNR: 24.71
SSIM: 0.804
image-super-resolution-on-ffhq-256-x-256-4xFSRCNN
FID: 139.78
MS-SSIM: 0.930
PSNR: 22.45
SSIM: 0.709
image-super-resolution-on-manga109-4xFSRCNN
PSNR: 27.90
SSIM: 0.8610

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Accelerating the Super-Resolution Convolutional Neural Network | Papers | HyperAI