python - YoloV5 中使用 Ultralytics 代码的类标签顺序

Question

我正在尝试使用Ultralytics 开源研究来训练自定义对象检测器。我在我们必须在此处生成 .yaml 文件的步骤中遇到了这个问题。这些标签名称的顺序应该是什么？它不像我们在 Tensorflow 中那样按字母顺序排列。我不希望我的模型在推理过程中贴错标签。

标签： pythonpytorchyoloyolov5

Answer 1

顺序是任意的。你可以选择任何你想要的。相关部分是，在下一步中，您必须提供.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。