首页|基于生成对抗网络的工业场景低质图像增强算法

基于生成对抗网络的工业场景低质图像增强算法

扫码查看
针对工业场景下图像模糊、分辨率低、边缘细节不明显等问题,提出一种基于生成对抗网络的低质图像增强算法.首先,设计退化网络获得与真实场景更为接近的低质图像,以此与现实高清图像获得特征映射关系;其次,在使用密集残差块(residual in residual dense block,RRDB)的基础上添加卷积注意力模块,增强RRDB网络的特征表达能力,以有效地捕获关键特征信息;最后,设计边缘增强网络模块结合改进的RRDB作为生成器,图像细节信息的捕捉与还原能力得到显著提升,并与判别器对抗生成更高质量的图像.实验结果表明,相较于现有常用的图像增强算法,所提算法能有效提升工业场景图像清晰度、保留图像细节并减少失真.定量指标峰值信噪比平均提升10.45%,结构相似性平均提升15.92%,运行速度快,能满足工业生产需求.
Low-Quality Image Enhancement Algorithm for Industrial Scenes Based on Generative Adversarial Networks
Aiming at the problems of image blurring,low resolution and unclear edge details in industrial scenes,this paper proposes a low-quality image enhancement algorithm based on generative adversarial net-works.Firstly,the degraded network is designed to obtain low-quality images closer to the real scene,so as to obtain the feature mapping relationship with the real high-definition images.Secondly,on the basis of u-sing residual in residual dense block(RRDB),a convolution attention module is added to enhance the fea-ture expression ability of RRDB network,so as to effectively capture key feature information.Finally,the edge enhancement network module is designed and combined with the improved RRDB as the generator,the ability of capturing and restoring image detail information is significantly improved,and the higher qual-ity image is generated by confrontation with the discriminator.The experimental results show that compared with the existing commonly used image enhancement algorithms,the proposed algorithm can effectively im-prove the clarity of industrial scene images,preserve image details and reduce distortion.The peak signal-to-noise ratio of the quantitative index is increased by 10.45% on average,and the structural similarity is in-creased by 15.92% on average.The operation speed is fast and can meet the needs of industrial production.

industrial scenedegradationdense residual blocksattentionedge enhancement

叶旭辉、倪蔚恒、陈燕、尹芹凯、张道德

展开 >

湖北工业大学机械工程学院,武汉 430068

武汉东湖学院机电工程学院,武汉 430212

工业场景 退化 密集残差块 注意力 边缘增强

国家自然科学基金面上项目湖北省农机装备补短板核心技术应用攻关项目

52075152HBSNYT202221

2024

组合机床与自动化加工技术
大连组合机床研究所 中国机械工程学会生产工程分会

组合机床与自动化加工技术

CSTPCD北大核心
影响因子:0.671
ISSN:1001-2265
年,卷(期):2024.(9)
  • 12