Skip to content

Code of our winning entry to NTIRE 2018 super-resolution challenge. https://www.vision.ee.ethz.ch/ntire18/

Notifications You must be signed in to change notification settings

WeiHan3/wdsr_ntire2018

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 

Repository files navigation

Wide Activation and Weight Normalization for Accurate Image Super-Resolution

By Jiahui Yu1, Yuchen Fan1, Jianchao Yang2, Ning Xu2, Zhaowen Wang3, Xinchao Wang4, Thomas S. Huang1

[1] University of Illinois at Urbana-Champaign, [2] Snap Inc., [3] Adobe Research, [4] Stevens Institute of Technology

Code and more details will be available soon. Please stay tuned.

Overall Performance

Network Parameters DIV2K (val) PSNR
EDSR Baseline 1,372,318 34.61
WDSR Baseline 1,190,100 34.77

We measured PSNR using DIV2K 0801 ~ 0900 (trained on 0000 ~ 0800) on RGB channels without self-ensemble which is identical to EDSR baseline model settings. Both baseline models have 16 residual blocks.

More results:

Number of Residual Blocks13
SR NetworkEDSRWDSREDSRWDSR
Parameters2.6M0.8M4.1M2.3M
DIV2K (val) PSNR33.21033.32334.04334.163
Number of Residual Blocks58
SR NetworkEDSRWDSREDSRWDSR
Parameters5.6M3.7M7.8M6.0M
DIV2K (val) PSNR34.28434.38834.45734.541

Comparisons of EDSR and our proposed WDSR for image bicubic x2 super-resolution on DIV2K dataset.

WDSR Network Architecture

Left: vanilla residual block in EDSR. Middle: wide activation. Right: wider activation with factorized convolution. The proposed wide activation has similar merits with MobileNet V2.

Weight Normalization vs. Batch Normalization and No Normalization

Training loss and validation PSNR with weight normalization, batch normalization or no normalization. Training with weight normalization has better performance.

About

Code of our winning entry to NTIRE 2018 super-resolution challenge. https://www.vision.ee.ethz.ch/ntire18/

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published