一种基于块循环扩散模型的渐进图像去雨方法技术

技术编号:40464417 阅读:61 留言:0更新日期:2024-02-22 23:18
本发明专利技术提供一种基于块循环扩散模型的渐进图像去雨方法,以提高扩散模型受噪声估计不准确而影响的图像恢复质量。本发明专利技术一个由粗到细的两阶段训练以构建渐进式图像去雨过程。首先,粗训练阶段通过对设计的噪声估计网络估计出的噪声进行约束,在此基础上,细训练阶段利用训练好的粗扩散模型选择固定采样步长来逐步产生精细的去雨结果,然后将其与干净图像进行约束去优化模型,以弥补不准确的噪声估计带来的影响。本发明专利技术充分利用扩散模型的生成能力,使渐进式的去雨结果更忠实于干净图像,并且采样4步即可恢复出高质量图像,为图像恢复任务提供了一个有前途的新方法。

【技术实现步骤摘要】

本专利技术属于图像复原,涉及一种基于块循环扩散模型的渐进图像去雨方法


技术介绍

1、图像复原旨在将因不同退化因素影响的低质量图像恢复出干净面貌,通常包括以下常见的任务,如图像去噪、图像去模糊、图像超分辨率等,图像去雨也属于此范畴内。由于他们是一个病态的问题,逐渐成为了计算机视觉和图像处理领域热点话题研究。从深度学习兴起到现在,由于卷积神经网络自身强大的数据学习能力,相比于传统方法,衍生出的大量基于卷积神经网络的图像复原方法展示出了独特的领先优势。另一方面,由transformer驱动的网络架构得益于注意力机制对长距离依赖的建模,已经在底层视觉领域大放异彩。

2、近年来,扩散模型作为一种新兴的生成模型,其本质在于通过前向扩散过程将图像逐渐扩散到纯噪声分布,然后通过学习去噪网络模拟逆向过程,在一定的时间序列内生成与原始数据分布一致的样本。由于其从噪声中逐渐恢复的机制,它不仅在各种图像生成任务中取得了显着的成功,而且在解决图像恢复任务方面也做了大量的工作。然而,对于旨在从有雨图像恢复出具有更清晰细节的图像去雨任务来说,仅仅通过约束噪声不能有效地学习本文档来自技高网...

【技术保护点】

1.一种基于块循环扩散模型的渐进图像去雨方法,其特征在于,步骤如下:

2.如权利要求1所述的一种基于块循环扩散模型的渐进图像去雨方法,其特征在于,所述的第三步,具体步骤如下:

3.如权利要求1或2所述的一种基于块循环扩散模型的渐进图像去雨方法,其特征在于,所述的第四步,具体步骤如下:

4.如权利要求1或2所述的一种基于块循环扩散模型的渐进图像去雨方法,其特征在于,所述的第五步,具体步骤如下:

5.如权利要求3所述的一种基于块循环扩散模型的渐进图像去雨方法,其特征在于,所述的第五步,具体步骤如下:

6.如权利要求1或2或5所述的一...

【技术特征摘要】

1.一种基于块循环扩散模型的渐进图像去雨方法,其特征在于,步骤如下:

2.如权利要求1所述的一种基于块循环扩散模型的渐进图像去雨方法,其特征在于,所述的第三步,具体步骤如下:

3.如权利要求1或2所述的一种基于块循环扩散模型的渐进图像去雨方法,其特征在于,所述的第四步,具体步骤如下:

4.如权利要求1或2所述的一种基于块循环扩散模型的渐进图像去雨方法,其特征在于,所述的第五步,具体步骤如下:

5.如权利要...

【专利技术属性】
技术研发人员:苏志勋王丽妍杨沁瑜
申请(专利权)人:大连理工大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1