南京大学学报(自然科学版) ›› 2019, Vol. 55 ›› Issue (4): 581–591.doi: 10.13232/j.cnki.jnju.2019.04.008

所属专题: 测试专题

• • 上一篇    下一篇

联合深度置信网络与邻域回归的超分辨率算法

黄炜钦1,2(),高凤强1,2,陈俊仁1,2,李婵1   

  1. 1. 厦门大学嘉庚学院信息科学与技术学院,漳州,363105
    2. 嘉庚学院微柏工业机器人创新实验室,漳州,363105
  • 收稿日期:2019-05-28 出版日期:2019-07-30 发布日期:2019-07-23
  • 通讯作者: 黄炜钦 E-mail:wqhuang@xujc.com
  • 基金资助:
    厦门大学嘉庚学院校级科研孵化项目(2018L03);福建省高校杰出青年科研人才培育计划(闽教科[2018]47号)

Image super⁃resolution algorithm joint deep belief network and neighborhood regression

Weiqin Huang1,2(),Fengqiang Gao1,2,Junren Chen1,2,Chan Li1   

  1. 1. School of Information Science & Technology, Xiamen University Tan Kah Kee College, Zhangzhou, 363105, China
    2. Tan Kah Kee College?VIBOT Industrial Robot Innovation Lab, Zhangzhou, 363105, China
  • Received:2019-05-28 Online:2019-07-30 Published:2019-07-23
  • Contact: Weiqin Huang E-mail:wqhuang@xujc.com

摘要:

为了提高重建的质量和速度,提出一种联合深度置信网络与邻域回归的超分辨率算法.一方面,结合字典学习与神经网络表示的联系对传统的深度置信网络进行调整,采用该网络模型实现字典学习,充分利用该模型突出的学习能力,使字典具有更好的特征表达能力,从而提高图像的重建质量.另一方面,在基于字典学习的超分辨率框架中融入邻域回归思想.首先,利用最近邻域算法确定字典原子的最近邻域映射关系;然后以此为基础,结合邻域回归方法,离线计算高、低分辨率投影矩阵;最后在重建过程中将该投影矩阵应用于图像重建.该方法避免了字典学习中的系数求解过程,降低了计算的复杂度,提高了重建的速度.实验表明,算法具有更高的峰值信噪比和结构相似度,同时极大地提高了图像的重建速度.

关键词: 超分辨率, 字典学习, 深度置信网络, 邻域回归

Abstract:

To improve the quality and speed of reconstruction,an image super?resolution algorithm joint deep belief network and neighborhood regression is proposed. For one thing,combine the relationship between dictionary learning and the neural network representation to adjust the traditional deep belief network and apply it to learn the dictionary,that make full use of the outstanding learning ability of the deep belief network to make the dictionary have the better ability to express feature. As a result,the quality of image reconstruction can be improved. For another,we integrate the idea of neighborhood regression in the framework of super?resolution based on dictionary learning. First of all,we determine the nearest neighborhood mapping relationship of the dictionary atom with the nearest neighborhood algorithm,and then based on it,the high resolution and low resolution projection matrix can be calculated offline using the neighborhood regression. Finally,the projection matrix can be applied to reconstruct images during the reconstruction process. This method avoids the process of solving coefficients in dictionary learning and reduces the computational complexity. Consequently, It has the ability to improve the speed of reconstruction.The experiments show that the algorithm has a higher Peak Signal?to?Noise Ratio and Structural Similarity,and it also increases the speed of image reconstruction greatly.

Key words: super?resolution, dictionary learning, deep belief network, neighborhood regression

中图分类号: 

  • TP391

图1

RBM模型"

图2

含有三层RBM的DBN模型"

图3

字典学习与神经网络表示的联系"

图4

样本特征图像的处理"

图5

基于DBN的字典学习模型"

图6

算法流程图"

图7

最近邻域数与重建质量关系"

表1

不同SR算法的重建图像对应的PSNR和SSIM值"

