Skip to content

vikashChouhan/SR-Caffe

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SR-Caffe

Image Super-Resolution use caffe.

This project implements two articles,"Image Super-Resolution Using Deep Convolutional Networks"(ECCV 2014) and "Accurate Image Super-Resolution Using Very Deep Convolutional Networks" (CVPR 2016 Oral Paper).

工程组织结构

├── dataset       (数据集)
│   ├── hdf5
│   │   ├── SRCNN-5
│   │   ├── SRCNN-Pool
│   │   └── VDSR-20
│   └── images
│       ├── 291
│       ├── 91
│       ├── 91-aug
│       └── test
├── external
│   └── caffe
├── pdf
├── test	(测试评估)
│   ├── images
│   │   ├── Set14
│   │   └── Set5
│   ├── matconvnet
│   ├── results
│   │   ├── SRCNN-5
│   │   ├── SRCNN-Pool
│   │   └── VDSR-20
│   │       ├── official
│   │       └── ours
│   └── utils
└── train	(训练)
    ├── SRCNN-5
    │   ├── caffemodel
    │   ├── convert
    │   ├── demo
    │   │   └── utils
    │   └── logs
    ├── SRCNN-Pool
    │   ├── caffemodel
    │   ├── convert
    │   ├── logs
    │   └── prototxt
    │       └── data_pool
    │           └── logs
    └── VDSR-20
        ├── caffemodel
        ├── convert
        └── logs

运行

1 数据准备

下载数据集 zip(train) zip(test)

git clone https://github.com/mindcont/SR-Caffe.git
cd SR-Caffe && mkdir dataset
wget http://cv.snu.ac.kr/research/VDSR/train_data.zip
wget http://cv.snu.ac.kr/research/VDSR/test_data.zip
unzip train_data.zip images/

# use matlab 
run train/data_aug.m

2 生成HDF5 格式,供caffe 训练

generate_train.m
generate_test.m

3 训练

run train_VDSR-20.sh or train_SRCNN-5.sh

4 测试评估

run demo_SR.m

博客

类似项目

About

Super-Resolution use Caffe

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • MATLAB 32.9%
  • Cuda 24.0%
  • C++ 18.0%
  • TeX 10.6%
  • Python 8.3%
  • Shell 2.6%
  • Other 3.6%