首页|基于联结扩张注意网络的图像去摩尔纹算法

基于联结扩张注意网络的图像去摩尔纹算法

扫码查看
原文链接
  • 国家科技期刊平台
  • NETL
  • NSTL
  • 万方数据
针对相机拍摄数字显示屏时图像产生的摩尔纹，本文提出了基于联结扩张注意网络的图像去摩尔纹算法。该算法先以轻量级基线U-Net网络为基础，将改进的动态特征金字塔模块DFP引入U-Net的跳跃连接中，并构成联结结构，实现多尺度特征提取与融合，再利用CBAM改进的软注意力机制RMAM来聚焦摩尔纹图像颜色信息，最后构建网络模型对其进行了训练及测试。结果表明，相较于其他算法，该算法具有较高的峰值信噪比和结构相似度。
Image Demoireing Algorithm Based on Associative Extended Attention Network
In order to solve the problem of moire patterns in digital display images taken by cameras, a demoireing algo-rithm based on connected extended attention network is proposed. Firstly, based on lightweight baseline U-Net network, dy-namic feature pyramid (DFP) module is introduced into U-Net skip connections to construct the connection structure and real-ize multi-scale feature extraction and fusion. Secondly, an improved soft attention mechanism based on CBAM, RMAM is used to focus the color information of moire images. Finally, the network model is built on the above foundation, and it is trained and tested. Experimental results show that the proposed algorithm achieves a higher peak signal-to-noise ratio and structural similarity compared to other existing algorithms.

image processingimage demoireingU-Net networkmulti-scalesoft attention mechanism

孙光灵、卢慧敏、陈冲、黄磊、苏亮亮

展开 >

安徽建筑大学 电子与信息工程学院,安徽 合肥 230601

合肥工业大学 智能互联系统安徽省实验室,安徽 合肥 230002

图像处理 图像去摩尔纹 U-Net网络 多尺度 软注意力机制

国家自然科学基金项目安徽省高校协同创新项目安徽省住房城乡建设科学技术计划项目安徽省住房城乡建设科学技术计划项目合肥工业大学"智能互联系统安徽省实验室"开放基金新时代育人质量项目

62001004GXXT-2021-0242023-YF0582023-YF113PA2021AKSK01072023zyxwjxalk112

2024

10.13757/j.cnki.cn34-1328/n.2024.02.009

安庆师范大学学报(自然科学版)
安庆师范学院

安庆师范大学学报(自然科学版)

影响因子：0.252
ISSN：1007-4260
年,卷(期)：2024.30(2)
  • 3