[发明专利]一种红外图像的处理方法、装置、介质及电子设备

说明书

本申请实施例公开了一种红外图像的处理方法、装置、介质及电子设备。其中，该方法包括：获取待处理的红外图像和可见光图像；其中，所述红外图像和可见光图像是在相同场景拍摄的；将所述待处理的红外图像和可见光图像作为配准对，输入至预先训练的卷积神经网络模型；根据所述卷积神经网络模型的输出结果，确定所述待处理的红外图像和可见光图像的融合参数；根据所述融合参数进行待处理的红外图像和可见光图像的融合处理。本技术方案，可以提高红外图像的信噪比和对比度，增强红外图像的视觉显示效果。

技术领域

本申请实施例涉及互联网技术领域，尤其涉及一种红外图像的处理方法、装置、介质及电子设备。

背景技术

随着科技水平的迅速发展，红外图像处理的方法技术也在逐步完善，在军事以及日常生活中有着越来越广泛的应用。红外图像处理技术是基于红外热像仪接收温度在绝对零度以上物体的热辐射后形成的红外图像并对其作进一步处理、分析和融合的技术。

然而，由于目标及其周围环境存在着热交换，加之空气对热辐射的散射和吸收作用，目标信号受背景的干扰较大，导致在红外图像中目标和背景的对比度较差，且目标边缘模糊。因此，如何能够增强红外图像的显示效果，成为本领域技术人员亟需解决的技术难题。

发明内容

本申请实施例提供一种红外图像的处理方法、装置、介质及电子设备，可以通过建立一种卷积神经网络模型对红外图像做进一步分析和融合，从而实现增强红外图像显示效果的目的。

第一方面，本申请实施例提供了一种红外图像的处理方法，所述方法包括：

获取待处理的红外图像和可见光图像；其中，所述红外图像和可见光图像是在相同场景拍摄的；

将所述待处理的红外图像和可见光图像作为配准对，输入至预先训练的卷积神经网络模型，卷积神经网络模型包括主干网络、传输网络和输出网络；所述主干网络的输入包括三个可见光图像通道和一个红外图像通道，所述主干网络的第一层卷积采用分组卷积，所述第一层卷积后的卷积层进行至少两次下采样，得到至少两个水平的特征；所述传输网络用于对得到的至少两个水平的特征进行特征融合；所述输出网络用于得到输出结果；

根据所述卷积神经网络模型的输出结果，确定所述待处理的红外图像和可见光图像的融合参数；

根据所述融合参数进行待处理的红外图像和可见光图像的融合处理。

第二方面，本申请实施例提供了一种红外图像的处理装置，该装置包括：

图像获取模块，用于获取待处理的红外图像和可见光图像；其中，所述红外图像和可见光图像是在相同场景拍摄的；

图像预处理模块，用于将所述待处理的红外图像和可见光图像作为配准对，输入至预先训练的卷积神经网络模型，卷积神经网络模型包括主干网络、传输网络和输出网络；所述主干网络的输入包括三个可见光图像通道和一个红外图像通道，所述主干网络的第一层卷积采用分组卷积，所述第一层卷积后的卷积层进行至少两次下采样，得到至少两个水平的特征；所述传输网络用于对得到的至少两个水平的特征进行特征融合；所述输出网络用于得到输出结果；

图像融合参数确定模块，用于根据所述卷积神经网络模型的输出结果，确定所述待处理的红外图像和可见光图像的融合参数；

图像融合处理模块，用于根据所述融合参数进行待处理的红外图像和可见光图像的融合处理。

第三方面，本申请实施例提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现如本申请实施例所述的红外图像的处理方法。

第四方面，本申请实施例提供了一种电子设备，包括存储器，处理器及存储在存储器上并可在处理器运行的计算机程序，所述处理器执行所述计算机程序时实现如本申请实施例所述的红外图像的处理方法。