11

yolov5 烟雾和火焰检测

 3 years ago
source link: https://blog.csdn.net/qq_34351067/article/details/115704848
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
yolov5 烟雾和火焰检测_tf_q568897492的博客-CSDN博客

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

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

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

yolov5.zip.mp4(源码和运行配置word)

fire_smoke_labels.zip.mp4(数据集的标签文件)

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

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

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

云注册地址

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

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

1环境安装

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

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

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

4推断视频

5运行结果

三个下载链接

源码yolov5.zip.mp4

数据集标签fire_smoke_labels.zip.mp4

数据集图片small_dataset.zip.mp4

测试效果视频


About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK