一种基于双阶段损失函数的目标检测模型训练方法

摘要

本发明涉及目标检测技术领域，提供一种基于双阶段损失函数的目标检测模型训练方法，包括：获取检测图片训练集中的训练图片，根据训练图片的标注信息，提取目标框的位置；将训练图片输入到神经网络模型进行训练，输出多个候选框；依次计算每个候选框与目标框之间的iou值，并根据候选框与目标框的iou值，确定候选框与目标框的iou损失函数；根据确定的损失函数，优化神经网络模型；依次利用检测图片训练集中的每一张训练图片，进行训练，得到目标检测模型。本发明损失函数可以根据iou值是否为零自动选择损失函数，提高网络的收敛速度和最终的模型检测精度。