一种红外图像检测模型的训练方法、装置及红外图像检测方法

本申请公开了一种红外图像检测模型的训练方法、装置及红外图像检测方法，可用于人工智能领域，该方法中，首先，获取包含红外图像的训练图片；而后，基于训练图片，通过预先构建的红外图像检测模型，输出缺陷检测结果；继而，基于缺陷检测结果与训练图片对应的缺陷标签的二分图匹配结果，计算第一损失函数；接着，计算第一损失函数与多层感知损失项之和，得到损失函数；最后，基于损失函数对红外图像检测模型进行迭代训练。由此，基于二分图匹配结果得到的第一损失函数与多层感知损失项，采用复合的损失函数对红外图像检测模型进行训练，训练后的红外图像检测模型可对红外图像中小目标实现更为准确的检测。