
YOLOv5火焰检测代码及训练完成的fire模型(含2000数据集)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:本项目提供了基于YOLOv5框架的火焰检测解决方案,包括完整的训练代码和使用2000张图像数据集训练得到的预训练模型。
训练好的YOLOv5火焰检测模型包含一个数据集,该数据集中有2000多张带有标签的火焰图片。这些标签以xml和txt两种格式提供,并且唯一的类别名为fire。配置好YOLOv5环境后可以直接使用这个模型进行工作。
参考的数据集及检测结果可以在相关博客文章中找到。需要注意的是,该数据集是基于pytorch框架开发的,代码采用python编写。
全部评论 (0)
还没有任何评论哟~