测试图像PSNR/SSIMBicubicL1SRSISRANRSRCNNDBNNR
BabyPSNR33.9134.2935.0835.1335.0135.26
SSIM0.92440.92260.94020.94150.94010.9425
BridPSNR32.5834.1134.5734.6034.9134.95
SSIM0.94220.95300.96150.96230.96230.9643
HeadPSNR32.8833.1733.5633.6333.5533.70
SSIM0.83860.83820.85720.86010.85810.8614
BarbaraPSNR26.2526.3926.7626.6926.6626.80
SSIM0.78130.78900.80810.80780.80760.8128
FacePSNR32.8233.1233.5333.6233.5833.68
SSIM0.83740.83740.85670.85990.85830.8612
ForemanPSNR31.1832.0533.1933.2333.3533.75
SSIM0.92310.92470.94260.94310.94420.9465
Ppt3PSNR23.7124.9825.2325.0325.3225.56
SSIM0.88740.90250.92040.91230.92740.9293
ZebraPSNR26.6327.9528.4928.4328.8729.07
SSIM0.83610.86480.87750.87940.88260.8874
平均值PSNR30.0030.7631.3031.3031.4031.60
SSIM0.87130.87900.89550.89580.89760.9007

图8

主观测试图像"

图9

Head在不同SR算法下的重建图像细节比较"

表2

Set5和Set14在不同SR算法下的平均重建时间(s)"

测试图像库L1SRSISRSRCNNDBNNR
Set523.511.023.440.30
Set1452.722.1110.000.62

图10

Zebra在不同SR算法下的重建图像细节比较"

图11

字典训练样本图像"

图12

Medical 1原图与重建图像比较"

图13

Medical 2原图与重建图像比较"

表3

原始医学图像与重建图像的无参考指标平均值的比较"

