Invention Grant
- Patent Title: 目标检测模型的训练方法、目标检测方法及装置
-
Application No.: CN202011378745.3Application Date: 2020-12-01
-
Publication No.: CN112200862BPublication Date: 2021-04-13
- Inventor: 秦泽奎 , 李强
- Applicant: 北京达佳互联信息技术有限公司
- Applicant Address: 北京市海淀区上地西路6号1幢1层101D1-7
- Assignee: 北京达佳互联信息技术有限公司
- Current Assignee: 北京达佳互联信息技术有限公司
- Current Assignee Address: 北京市海淀区上地西路6号1幢1层101D1-7
- Agency: 北京清亦华知识产权代理事务所
- Agent 韩海花
- Main IPC: G06T7/70
- IPC: G06T7/70 ; G06Q10/04 ; G06N3/04 ; G06K9/62

Abstract:
本公开关于一种目标检测模型的训练方法、目标检测方法及装置,训练方法包括:将获取的训练图像的特征图输入目标检测模型的级联头部网络,以得到各级头部网络进行中心点预测输出的预测热力图,将第一级头部网络输出的预测热力图内各位置点均作为目标位置点,对其余各级头部网络,将前一级头部网络输出的预测热力图中概率大于阈值的位置点,作为后一级头部网络输出的预测热力图中的目标位置点,根据各级头部网络输出的预测热力图与对应的标签热力图在目标位置点的概率差异,训练目标检测模型。本公开通过多个级联头部网络,将属于目标中心点的概率小于阈值的位置点逐级过滤掉,减少了预测位置点的数量,降低了训练难度,提高了模型的检测精度。
Public/Granted literature
- CN112200862A 目标检测模型的训练方法、目标检测方法及装置 Public/Granted day:2021-01-08
Information query