高级检索
金燕, 黄梦佳, 姜智伟. 基于聚集残差生成对抗网络的图像去模糊[J]. 计算机辅助设计与图形学学报, 2022, 34(1): 84-93. DOI: 10.3724/SP.J.1089.2022.18839
引用本文: 金燕, 黄梦佳, 姜智伟. 基于聚集残差生成对抗网络的图像去模糊[J]. 计算机辅助设计与图形学学报, 2022, 34(1): 84-93. DOI: 10.3724/SP.J.1089.2022.18839
Jin Yan, Huang Mengjia, Jiang Zhiwei. Image Deblurring Based on Aggregate Residual Adversary Networks[J]. Journal of Computer-Aided Design & Computer Graphics, 2022, 34(1): 84-93. DOI: 10.3724/SP.J.1089.2022.18839
Citation: Jin Yan, Huang Mengjia, Jiang Zhiwei. Image Deblurring Based on Aggregate Residual Adversary Networks[J]. Journal of Computer-Aided Design & Computer Graphics, 2022, 34(1): 84-93. DOI: 10.3724/SP.J.1089.2022.18839

基于聚集残差生成对抗网络的图像去模糊

Image Deblurring Based on Aggregate Residual Adversary Networks

  • 摘要: 针对现有图像去模糊算法存在的处理模糊种类单一、耗时长等问题,提出了一种基于聚集残差生成对抗网络的图像去模糊算法.首先,利用生成对抗网络,生成重建图像判别标签,使最后生成的图像更加接近清晰图像;其次,结合聚集残差网络与通道注意力模块,构成特征提取模块,提取中间层的有用特征信息;最后,采用WGAN的Wasserstein-1距离与感知损失结合作为损失函数训练模型,保证生成图像与清晰图像在内容上的一致性.在PyTorch环境下用GOPRO数据集和Kohler数据集测试所提算法,并与L0范数先验、暗通道先验、特异性去模糊、DeepDeblur,DeblurGAN等算法进行对比.实验结果表明,所提算法应用于复原运动模糊图像和高斯模糊图像时,峰值信噪比等评价指标均高于其他算法,并且耗时更短.

     

    Abstract: In order to solve the problems that the existed deblurring algorithms canonlyprocess a single blur type and need long running time, an image deblurring algorithm based on the aggregation residual generation confrontation network is proposed. Firstly, the generation confrontation network is used to generate the reconstructed image discriminant label, so that the final image is closer to the clear image. Secondly, the feature extraction module is combined with the aggregate residual network and the channel attention module to extract the useful feature information of the intermediate layer. Finally, the combination of WGAN’s Wasserstein-1 distance and perceptual loss is used as the loss function to train the model to ensure the consistency of the generated image and the clear image. In the PyTorch environment, the proposed algorithm is tested with the GOPRO dataset and the Kohler dataset, and compared with other algorithms, such as L0 norm prior, dark channel prior, specific deblurring, DeepDeblur and DeblurGAN. Experimental results show that when restoring motion blurred images and Gaussian blurred images, the proposed algorithm outperforms other algorithms in terms of evaluating indicator such as PSNR and the running time.

     

/

返回文章
返回