1.笔者基于pytorch库,采用开源的YOLOv5框架训练出了基于深度学习的能通过计算机视觉来实现目标检测功能的火焰识别模块。进而,笔者便可以使用Pycharm这款IDE来借助pytorch环境进行开发。
2.数据集标注
labelimg是一款数据集标注工具,可以将标注的数据保存成不同格式。通过cmd执行命令来安装
pip install labelimg -i https://pypi.tuna.tsinghua.edu.cn/simple
然后创建数据集文件,创建class.txt,里面输入fire类&