公开(公告)号：CN117611860A

公开(公告)日：2024-02-27

申请号：CN202311585637.7

申请日：2023-11-24

Applicant: 北京京航计算通讯研究所

Inventor： 章宽爽 ,  王学勇 ,  李晶 ,  蔡宇 ,  聂立东 ,  曲昭然

IPC: G06V10/762 ,  G06V10/774 ,  G06V20/70 ,  G06V10/80 ,  G06V10/82

Abstract: 本发明涉及一种基于锚定框的目标检测模型的训练方法，所述方法包括以下步骤：对生产现场历史图像中的所有检测目标进行标注并设定锚定框构建训练样本集；构建目标检测模型，基于所述训练样本集对所述目标检测模型进行训练得到训练好的目标检测模型；每次训练时，根据获取的预测框计算锚定框召回率，若召回率达不到设定阈值则对锚定框进行更新，基于更新后的锚定框进行下一次的训练。本发明通过对图像中的检测目标进行标注以及设定锚定框，并且通过聚类算法对锚定框进行更新，基于更新的锚定框对目标检测模型进行训练，提高了目标检测模型的训练速度。