经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 大数据云AI » 人工智能基础 » 查看文章
SRCNN
来源:cnblogs  作者:ceo1207  时间:2018/11/9 13:33:10  对本文有异议

超分的开篇之作,2014 ECCV 港中文 Chao Dong

三层网络,文中还对各层网络的意义做出了解释

使用caffe训练模型,matlab做inference,代码见 http://mmlab.ie.cuhk.edu.hk/projects/SRCNN.html 

700*700像素的图重建出来需要30s+,效果还是比较朦胧

 

可以用自己的图片做一下直观了解

  1. path = 'E:\Download\超分辨率\test\test';
  2.  
  3. list = dir(path);
  4. for i1 = 1: size(list,1)
  5. name = list(i1).name;
  6. if name == '.'
  7. continue;
  8. end
  9. fullName = fullfile(path, name);
  10. demo_SR(fullName); % 需要把作者的demo源文件改为function
  11. end

  

评估方法

计算PSNR、SSIM,计算的方式可以参见VDSR的matlab代码

常用评估数据集 set5,set14,B100,urban100,之后的新paper,还用了manga109、部分DIV2K等

 

 

用二维坐标表示性能也相当直观

 

 

缺点:

1、Works for only a single scale
2、卷积感受野太小
3、学习速率 1e-5 太慢,需要加快训练速度。而且inference的速度也比较慢

 

如何训练model

 

关于训练数据的说明:

For a fair comparison with traditional example-based methods, we use the same training set, test sets, and protocols as in [20]. Speci cally, the training set consists of 91 images. The Set5 [2] (5 images) is used to evaluate the performance of upscaling factors 2, 3, and 4, and Set14 [28] (14 images) is used to evaluate the upscaling factor 3. In addition to the 91-image training set, we also investigate a larger training set in Section 5.2. 

 

Anchored Neighborhood Regression for Fast Example-Based Super-Resolution

http://www.vision.ee.ethz.ch/~timofter/ICCV2013_ID1774_SUPPLEMENTARY/index.html


而cvpr2018的SR文章使用的训练集,基本就是比较大的数据集了,其实选什么都可以的,只要不包含test data即可。

DIV2K consists of 800 training images, 100 validation images, and 100 test images. We train all of our models with 800 training images and use 5 validation images in the training process. For testing, we use five standard benchmark datasets: Set5 [1], Set14 [33], B100 [18], Urban100 [8], and Manga109 [19]. The SR results are evaluated with PSNR and SSIM [32] on Y channel (i.e., luminance) of transformed YCbCr space.

 整理了SRCNN的训练数据 91,测试数据集 set5,set14,B100,urban100

 链接: https://pan.baidu.com/s/1f5CrntYV2RgsAVoDx3hvUg 提取码: jv2f 复制这段内容后打开百度网盘手机App,操作更方便哦

 

keras版本的模型训练代码,可以参考一下 

https://github.com/DeNA/SRCNNKit/tree/master/script

 

 

VDSR 

CVPR 2016 首尔大学

https://cv.snu.ac.kr/research/VDSR/

 

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号