基于密集连接注意力块的双生成器图像修复算法

作     者：胡海燕 李硕 刘斌 HU Haiyan;LI Shuo;LIU Bin

作者机构：国网陕西省电力有限公司榆林供电公司陕西榆林719000 陕西科技大学电子信息与人工智能学院陕西西安710021 

基　　金：国家自然科学基金项目(61871260) 

出 版 物：《微型电脑应用》 (Microcomputer Applications)

年 卷 期：2024年第40卷第2期

页      码：1-5页

摘      要：针对图像修复痕迹明显、模型训练不稳定等问题,设计一种结合密集连接注意力块的图像修复算法。在生成器中引入精修复和粗修复二阶段修复网络,并在精修复网络中使用4个通道注意力块设计的密集连接注意力块;同时,增设VGG16特征提取模型,引入WGAN-GP作为判别器损失函数,以多损失融合的方式提高图像的修复效果。在CelebA数据集上验证模型的修复效果,该算法在主客观指标上均优于DCGAN、CE和DD这3种主流算法。

主 题 词：图像修复 生成对抗网络 通道注意力块 密集连接网络 VGG16 

学科分类：1305[艺术学-设计学类] 13[艺术学] 08[工学] 081104[081104] 0804[工学-材料学] 081101[081101] 0811[工学-水利类] 

D　O　I：10.3969/j.issn.1007-757X.2024.02.002

馆 藏 号：203126132...