一种基于目标检测和关键点检测的吸烟识别方法及系统技术方案

本发明专利技术公开了一种基于目标检测和关键点检测的吸烟识别方法及系统，方法包括：获取待检测图像，基于深度学习算法对所述待检测图像进行目标检测，获取人脸目标框位置及香烟目标框位置；基于姿态估计网络对所述待检测图像的人脸进行指定关键点检测，得到指定关键点的位置；获取指定关键点的位置包含在人脸的目标框位置的待检测图像，作为目标图像；根据所述目标图像、人脸目标框位置及指定关键点的位置，得到嘴巴目标框位置；计算香烟目标框位置与嘴巴目标框位置的交并比，若所述交并比大于预设阈值，则判定为正在吸烟。本发明专利技术实施例结合人体关键点目标检测的方法，可以提高人体的检测精度，与香烟结合判断时能减少误报，提升香烟的检测精度。的检测精度。的检测精度。

【技术实现步骤摘要】
一种基于目标检测和关键点检测的吸烟识别方法及系统


[0001]本专利技术涉及图像识别
，尤其涉及一种基于目标检测和关键点检测的吸烟识别方法及系统。

技术介绍

[0002]随着深度学习的不断发展，快速成型技术日趋成熟，吸烟识别广泛应用于各类安监系统中，为了满足监控场景和识别需求，需要图像中的吸烟识别的准确性。在现有技术中，提供了一种图像切割的方法，该方法对于较大像素的图片有提升效果，但是在1080p像素的相机上提升效果不大，尤其是对于吸烟这类细长直的小目标的识别仍然较困难，识别精度较低。
[0003]因此现有技术还有待于进一步发展。

技术实现思路

[0004]针对上述技术问题，本专利技术实施例提供了一种基于目标检测和关键点检测的吸烟识别方法及系统，能够解决现有技术中吸烟行为识别精度较低的技术问题。
[0005]本专利技术实施例的第一方面提供一种基于目标检测和关键点检测的吸烟识别方法，方法包括：
[0006]获取待检测图像，基于深度学习算法对所述待检测图像进行目标检测，获取人脸目标框位置及香烟目标框位置本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于目标检测和关键点检测的吸烟识别方法，其特征在于，方法包括：获取待检测图像，基于深度学习算法对所述待检测图像进行目标检测，获取人脸目标框位置及香烟目标框位置；基于姿态估计网络对所述待检测图像的人脸进行指定关键点检测，得到指定关键点的位置；获取指定关键点的位置包含在人脸的目标框位置的待检测图像，作为目标图像；根据所述目标图像、人脸目标框位置及指定关键点的位置，得到嘴巴目标框位置；计算香烟目标框位置与嘴巴目标框位置的交并比，若所述交并比大于预设阈值，则判定为正在吸烟。2.根据权利要求1所述的基于目标检测和关键点检测的吸烟识别方法，其特征在于，所述获取待检测图像，基于深度学习算法对所述待检测图像进行目标检测，获取人脸目标框位置及香烟目标框位置，包括：获取待检测图像，将所述待检测图像输入训练好的yolov5模型进行目标检测，得到人脸目标框位置及香烟目标框位置。3.根据权利要求2所述的基于目标检测和关键点检测的吸烟识别方法，其特征在于，所述获取待检测图像，基于深度学习算法对所述待检测图像进行目标检测，获取人脸目标框位置及香烟目标框位置前，包括：预先构建初始的yolov5网络；采集人脸及香烟的数据集，对所述数据集进行标注，生成标注数据集；基于所述标注数据集对所述初始的yolov5网络进行训练，生成训练好的yolov5模型。4.根据权利要求3所述的基于目标检测和关键点检测的吸烟识别方法，其特征在于，所述基于姿态估计网络对所述待检测图像的人脸进行指定关键点检测，得到指定关键点的位置，包括：基于openpose对所述待检测的人脸进行鼻子及眼睛对应的指定关键点检测，得到鼻子及眼睛对应的指定关键点的位置。5.根据权利要求4所述的基于目标检测和关键点检测的吸烟识别方法，其特征在于，所述根据所述目标图像、人脸目标框位置及指定关键点的位置，得到嘴巴目标框位置，包括：获取所述目标图像中的人脸目标框位置；获取目标图像中鼻子的关键点的横轴坐标，将横轴坐标向横轴延展指定宽度，得到嘴巴目标框的横轴数据；获取目标图像中鼻子的关键点的纵轴坐标，将纵轴坐标向纵轴延展指定高度...

【专利技术属性】
技术研发人员：张金圣，刘奔，刘晨飞，
申请(专利权)人：深圳市铁越电气有限公司，
类型：发明
国别省市：