我试图使用Ultralytics的开源研究来训练一个自定义对象检测器。我在这里必须生成.yaml文件的步骤中遇到了这个问题。这些标签名称的顺序应该是什么?它不像我们在Tensorflow中那样按字母顺序排列。我不希望我的模型在推理过程中错误标记。
顺序任意。你可以选择任何你想要的。相关的部分是，在下一步中，您必须为每个图像提供一个.txt，其中:
每行为
class x_center y_center width height格式
在本例中，class将是0到N-1之间的整数，其中N是您在.yaml文件中定义的类的数量。
那么，如果在.yaml文件中你有:
nc: 3
classes: ['cat', 'dog', 'car']
在my_image.txt中你有:
0 0.156 0.321 0.254 0.198
2 0.574 0.687 0.115 0.301
那么，这意味着，在这个图像中，您有一个cat和一个car。