本发明涉及图像处理,特别是指一种基于多重注意力机制的图像恢复方法及系统。
背景技术:
1、在现代图像处理和计算机视觉领域,图像恢复是一项重要的任务,特别是在监控、医疗成像、遥感等领域,图像恢复的目标是从退化或损坏的图像中恢复出高质量的原始图像,常见的图像退化包括模糊、噪声、低分辨率等,其中模糊退化尤为常见且复杂,通常由相机运动、焦距失准、大气湍流等因素引起。
2、随着社会的快速发展,图像恢复技术越来越智能化,通过图像恢复技术,可以解决由于成像条件、设备性能等带来的图像退化问题,从而提高图像的清晰度、分辨率和可用性,为各类图像分析和视觉任务提供可靠的数据支持,从而满足用户对高质量图像的需求,提升用户的满意度。
3、目前,针对模糊退化的图像恢复方法主要包括传统方法和基于深度学习的方法。传统方法通常基于物理模型,如点扩散函数(psf)建模,通过逆滤波、维纳滤波等技术进行图像恢复。这些方法在特定条件下效果较好,但在实际应用中往往受限于对退化模型的准确性和鲁棒性不足,难以应对复杂的模糊类型。同时传统的深度学习方法通常依赖于单一的特征提取方式,无法有效捕捉图像中的多尺度、多类型信息,导致恢复效果不佳,随着深度学习的发展,随着深度学习的发展,基于卷积神经网络(cnn)的方法逐渐成为主流。基于卷积神经网络的方法通过大量数据训练,能够自动学习到从模糊图像到清晰图像的映射关系。
4、然而,现有的深度学习方法在处理复杂模糊退化时仍面临挑战,由于在处理局部细节时往往忽视了全局上下文信息,导致恢复的图像在细节和结构上存在失真,并且在面对多样化的模糊类型时,泛化能力较弱,容易出现过拟合现象,尤其是在多尺度、多类型的模糊退化情况下,模型的泛化能力和恢复质量有待提高。
技术实现思路
1、为了解决现有技术存在的深度学习方法在处理复杂模糊退化时,由于在处理局部细节时往往忽视了全局上下文信息,导致恢复的图像在细节和结构上存在失真,并且在面对多样化的模糊类型时,泛化能力较弱,容易出现过拟合现象的问题,本发明提供了一种基于多重注意力机制的图像恢复方法及系统。
2、本发明实施例提供的技术方案如下:
3、第一方面
4、本发明实施例提供的一种基于多重注意力机制的图像恢复方法,包括:
5、s1:获取待处理图像;
6、s2:通过卷积层,提取待处理图像的浅层特征;
7、s3:根据浅层特征,通过三重注意力模块,计算注意力特征;通过特征聚合模块,对浅层特征进行处理,提取出深层特征;
8、s4:对注意力特征、深层特征和浅层特征进行融合,得到融合特征;
9、s5:根据融合特征,对待处理图像进行图像恢复,并输出恢复后的图像。
10、第二方面
11、本发明实施例提供的一种基于多重注意力机制的图像恢复系统,包括:
12、处理器;
13、存储器,存储器上存储有计算机可读指令,计算机可读指令被处理器执行时,实现如第一方面的基于多重注意力机制的图像恢复方法。
14、第三方面
15、本发明实施例提供的一种计算机可读存储介质,其上存储有计算机程序,该程序被处理器执行时实现如第一方面的基于多重注意力机制的图像恢复方法。
16、本发明实施例提供的技术方案带来的有益效果至少包括:
17、在本发明中,通过获取待处理图像并逐步提取浅层和深层特征,结合三重注意力模块和特征聚合模块,能够从多维度、多层次全面地增强图像特征。三重注意力模块通过空间、通道和全局注意力的机制,精准捕捉关键细节,优化特征表示,特征聚合模块则有效融合浅层和深层特征,实现对多尺度信息的综合利用,最终,特征融合后的重建步骤保证了恢复图像的高质量,使得该方法在处理复杂退化图像时具有更高的鲁棒性和准确性。该方法通过引入多重注意力机制,能够在不同尺度和不同类型的信息之间进行有效的特征融合和选择,从而提高图像恢复的质量和泛化能力,增强图像的清晰度、分辨率和可用性,为各类图像分析和视觉任务提供可靠的数据支持,从而满足用户对高质量图像的需求,提升用户的满意度。
1.一种基于多重注意力机制的图像恢复方法,其特征在于,包括:
2.根据权利要求1所述的基于多重注意力机制的图像恢复方法,其特征在于,所述s2具体为:
3.根据权利要求1所述的基于多重注意力机制的图像恢复方法,其特征在于,所述s3中根据所述浅层特征,通过三重注意力模块,计算注意力具体包括:
4.根据权利要求3所述的基于多重注意力机制的图像恢复方法,其特征在于,所述s301具体包括:
5.根据权利要求3所述的基于多重注意力机制的图像恢复方法,其特征在于,所述s302具体包括:
6.根据权利要求3所述的基于多重注意力机制的图像恢复方法,其特征在于,所述s303具体包括:
7.根据权利要求1所述的基于多重注意力机制的图像恢复方法,其特征在于,所述s3中通过特征聚合模块,对所述浅层特征进行处理,提取出深层特征具体包括:
8.根据权利要求1所述的基于多重注意力机制的图像恢复方法,其特征在于,所述s4具体包括:
9.一种基于多重注意力机制的图像恢复系统,其特征在于,包括:
10.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,该程序被处理器执行时实现如权利要求1至8中任一项所述的基于多重注意力机制的图像恢复方法。