无参考指标平均梯度信息熵Brenner函数能量函数
原始图2.94436.670142617383379712
DBNNR3.65276.708860875685284806
1 潘宗序,禹晶,肖创柏等. 基于自适应多字典学习的单幅图像超分辨率算法. 电子学报,2015,43(2):209-216.
Pan Z X,Yu J,Xiao C B,et al.Single image super resolution based on adaptive multi?dictionary learning. Acta Electronica Sinica,2015,43(2):209-216.
2 苏衡,周杰,张志浩. 超分辨率图像重建方法综述. 自动化学报,2013,39(8):1202-1213.
Su H,Zhou J,Zhang Z H.Survey of super?resolution image reconstruction methodsActa Automatica Sinica,2013,39(8):1202-1213.)
3 FreemanW T,JonesT R,PasztorE C. Example?based super?resolution. Computer Graphics and Applications IEEE,2002,22(2):56-65.
4 YangJ C,WrightJ,HuangT,et al. Image super?resolution as sparse representation of raw image patches∥Proceedings of 2008 IEEE Conference on Computer Vision and Pattern Recognition. Anchorage,AK,USA:IEEE Press,2008:1-8.
5 ZeydeR,EladM,ProtterM. On single image scale?up using sparse?representations∥Proceeding of 7th International Conference on Curves and Surfaces. Springer Berlin Heidelberg,2010:24-30.
6 WoldS,EsbensenK,GeladiP. Principal component analysis. Chemometrics and Intelligent Laboratory Systems,1987,2(1-3):37-52.
7 TimofteR,DeV,Van GoolL. Anchored neighborhood regression for fast example?based super?resolution∥Proceedings of 2013 IEEE International Conference on Computer Vision. Sydney,Australia:IEEE,2013:1921-1927.
8 DongC,LoyC C,HeK M,et al. Image super?resolution using deep convolutional networks. IEEE Transactions on Pattern Analysis and Machine Intelligence,2014,38(2):295-307.
9 DonohoD L. For most large underdetermined systems of equations,the minimal 1?norm near?solution approximates the sparsest near?solution. Communications on Pure and Applied Mathematics,2006,59(7):907-934.
10 EnganK,AaseS O,HusoyJ H. Frame based signal compression using method of optimal directions (MOD)∥Proceedings of the 1999 IEEE International Symposium on Circuits and System. Orlando,FL,USA:IEEE,1999:1-4.
11 AharonM,EladM,BrucksteinA. K?SVD:an algorithm for designing overcomplete dictionaries for sparse representation. IEEE Transactions on Signal Processing,2006,54(11):4311-4322.
12 KeyvanradM A,HomayounpourM M. A brief survey on deep belief networks and introducing a new object oriented toolbox (DeeBNet). 2014,arXiv:1408.3264.
13 SmolenskyP. Information processing in dynamical systems:foundations of harmony theory. Boston:MIT Press,1986,194-281.
14 KrizhevskyA. Learning multiple layers of features from tiny images. Master Dissertation. Toronto:University of Toronto,2009.
15 HintonG E,OsinderoS,TehY W. A fast learning algorithm for deep belief nets. Neural Computation,2014,18(7):1527-1554.
16 蒋建国,陈亚运,齐美彬等. 基于自相似性和稀疏表示的图像超分辨率重建. 光电工程,2015,42(12):74-81.
Jiang J G,Chen Y Y,Qi M B,et al.Image super?resolution reconstruction based on selfsimilarity and sparse representation. Opto?Electronic Engineering,2015,42(12):74-81.
17 BevilacquaM,RoumyA,GuillemotC,et al. Low?complexity single image super?resolution based on nonnegative neighbor embedding∥Proceedings of British Machine Vision Conference. Guildford,British:BMVA Press,2012:1-10.
18 李祚林,李晓辉,马灵玲等. 面向无参考图像的清晰度评价方法研究. 遥感技术与应用,2011,26(2):239-246.
Li Z L,Li X H,Ma L L,et al.Research of definition assessment based on no?reference digital image quality. Remote Sensing Technology & Application,2011,26(2):239-246.
19 路鹤晴,朱国英,卓维海等. 医用X射线CT辐射剂量影响因素研究. 上海医学影像,2008,17(2):93-96,136.
Lu H Q,Zhu G Y,Zhuo W H,et al.An evaluation on influence factors of radiation dose for the medical X?ray computed tomography. Shanghai Medical Imaging,2008,17(2):93-96,136.
20 ClarkK,VendtB,SmithK,et al. The cancer imaging archive (TCIA):maintaining and operating a public information repository. Journal of Digital Imaging,2013,26(6):1045-1057.
[1] 罗鸣威2,王怀登1*,丁 尧1,袁 杰2. 变焦序列图像超分辨率重建算法研究[J]. 南京大学学报(自然科学版), 2017, 53(1): 165-.
[2] 王喆正, 唐 晔, 杨育彬 *
. 利用图像类标信息的自调式字典学习方法
[J]. 南京大学学报(自然科学版), 2015, 51(2): 320-327.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
[1] 许 林,张 巍*,梁小龙,肖 瑞,曹剑秋. 岩土介质孔隙结构参数灰色关联度分析[J]. 南京大学学报(自然科学版), 2018, 54(6): 1105 -1113 .
[2] 徐扬,周文瑄,阮慧彬,孙雨,洪宇. 基于层次化表示的隐式篇章关系识别[J]. 南京大学学报(自然科学版), 2019, 55(6): 1000 -1009 .
[3] 刘作国,陈笑蓉. 汉语句法分析中的论元关系模型研究[J]. 南京大学学报(自然科学版), 2019, 55(6): 1010 -1019 .
[4] 游杰, 胡广, 张玺华, 沈安江, 彭瀚霖, 田兴旺, 赵东方. 微生物碳酸盐岩同生⁃早成岩阶段有机质降解示踪:以四川盆地灯影组四段为例[J]. 南京大学学报(自然科学版), 2020, 56(3): 308 -321 .
[5] 戴海亮,沈斌,李开开,张小涛,徐学敏,许智超,周晶晶. 地质条件约束下川北二叠系大隆组富有机质页岩热模拟生烃过程及特征研究[J]. 南京大学学报(自然科学版), 2020, 56(3): 382 -392 .
[6] 刘文平,周政,吴娟,罗超,吴伟,姜磊,焦堃,叶玥豪,邓宾. 川南盆地长宁页岩气田五峰组⁃龙马溪组成藏动力学过程及其意义[J]. 南京大学学报(自然科学版), 2020, 56(3): 393 -404 .
[7] 高云樵,马建敏. 变精度区间集概念格[J]. 南京大学学报(自然科学版), 2020, 56(4): 437 -444 .
[8] 任睿,张超,庞继芳. 有限理性下多粒度q⁃RO模糊粗糙集的最优粒度选择及其在并购对象选择中的应用[J]. 南京大学学报(自然科学版), 2020, 56(4): 452 -460 .