文章目录
演示视频
【基于YOLOv5的反光衣检测预警系统】可检测图片、视频、摄像头，支持GPU加速检测以及语音播报预警
系统界面
项目环境
项目运行所需环境为YOLOv5的环境+PyQt5的环境即可。
系统界面功能
①图片检测，视频检测、摄像头检测
②检测图像、视频与检测速度显示
③可保存结果、调节置信度和交并比
④支持GPU和CPU两种模型加载模型
⑤检测出有人未穿反光衣会语言播报与文字显示预警
项目框架介绍
| 新增文件 | 文件说明 |
|---|---|
| A_identify.py | 写了Identify目标检测类，该类用于初始化模型相关参数并且加载模型、其中写了一个类函数用于目标检测，功能为输入图像或者视频流图像返回检测图像与标签列表 |
| A_main.py | 写了MainUi界面类，该类用于PyQt5编写可视化界面，以及写了相关功能函数（预警功能等）实现响应 |
| A_input文件夹 | 该文件夹放了用来测试的图像视频 |
| A_output文件夹 | 该文件夹用于存放检测保存的检测图像、检测视频和摄像头实时检测视频 |
| A_pt文件夹 | 该文件夹用于放系统所读取的反光衣检测pt模型，如果需要更换检测模型，则将模型换进此文件夹即可 |
| A_warn文件夹 | 该文件夹用于放系统所读取的预警音频mp3，如果需要更换预警音频，则将音频换进此文件夹即可 |
| runs文件夹 | 该文件夹里的train/exp为YOLOv5训练了2000多张反光衣图像出来的模型结果 |
如下图为整个项目目录，除上方表格所述文件，其余文件为YOLOv5-v7.0的源项目文件。
项目代码获取
完整代码与数据集下载： 在上方演示视频置顶评论获取！