一种基于深度学习的两阶段红外图像湍流抑制方法及系统

Abstract

本发明提供了基于深度学习的两阶段红外图像湍流抑制方法及系统。方法，包括：获取待恢复红外湍流图像；将待恢复红外湍流图像输入预训练的去湍流模型，预测得到去湍流图像；预训练的去湍流模型包括：一阶段去模糊模型和二阶段去畸变模型，一阶段去模糊模型为基于Transformer的U型神经网络，在多个尺度上通过多头注意力机制和前馈神经网络对待恢复红外湍流图像进行特征提取，并映射得到待识别畸变图像；二阶段去畸变模型针对待识别畸变图像进行通道打乱的混洗处理得到待识别混洗图像，并将待识别混洗图像与待恢复红外湍流图像做像素叠加处理，得到去湍流图像。提高了对红外湍流图像的恢复效果，降低了恢复过程中的资源消耗。

Images