2.118

影响因子

    高级检索

    基于SE模块与空洞卷积的图像修复算法

    Image Inpainting Algorithm Based on SE Module and Atrous Convolution

    • 摘要: 提出一种基于生成对抗网络的两阶段图像修复模型,该模型融入改进的残差块和通道注意力机制。在模型的第 1 阶段,通过训练一个边缘轮廓生成器来输出初步的边缘轮廓修复图;在第2阶段,利用这些修复后的边缘轮廓作为先验知识,进一步通过图像完成网络生成最终的完整修复图像。通过在CelebA-HQ人脸数据集上进行实验,该算法与原始模型及CA模型相比有显著的提升,尤其是在平均绝对误差、结构相似性指数和峰值信噪比等关键性能指标上表现出色。

       

      Abstract: This paper proposes a two-stage image inpainting model based on Generative Adversarial Networks(GANs), which integrates improved residual blocks and a channel attention mechanism. In the first stage of model inpainting, an edge contour generator is trained to output an initial edge contour inpainting map. In the second stage, the image completion network is further employed to generate the final complete inpainted image using these repaired edge contours as prior knowledge. Experiments on the CelebA-HQ face dataset show significant improvements of this algorithm over the original and CA models, especially in key performance indicators such as mean absolute error, structural similarity index, and peak signal-to-noise ratio.

       

    /

    返回文章
    返回