本技术实施例公开一种检测模型的训练方法、检测方法、装置、设备及介质,该方法包括:利用训练样本集对检测模型进行训练,其中,所述训练样本集中包含有原始特征图及其对称特征图;对于每一个训练样本,确定检测模型所预测的头部关键点与该训练样本中标注的头部关键点的第一欧式距离,以及预测的肩部关键点与该训练样本中标注的肩部关键点的第二欧式距离;对预测的肩部关键点进行归一化,并确定肩部归一化后所对应的头部关键点在头部先验概率分布中对应的概率值;根据该概率值、第一欧式距离和第二欧式距离,计算头部关键点对应的加权损失值,当加权损失值达到收敛时,检测模型训练完成;通过采用上述技术方案,有效提高头肩关键点的检测精度。
背景技术
目标检测是计算机视觉领域中一项经典的研究课题,它的任务是指出一张图中目标的准确位置,在无人驾驶、人流统计分析等领域具有广泛的应用。近年来,得益于深度卷积网络的引入,目标检测算法发展迅速,将检测性能提高到了新的高度。
目前,对图像中头部和肩部关键点的检测方案一般有两种:一种是基于传统的图像处理方法先进行局部直线段检测,然后通过启发式准则搜索头肩关键点。一种是人体姿态估计中的人体关键点检测方法,这类方法通常采用CNN提取特征并预测关键点的热力图,然后通过图模型的方法建立关键点之间的连接关系。
但是,基于传统的图像处理的头肩关键点检测方法,依赖直线段检测的精度和召回率,容易受到光照和图像质量的影响出现严重退化的情形。另外基于规则的判断方法能够检索到多个符合规则的头肩关键点,因此容易出现虚警。基于深度学习的人体关键点检测方法由于待估计关键点较多、下半身关键点容易受到视角和姿态的影响,因此估计难度较大,无法得到头肩关键点在图像中的准确位置。
实现思路