国内每年都会发生大大小小的火灾,造成生命、财产的损失。但是很多火灾如果能够早期发现,并及时提供灭火措施,将会大大较小损失。本套源码采用yolov5-7.0目标检测算法结合pyside6可视化界面源码,当检测到火灾时,能否发出语音警报,对于火灾的警报工作有着重要的作用。
源码下载地址:yolov5_7.0-pyside6火焰烟雾识别源码
- 支持更改权重模型自定义识别(前提是未改进的网络)
- 支持图片、视频、摄像头检测
- 可以选择是否保存结果
- 当程序检测到火焰时,可以选择是否发出警报声
- pt的ui源文件(可以拖动到qt编辑器进行修改)
环境配置+运行: 1、在anaconda新建独立环境
conda create -n yolo5 python=3.8 -y
2、在新建的环境里面安装依赖
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
3、右键运行wzq.py 说明:如果默认采用上面的pip安装依赖,默认会安装CPU版。 如需安装GPU版,可以卸载torch和torchvision再重新安装,安装代码如下:
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch -y
代码默认采用fire.pt(基于yolov5 7.0版本训练的火焰和烟雾数据集)权重模型进行测试演示,如需检测其他物体, 可以修改成自己的权重,自定义检测,但必须是yolo5 7.0官方版训练的。否则可能会报网络模型异常。
提供了ui源文件,可以拖进QT组件进行编辑修改
更多功能,可以查看B站介绍:基于yolov5+pyside6开发的火灾烟雾警报图形化目标检测系统_哔哩哔哩_bilibili
源码下载地址:yolov5_7.0-pyside6火焰烟雾识别源码