基于改进知识一致性注意力的图像修复算法 |
| |
引用本文: | 李海燕,尹浩林,钟杏苑,张榆锋.基于改进知识一致性注意力的图像修复算法[J].华中科技大学学报(自然科学版),2023(9):110-117. |
| |
作者姓名: | 李海燕 尹浩林 钟杏苑 张榆锋 |
| |
作者单位: | 云南大学信息学院 |
| |
基金项目: | 国家自然科学基金资助项目(62266049);;国家级大学生创新创业训练计划支持项目(202110673087); |
| |
摘 要: | 为有效解决修复大面积不规则洞孔出现的纹理模糊、结构失真等问题,提出了基于改进知识一致性注意力机制图像修复算法.首先使用部分卷积对待修复图像进行处理,随后将处理的特征图送入包含混合空洞卷积(HDC)与改进知识一致性注意力机制(KCA)的特征推理模块.推理完成后将输出特征图输入至部分卷积与特征推理模块进行循环推理,逐步提升网络的修复能力,循环完成后对输出图像进行特征合并.最后将合并特征图进行去残差瓶颈层以增强修复图像的结构完整性.提出模型使用组归一化方式(GN)加快损失函数收敛速度.在公开数据集上验证提出算法的性能,主客观实验结果表明:提出算法能有效修复大面积连续不规则区域,能够较好地避免修复失真,其峰值信噪比和结构相似度及运算速度优于对比算法.
|
关 键 词: | 图像修复 混合空洞卷积 不规则洞孔 注意力机制 组归一化 |
|
|