本文主要是简述如何训练自己的数据集,这里以FLIR的红外数据集为例,配置运行以及原理不再叙述
1.修改Makefile
这步一般已经在配置运行的时候已经做过了,没什么大问题
2.将自己的数据集处理成yolo训练所需的数据集格式
FLIR官方开源的数据集结构如下:
这里的标签文件是一个json文件,而yolo需要的标签文件是txt文件,处理代码链接:https://blog.csdn.net/hello_levy/article/details/105212876
处理后的标签如下,第一个整数代表的是类别,我这里2代表的是car,后面是目标框的中心点和长宽归一化后得到的坐标
每张图片对应一个txt文件: