一种基于双主干网络的吸烟行为检测模型构建方法技术

本发明专利技术公开了一种基于双主干网络的吸烟行为检测模型构建方法。该模型由五个部分组成：ROI主干网络、空间约束网络、香烟主干网络、香烟颈部网络和香烟检测头组成。通过迁移学习，利用ROI主干网络获取各尺度的吸烟姿态特征；利用空间约束网络，面向将各尺度的吸烟姿态特征图，提取吸烟姿态Mask；利用不同尺度的吸烟姿态Mask对香烟主干网络获取的各尺度香烟特征进行空间滤波；滤波后的香烟特征经过颈部网络和检测头输出香烟目标的坐标与置信度。采用本发明专利技术方法，将姿态检测、图像分割和目标检测相融合；通过上下文的空间语义约束，有效减少图像中类似香烟物体的误检率，提高检测准确性。确性。确性。

【技术实现步骤摘要】
一种基于双主干网络的吸烟行为检测模型构建方法


[0001]本专利技术属于深度学习目标检测领域，具体为一种基于双主干网络的吸烟行为检测模型构建方法。
技术背景
[0002]香烟目标检测属于小目标检测，在复杂的场景中存在着大量与香烟类似的物体，因此直接检测香烟会造成大量的误检和漏检情况的发生。传统“先人脸、后香烟”分段式的检测方法可以有效地提高检测的准确性，然而分段式检测需要两次调用YOLO模型，计算时间较长，实时性不佳。

技术实现思路

[0003]专利技术目的：针对上述问题，本专利技术提出一种基于双主干网络的吸烟行为检测模型构建方法。针对现有方法存在的不足，利用ROI主干网络与FCN空间约束网络产生吸烟姿态Mask；在香烟主干网络的特征提取过程中进行空间滤波，减少类似香烟物体的误检；同时将吸烟姿态检测、图像分割、香烟检测相融合，设计端到端的吸烟检测模型，增强检测准确性。
[0004]本专利技术有效利用机器学习和深度学习方法，优化现有YOLO模型的主干网络结构，使香烟主干网络在输出特征之前，考虑到吸烟姿态与香烟目标的上下文关本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于双主干网络的吸烟行为检测模型构建方法，其特征在于：包括如下步骤：步骤1)以YOLOv5作为模型，利用吸烟姿态数据集，通过迁移学习获取ROI主干网络的参数；步骤2)以YOLOv5作为模型，利用香烟数据集，通过迁移学习获取香烟检测模型，包括香烟主干网络、香烟颈部网络和香烟检测头的参数；步骤3)以全卷积网络FCN作为空间约束网络，将ROI主干网络与香烟主干网络并联，构建双主干网络吸烟行为检测模型；步骤4)利用构建好的双主干网络吸烟行为检测模型，对待测图像进行吸烟行为检测。2.根据权利要求1所述的基于双主干网络的吸烟行为检测模型构建方法，其特征在于：所述的步骤1)中，以YOLOv5作为模型，利用吸烟姿态数据集，通过迁移学习获取ROI主干网络的参数，具体步骤如下：首先，收集嘴部正在吸烟的图像，标注、制作“吸烟姿态数据集”；标注框内要求包含人脸与香烟；然后，利用“吸烟姿态数据集”训练YOLOv5模型；将训练好的吸烟姿态YOLOv5模型中的主干网络CSPDarknet保存作为ROI主干网络。3.根据权利要求1所述的基于双主干网络的吸烟行为检测模型构建方法，其特征在于：所述的步骤2)中，以YOLOv5作为模型，利用香烟数据集，通过迁移学习获取香烟检测模型，包括香烟主干网络、香烟颈部网络和香烟检测头的参数，具体步骤如下：首先，收集香烟图像，标注、制作“香烟数据集”；然后，利用“香烟数据集”训练YOLOv5模型；训练好的香烟检测YOLOv5模型包括：香烟主干网络、香烟颈部网络和香烟检测头...

【专利技术属性】
技术研发人员：王晗，金文，刘文杰，张新宇，姚正发，王文忠，王欢，张翟容，
申请(专利权)人：江苏金海星导航科技有限公司，
类型：发明
国别省市：