sumfu的博客

原创 基于YOLOv5的火焰图像识别

1.笔者基于pytorch库，采用开源的YOLOv5框架训练出了基于深度学习的能通过计算机视觉来实现目标检测功能的火焰识别模块。进而，笔者便可以使用Pycharm这款IDE来借助pytorch环境进行开发。然后创建数据集文件，创建class.txt，里面输入fire类，导入要标注的图片文件，从cmd进入目录后使用命令。labelimg是一款数据集标注工具，可以将标注的数据保存成不同格式。通过cmd执行命令来安装。来调用labelimg工具标注数据集。3即可载入模型进行训练。