本文介绍了如何针对YOLOv5的检测结果进行代码改进，当检测到特定目标如安全帽时，触发语音告警。通过在detect.py中添加条件判断和语音播放代码，实现了当检测到person类别的置信度超过一定阈值时，播放警告音频。此外，文章还展示了改进后的检测效果和提供的相关资源，包括数据集、权重文件和检测界面程序。
一、问题描述
在之前的博文中，好多朋友通过私信和评论给我留言，问到了一个问题：“如果有两个标签，怎么让它识别到其中一个才报警？” 其实下面这位C友已经给出解决方法了，但是大家可能对YOLOv5各个部分的代码还没了解透，所以自己改还有难度。
在这里，我把我自己改进的方法说一下，大家可以改着试试看。
二、改进方法
我的方法比较简单，只要在detect.py里的画框的代码那里加个判断即可。
在这里，我以检测安全帽为例子操作一下。
代码版本：YOLOv5-6.2
检测权重：安全帽检测的开源权重，s版本
检测对象：person和hat两个
步骤一
在detect.py中，找到这一部分代码行，截图如下：
步骤二
然后，输入代码
if names[
最低0.47元/天 解锁文章
扫一扫
258
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
