[发明专利]图像检测模型的训练方法及装置、目标检测方法及装置

摘要

本申请提供了一种图像检测模型的训练方法及装置、目标检测方法及装置，涉及人工智能技术领域。该模型训练方法包括：将目标训练图像输入初始模型；通过骨干网络提取目标训练图像的特征响应图；通过分类子网络对特征响应图进行分类处理，并基于分类处理结果与类别标签计算得到分类损失值；通过回归子网络对特征响应图进行回归处理，并基于回归处理结果与回归标签计算得到回归损失值；通过掩码子网络对特征响应图进行掩码处理，并基于掩码处理结果与掩码标签计算得到掩码损失值；基于分类损失值、回归损失值和掩码损失值对初始模型进行训练，得到目标检测模型。本申请改善了模型训练方法，有效提升了经训练得到的目标检测模型对图像的检测效果。