项目作者: opteroncx

项目描述 :
SESR: Single Image Super Resolution with Recursive Squeeze and Excitation Networks
高级语言: Python
项目地址: git://github.com/opteroncx/SESR.git
创建时间: 2018-01-17T12:16:49Z
项目社区:https://github.com/opteroncx/SESR

开源协议:

下载


SESR

SESR: Single Image Super Resolution with Recursive Squeeze and Excitation Networks (To appear in ICPR 2018)

https://arxiv.org/abs/1801.10319

Quality for scale x4


  1. Trained on div2k, r=4

DataSet/Method PSNR/SSIM
Set5 32.05/ 0.897
Set14 28.54/ 0.789
BSD100 27.51/ 0.743
Urban100 25.83/ 0.785

Compare with other methods

Requirement

  1. Python 2.7
  2. Pytorch 0.2.0
  3. opencv-python
  4. numpy

Train

  1. python train.py --cuda

Evaluate

  1. python test.py --cuda

Do Super resolution on your own images

  1. python test.py --cuda --mode sr --testdir path_to_your_image

Reference

https://github.com/twtygqyy/pytorch-LapSRN
https://github.com/jiny2001/dcscn-super-resolution
https://github.com/jmiller656/EDSR-Tensorflow

https://github.com/grevutiu-gabriel/python-ssim/blob/master/python-ssim.py