将未知标签转换为 Yolov5

Question

我拥有一个标签格式未知的图像数据集，即：

angry_actor_104.jpg 0 28 113 226 141 22.9362 0

表示图片如下：

image_name face_id_in_image face_box_top face_box_left face_box_right face_box_bottom face_box_cofidence expression_label

我的问题是：这个怎么转换成yolov5格式？

我一直在寻找这个很长一段时间，希望有人能提供帮助。 非常感谢您。

Answer 1

由于格式未知，您不太可能找到现有代码来完全处理转换，但我可以分享一些入门技巧。

1. 注释文件没有足够的信息来转换为 Yolo 格式。因为要转换为 Yolo，您还需要知道图像的尺寸。如果所有图像的尺寸都相同，那么它会更容易，但如果所有图像都不同，那么您将需要额外的代码来提取图像的尺寸。我将在下面解释原因。

2. 完成后，您需要像这样以特定的直接结构获取图像和标签，每张图像有一个 txt 文件：

   /images/actor1.jpg
   /images/actor2.jpg
   /labels/actor1.txt
   /labels/actor2.txt
   
   

3. 这是您要将注释文件放入的形状。

   face_id_in_image x_center_image y_center_image width height
   

   https://stackoverflow.com/a/66563144/5183735 此处对值的含义进行了清晰的描述。

   现在您需要做一些数学运算来计算这些值。

   • 宽度 = (face_box_right - face_box_left)/image_width
   • 高度 = (face_box_bottom - face_box_top)/image_height
   • x_center_image = face_box_left/image_width + (width/2)
   • y_center_image = face_box_top/image_height + (height/2)

我有一些代码可以帮助您阅读文本文件并在此处保存文本文件。 https://github.com/pylabel-project/pylabel/blob/main/pylabel/exporter.py 和 https://github.com/pylabel-project/pylabel/blob/main/pylabel/importer.py。

如果您能够分享您的确切文件，我也许可以找到一些转换它们的快捷方式。