如何通过LabelImg为YOLO v3创建自定义数据集

Question

我使用了LabelImg Save as YOLO选项将我的标签保存为.txt格式，格式如下

6 0.333984 0.585938 0.199219 0.160156

但我希望它是这样的格式

path/to/img1.jpg 50,100,150,200,0 30,50,200,120,3
path/to/img2.jpg 120,300,250,600,2

我该如何做到这一点？

Answer 1

YOLO使用相对像素值，而不是原始像素值。换句话说，格式是：

中心x中心y宽度高度

其中-x是宽度的中心百分比。换句话说，如果图像的宽度为800px，而中心x为400px，则中心x将写为0.5。

因此，您的LabellMg值对于训练YOLO来说已经是正确的。此外，在YOLO v3中，您实际上需要它们都是单独的.txt文件，而不是放在一个大而长的文件中。所以你已经可以走了。

Answer 2

不同意上面的答案。并不是所有的实现都需要宽度、中心或高度的百分比，并且Yolo I使用的实现需要单个train.txt文件。一个特定的数字，例如https://github.com/qqwweee/keras-yolo3，需要问题中提到的确切格式，但这4个数字是坐标右上角x，右上角y，右下角x，右下角y，后跟类号。不过，您可以使用这些文本文件，并在csv中将它们合并在一起，在列中也包含图像的名称。这可以使用glob或pandas库来完成。您可以一次在csv中计算整个列的宽度和高度。然后立即将路径添加到complete列，并将其转换为文本文件，即可输入。