应用二次IoU损失函数的无锚框目标检测方法
本发明提供一种应用二次IoU损失函数的无锚框目标检测方法,其特征在于,包括:将待检测图像进行预处理再输入至预先训练好的目标检测模型进行推理得到对应的图像检测结果,其中,目标检测模型预先通过如下训练得到:获取训练用图像数据集以及监督信息;构建初始目标检测模型,并将训练用图像数据集以及监督信息输入至该模型;使用二次IoU损失函数求定位损失,并使用Focal Loss损失函数求分类损失;分别用定位损失以及分类损失对初始目标检测模型的模型参数进行求导,再使用反向传播对模型参数进行更新;判断更新后的模型参数是否达到终止条件,当判断为是则进入下一步,否则进入训练过程第二步;保存更新后的模型参数并进行加载得到目标检测模型。
复旦大学
2021-01-12