[发明专利]图像处理模型训练方法、装置、计算机设备和存储介质

说明书

本申请涉及一种图像处理模型训练方法、装置、计算机设备、存储介质和计算机程序产品。所述方法包括：获取待训练图像对，以及所述待训练图像对中红外图像对应的第一分辨率字典，以及所述待训练图像对中可见光图像对应的第二分辨率字典；对所述红外图像和所述可见光图像进行特征提取，得到对应的三维红外矩阵和三维可见光矩阵；基于所述第一分辨率字典和所述第二分辨率字典，对所述三维红外矩阵进行预测处理，得到所述红外图像对应的预测可见光矩阵；计算所述预测可见光矩阵和所述三维可见光矩阵之间的误差损失，基于所述误差损失对图像处理模型进行训练，得到目标图像处理模型。采用本方法能够提高图像处理模型的准确性。

技术领域

本申请涉及计算机技术领域，特别是涉及一种图像处理模型训练方法、装置、计算机设备、存储介质和计算机程序产品。

背景技术

随着计算机技术的发展，出现了智能检测技术，智能检测技术通过对自动采集的图像或者视频中的图像帧进行分析，得到检测对象的检测结果。

传统技术中，对于光线不充足的的环境，自动采集的图像一般为红外图像，通过对红外图像进行分析得到的检测结果，由于红外图像的低分辨率，导致了检测结果准确率低的问题。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提高图像处理模型准确性的图像处理模型训练方法、装置、计算机设备、计算机可读存储介质和计算机程序产品。

第一方面，本申请提供了一种图像处理模型训练方法。所述方法包括：

获取待训练图像对，以及所述待训练图像对中红外图像对应的第一分辨率字典，以及所述待训练图像对中可见光图像对应的第二分辨率字典；

对所述红外图像进行特征提取，得到所述红外图像对应的三维红外矩阵，对所述可见光图像进行所述特征提取，得到所述可见光图像对应的三维可见光矩阵；

基于所述第一分辨率字典和所述第二分辨率字典，对所述三维红外矩阵进行预测处理，得到所述红外图像对应的预测可见光矩阵；

计算所述预测可见光矩阵和所述三维可见光矩阵之间的误差损失，基于所述误差损失对图像处理模型进行训练，得到目标图像处理模型；所述目标图像处理模型用于将红外图像变换成为可见光图像。

在一个实施例中，所述对所述红外图像进行特征提取，得到所述红外图像对应的三维红外矩阵，对所述可见光图像进行所述特征提取，得到所述可见光图像对应的三维可见光矩阵包括：

将所述红外图像划分成为预设尺寸的红外图像块，将所述可见光图像划分成为所述预设尺寸的可见光图像块；

对所述红外图像块进行所述特征提取，得到所述红外图像块对应的三维红外矩阵，对所述红外图像块对应的可见光图像块进行所述特征提取，得到所述可见光图像块对应的三维可见光矩阵；

所述基于所述第一分辨率字典和所述第二分辨率字典，对所述三维红外矩阵进行预测处理，得到所述红外图像对应的预测可见光矩阵，包括：

基于所述第一分辨率字典和所述第二分辨率字典，对所述红外图像块对应的三维红外矩阵进行预测处理，得到所述红外图像块对应的预测可见光矩阵；

所述计算所述预测可见光矩阵和所述三维可见光矩阵之间的误差损失，包括：

计算所述红外图像块对应的预测可见光矩阵和所述可见光图像块对应的三维可见光矩阵之间的误差损失。

在一个实施例中，所述基于所述第一分辨率字典和所述第二分辨率字典，对所述红外图像块对应的三维红外矩阵进行预测处理，得到所述红外图像块对应的预测可见光矩阵包括：

基于所述第一分辨率字典，对所述三维红外矩阵进行去稀疏处理，得到复杂特征矩阵；