详细介绍查看文章:https://blog.51cto.com/wzqgzs/12778466
拍下发送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站介绍:https://www.bilibili.com/video/BV1Te411Y7PP/?vd_source=cefeccaf83f573586d8360b3603f1de3
