yolov5_7.0-pyside6火焰烟雾识别源码

作品简介

详细介绍查看文章: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



创作时间: