基于YOLOV5算法的人头检测方法、模型训练方法和装置制造方法及图纸

本申请公开一种基于YOLOV5算法的人头检测方法、模型训练方法和装置，涉及图像处理技术领域。人头检测方法包括：获取待检测图像；确定与待检测图像对应的多个预设尺寸；将待检测图像和多个预设尺寸，输入到预设的多尺寸头肩检测模型中进行检测，确定待检测图像中包括的人头数量；其中，多尺寸头肩检测模型是使用获取到的样本图像和随机选取的多个训练尺寸对初始头肩检测模型进行训练，获得的模型，初始头肩检测模型是基于YOLOV5算法构建的模型。能够通过多尺寸头肩检测模型增强对待检测图像的检测精准性，并使用多个预设尺寸适应性对待检测图像中的不同区域进行人头数量的检测，提升检测准确性。升检测准确性。升检测准确性。

【技术实现步骤摘要】
基于YOLOV5算法的人头检测方法、模型训练方法和装置


[0001]本申请涉及图像处理
，具体涉及一种基于YOLOV5算法的人头检测方法、模型训练方法和装置。

技术介绍

[0002]随着视频监控系统在不同领域中的广泛应用，用户可以对获取到的监控视频数据中的人员数量进行统计，以获取在不同场景下出现的人员情况。例如，可以通过公共区域安装的监控摄像装置对出现在该公共区域内的人物的头部特征进行检测，可以快捷高效的确定出现在该公共区域内的人员数量，能够有效降低目标被遮挡的比例，提升对公共区域的人员数量的监控准确性。
[0003]但是，在一些复杂的场景下(例如，在高铁候车室中)，出现的人头的数量过多，摄像装置的安装位置低于人物的身高的情况下，使得通过摄像装置获取到的图像中，各个人物对应的图像面积差异较大，例如，距离摄像装置远的行人，对应的图像面积会很小；而距离摄像装置近的行人，对应的图像面积会很大，使得对出现在高铁候车室中的人员数量的统计结果不准确，无法精准的控制高铁候车室中的人员数量。

技术实现思路

[0004]为此，本申请本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于YOLOV5算法的人头检测方法，其特征在于，所述方法包括：获取待检测图像；确定与所述待检测图像对应的多个预设尺寸；将所述待检测图像和多个所述预设尺寸，输入到预设的多尺寸头肩检测模型中进行检测，确定所述待检测图像中包括的人头数量；其中，所述多尺寸头肩检测模型是使用获取到的样本图像和随机选取的多个训练尺寸对初始头肩检测模型进行训练，获得的模型，所述初始头肩检测模型是基于YOLOV5算法构建的模型。2.根据权利要求1所述的方法，其特征在于，所述确定与所述待检测图像对应的多个预设尺寸，包括：获取所述待检测图像中预设区域中的待检测人头与图像获取装置之间的位置关系信息；依据所述位置关系信息和预设基准距离，确定多个所述预设尺寸。3.根据权利要求1所述的方法，其特征在于，多个所述预设尺寸包括：尺寸依次递增的第一检测尺寸、第二检测尺寸和第三检测尺寸；所述将所述待检测图像和多个所述预设尺寸，输入到预设的多尺寸头肩检测模型中进行检测，确定所述待检测图像中包括的人头数量，包括：将所述待检测图像和所述第一检测尺寸输入到所述多尺寸头肩检测模型中进行检测，获得第一检测结果；将所述待检测图像和所述第二检测尺寸输入到所述多尺寸头肩检测模型中进行检测，获得第二检测结果；将所述待检测图像和所述第三检测尺寸输入到所述多尺寸头肩检测模型中进行检测，获得第三检测结果；依据所述第一检测结果、所述第二检测结果和所述第三检测结果，确定所述待检测图像中包括的人头数量。4.根据权利要求3所述的方法，其特征在于，所述第二检测尺寸是在确定所述待检测图像中的待检测人头与图像获取装置之间的待检测距离等于预设基准距离的情况下，确定的尺寸。5.根据权利要求3或4所述的方法，其特征在于，所述第一检测结果包括第一待处理头肩图像；所述第二检测结果包括第二待处理头肩图像；所述第三检测结果包括第三待处理头肩图像；所述依据所述第一检测结果、所述第二检测结果和所述第三检测结果，确定所述待检测图像中包括的人头数量，包括：依据预设变换尺寸，分别对所述第一待处理头肩图...

【专利技术属性】
技术研发人员：雷庆庆，王晓，毛少将，任峰，
申请(专利权)人：通号通信信息集团有限公司，
类型：发明
国别省市：