如何将yolo注解转成coco格式。 Json？

Question

我想将我的 yolo 格式标签转换为 coco 格式我试过了 https://github.com/Taeyoung96/Yolo-to-COCO-format-converter 和标签他们都有一个错误。

我想在 detectron 2 上训练，但由于 json 文件错误，它无法加载数据集。

谢谢大家

Answer 1

你能用这个 tool 试试吗（免责声明：我是作者）？它（还）不是 Python 包，所以你需要先下载 repo。这应该类似于：

from ObjectDetectionEval import *
from pathlib import Path


def main() -> None:
  path = Path("/path/to/annotations/")  # Where the .txt files are
  names_file = Path("/path/to/classes.names")
  save_file = Path("coco.json")

  annotations = AnnotationSet.from_yolo(gts_path).map_labels(names)
  
  # If you need to change the labels
  # names = Annotation.parse_names_file(names_file)
  # annotations.map_labels(names)

  annotations.save_coco(save_file)


if __name__ == "__main__":
    main()

如果您需要更多控制（坐标格式、图像位置和扩展名等），您应该使用更通用的 AnnotationSet.from_txt()。如果它不适合您的需要，您可以使用 AnnotationSet.from_folder().

轻松实现自己的解析器

如何将yolo注解转成coco格式。 Json？

How to convert yolo annotations to coco format. Json?

image

object-detection

annotation-processing

data-annotations

deep-learning