[发明专利]图像重建方法、训练方法、检测方法、装置和存储介质

说明书

本申请公开了一种图像重建方法、训练方法、检测方法、装置和存储介质，该方法包括：基于重建网络对待处理红外图像进行重建处理，得到重建红外图像；其中，重建网络是基于红外样本图像进行训练的，红外样本图像包括第一红外训练图像与第二红外训练图像，第一红外训练图像的内容与第二红外训练图像的内容相同，第一红外训练图像的分辨率小于第二红外训练图像的分辨率，且第一红外训练图像的视场角与第二红外训练图像的视场角匹配。通过上述方式，本申请能够提升图像重建的质量。

技术领域

本申请涉及图像处理技术领域，具体涉及一种图像重建方法、训练方法、检测方法、装置和存储介质。

背景技术

与可见光图像相比，红外图像不受光照以及遮挡的影响，因此被广泛应用在智能交通或越线报警等场景中；但是不易获取分辨率高、清晰的红外图像，常见的红外图像大多比较模糊、清晰度不高，致使执行后续操作(比如：红外目标检测任务)时效果不佳，因此需要提升红外图像的视觉效果，但是相关技术方案中所采用的方案对红外图像的提升效果较差。

发明内容

本申请提供图像重建方法、训练方法、检测方法、装置和存储介质，能够提升图像重建的质量。

为解决上述技术问题，本申请采用的技术方案是：提供一种图像重建方法，该方法包括：获取待处理红外图像；基于重建网络对待处理红外图像进行重建处理，得到重建红外图像；其中，重建网络是基于红外样本图像进行训练的，红外样本图像包括第一红外训练图像与第二红外训练图像，第一红外训练图像的内容与第二红外训练图像的内容相同，第一红外训练图像的分辨率小于第二红外训练图像的分辨率，且第一红外训练图像的视场角与第二红外训练图像的视场角匹配。

为解决上述技术问题，本申请采用的技术方案是：提供一种重建网络的训练方法，该方法包括：获取第一红外图像与第二红外图像，第一红外图像的分辨率小于第二红外图像的分辨率；对第一红外图像与第二红外图像进行特征匹配，生成第三红外图像；基于第三红外图像，对第一红外图像进行模板匹配处理，得到第一红外图像中的第一红外训练图像；基于第一红外训练图像，对第三红外图像进行处理，得到第三红外图像中与第一红外训练图像匹配的第二红外训练图像；利用训练样本对重建网络进行训练，得到训练完的重建网络，训练样本包括第一红外训练图像与第二红外训练图像。

为解决上述技术问题，本申请采用的技术方案是：提供一种红外目标检测方法，该方法包括：获取待处理红外图像；基于重建网络对待处理红外图像进行重建处理，得到重建红外图像；对重建红外图像进行目标检测处理，得到目标检测结果；其中，重建网络采用上述技术方案中的重建网络的训练方法训练得到。

为解决上述技术问题，本申请采用的另一技术方案是：提供一种图像处理装置，该图像处理装置包括互相连接的存储器和处理器，其中，存储器用于存储计算机程序，计算机程序在被处理器执行时，用于实现上述技术方案中的图像重建方法、重建网络的训练方法或红外目标检测方法。

为解决上述技术问题，本申请采用的另一技术方案是：提供一种计算机可读存储介质，该计算机可读存储介质用于存储计算机程序，计算机程序在被处理器执行时，用于实现上述技术方案中的图像重建方法、重建网络的训练方法或红外目标检测方法。

通过上述方案，本申请的有益效果是：利用训练样本对重建网络进行训练，得到训练完的重建网络，该红外样本图像包括第一红外训练图像与第二红外训练图像，第一红外训练图像的内容与第二红外训练图像的内容相同，第一红外训练图像的分辨率小于第二红外训练图像的分辨率，且第一红外训练图像的视场角与第二红外训练图像的视场角匹配；在实际应用中，将获取到的待处理红外图像输入训练完的重建网络，便可得到重建红外图像。由于先输入至重建网络的图像为内容相同、分辨率不同且视场角匹配的图像，能够克服输入至重建网络的图像的视场角不匹配的问题，使得重建网络能够更好地学习高分辨率的红外图像的特征，以便在实际使用时对低质红外图像进行更好的重建，改善重建网络的重建效果，在提升低质红外图像的分辨率的同时，还可恢复低质红外图像的纹理细节以及图像中目标的轮廓，进而提升对低质红外图像进行目标检测的准确率。