我正在尝试使用Ultralytics 开源研究来训练自定义对象检测器。我在我们必须在此处生成 .yaml 文件的步骤中遇到了这个问题。这些标签名称的顺序应该是什么?它不像我们在 Tensorflow 中那样按字母顺序排列。我不希望我的模型在推理过程中贴错标签。
问问题
429 次
1 回答
2
顺序是任意的。你可以选择任何你想要的。相关部分是,在下一步中,您必须提供.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
。
于 2021-08-14T13:15:39.307 回答