yolov5 烟雾和火焰检测

       随着社会经济的高速发展,工业、企业园区,住宅日益增多,存在一些消防安全隐患,导致火灾事故屡屡发生。

       针对住宅,工业园区,森林,加油站等室内外场景,深入了解,并通过目标检测算法,在图像中自动标注疑似烟雾、火焰,实现机器智能值守,具有重大意义。

       运行环境是pytorch1.81 gpu,可以在免费云gpu上直接运行,一共1个压缩包

yolov5源码和数据集.zip.mp4(数据集图片烟雾和火焰分别标注2类,一共2000张)

因为网络问题无法上传zip文件

需要把.mp4后缀去掉改为zip文件才能打开。

云注册地址

https://gpushare.com/auth/register?user=18*****1785&fromId=c4100216220&source=link

具体云gpu环境配置看word文档。

1环境安装

yolov5 烟雾和火焰检测_第1张图片

 

2安装wandb及训练,安装wandb具体细节可以查csdn上的教程。

yolov5 烟雾和火焰检测_第2张图片

3 训练后的[email protected]大概55多

加大训练批次还能继续提升。

yolov5 烟雾和火焰检测_第3张图片

4推断视频

yolov5 烟雾和火焰检测_第4张图片

 

5运行结果

 

yolov5 烟雾和火焰检测_第5张图片

 

yolov5 烟雾和火焰检测_第6张图片

 

下载链接

yolov5源码和数据集.zip.mp4

 

测试效果视频

你可能感兴趣的:(深度学习,pytorch,yolov5,火焰检测,烟雾检测,火灾检测)