简介

吸烟不仅对个人健康有害，也可能在某些特定场合带来安全隐患。为了有效地监控公共场所和工作环境中的吸烟行为，我们开发了一种基于YOLOv5目标检测模型的吸烟检测系统。本报告将详细介绍该系统的实际应用与实现，包括系统架构、功能实现、使用说明、检测示例、数据集获取与介绍、YOLOv5模型介绍及其训练过程。

系统架构

系统组成

1. 用户界面（GUI）：基于PyQt5开发，支持图像、视频和实时摄像头检测功能。
2. 检测模型：基于YOLOv5的目标检测模型，用于识别吸烟行为。
3. 视频处理模块：处理视频流，实现实时吸烟检测。
4. 数据管理模块：负责数据的加载、保存及标注。

工作流程

1. 用户加载图像/视频或启动摄像头。
2. 系统调用YOLOv5模型进行吸烟行为检测。
3. 检测结果显示在GUI上，包括吸烟行为的位置和类别。
4. 用户可以保存检测结果。

功能实现