YOLO-SSDA——基于改进YOLOv9的吸烟行为检测方法  

YOLO-SSDA—Smoking Behavior Detection Method Based on Improved YOLOv9

作　　者：何超勋 陈智霖 黄声勇 彭道福 

机构地区：[1]广东省汕尾市城区香洲头汕尾供电局,广东 汕尾

出　　处：《计算机科学与应用》2024年第9期56-65,共10页Computer Science and Application

摘　　要：针对复杂危险环境下传统吸烟行为检测方法成本较高、实时性较差、识别率低、容易漏检和小目标误判率高等问题,提出一种基于改进YOLOv9的吸烟行为检测算法YOLO-SSDA。YOLO-SSDA通过引入注意力机制模块SimAM,加强骨干网络对香烟特征的提取能力;利用Soft-NMS算法避免在过滤重叠定位框时出现漏检可能;利用距离关联算法计算香烟与人体之间的距离关系,减少香烟识别误判率。本文通过网络采样、AI生成与现场采集等方式构建了实验数据集并进行了大量实验测试,实验结果表明:YOLO-SSDA算法准确率为94%,召回率为89.6%,mAP50值为94.4%,处理单振图像输入的平均耗时为39 ms,与其他算法相比具有较高的检测准确率和实时性能。Aiming at the problems of high cost, poor real-time performance, low recognition rate, easy missed detection, and high misjudgment rate of small targets in traditional smoking behavior detection methods in complex and dangerous environments, a smoking behavior detection algorithm YOLO-SSDA based on improved YOLOv9 is proposed. By introducing the attention mechanism module SimAM, the backbone network’s ability to extract cigarette features is strengthened;using Soft NMS algorithm to avoid the possibility of missed detection when filtering overlapping positioning boxes;using distance correlation algorithm to calculate the distance relationship between cigarettes and the human body, in order to reduce the misjudgment rate of cigarette recognition. This article constructed an experimental dataset through network sampling, AI generation, and on-site collection, and conducted extensive testing based on it. The results show that the YOLO-SSDA algorithm proposed in this paper has an accuracy of 94%, a recall rate of 89.6%, an mAP50 value of 94.4%, and an average processing time of 39 ms for single vibration image input. Compared with other algorithms, it has higher detection accuracy and real-time performance.

关 键 词：吸烟检测 YOLOv9 NMS算法 SimAM 距离关联 

分 类 号：TP3[自动化与计算机技术—计算机科学与技术]