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