- 博客(1)
- 资源 (1)
- 收藏
- 关注
原创 基于YOLOv5的火焰图像识别
1.笔者基于pytorch库,采用开源的YOLOv5框架训练出了基于深度学习的能通过计算机视觉来实现目标检测功能的火焰识别模块。进而,笔者便可以使用Pycharm这款IDE来借助pytorch环境进行开发。然后创建数据集文件,创建class.txt,里面输入fire类,导入要标注的图片文件,从cmd进入目录后使用命令。labelimg是一款数据集标注工具,可以将标注的数据保存成不同格式。通过cmd执行命令来安装。来调用labelimg工具标注数据集。3即可载入模型进行训练。
2023-05-15 13:20:24 774 1
基于YOLOv5的火焰图像识别工程
要想对特定的视频文件执行火焰识别,需要在detect.py中传入视频文件的名称或路径,然后处理好的视频会生成在runs/detect目录下。自己先把环境配置好,网上都有教程,如果载入训练模型出现bug,给我私信即可。
2023-05-15
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人