基于改进目标检测算法的吸烟行为检测

摘要

本文提出了一种基于YOLOv5算法和图像处理算法的吸烟行为检测方法。针对香烟这种小目标物体检测,采用K-means算法,通过增加小目标检测层的方法对YOLOv5算法进行改进,实现了检测精度的提升。在自制数据集上,误检率为0%,平均精度为92.3%,比YOLOv5s算法高6.7%。