• 改进SDE扩散模型的真实图像去模糊算法

    Real image deblurring algorithm based on improved SDE diffusion model

    • 在图像去模糊中,扩散模型具有较好性能,细节重建方面尤其突出。但是,扩散模型中去噪U-Net网络,难以处理高分辨率图像,且非线性表征能力较弱,无法准确聚焦关键信息,使得恢复的图像存在明显伪影。针对上述问题,基于随机微分方程扩散模型,提出了一种结合多尺度门控扩展单元的图像去模糊算法。首先,结合注意力机制,设计了残差连接的多头深度残差块。在增大感受野的同时,改善了模型处理高分辨率图像的能力,并保留了原始信息。其次,为了提高模型的非线性表征能力,得到准确的噪声分布,设计了多尺度门控扩展单元模块。最后,在多个数据集上对模型进行了验证。实验结果表明,所提方法恢复的图像在视觉上更接近真实情况,纹理细节更加丰富,其PSNR值和SSIM值优于其他常见模型。

       

      Abstract: The diffusion model has good performance in image deblurring, especially in detail reconstruction. However, the denoising U-Net network in the diffusion model is difficult to process high-resolution images, and its non-linear representation ability is weak, making it difficult to accurately focus on key information, resulting in significant artifacts in the restored images. A multi-scale gated extended unit image deblurring algorithm is proposed based on the stochastic differential equation diffusion model to address the above issues. Firstly, in combination with the attention mechanism, a multi-head deep residual block with residual connection is designed. While increasing the receptive field, the model's ability to process high-resolution images has been improved, and the original information has been retained. Secondly, in order to enhance the nonlinear representation ability of the model and obtain an accurate noise distribution, a multi-scale gated expansion unit module is designed. Finally, the model is verified on multiple datasets. The experimental results show that the images restored by the proposed method are visually closer to the real situation, with richer texture details. Its PSNR value and SSIM value are superior to those of other common models.

       

    /

    返回文章
    返回