如何将Yolo注释转换为可可格式。 json？

Question

我想以yolo格式将标签转换为可可格式 我尝试过 https://github.com/github.com/taeyoung96/yolo-to-coco-format-to-coco-format--coco-format--coco-format--转换器 和 Pylabel 他们都有一个错误。

我想在检测2上训练，但是由于错误的JSON文件，它无法加载数据集。

Answer 1

您可以使用 globox 包（免责声明：我是作者）。这应该类似于：

from pathlib import Path

import globox


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

  annotations = globox.AnnotationSet.from_yolo(gts_path)
  annotations.save_coco(save_file)


if __name__ == "__main__":
    main()

如果您需要使用 .name 文件映射标签：

names_file = Path("/path/to/classes.names")
names = AnnotationSet.parse_names_file(names_file)
annotations.map_labels(names)

您可以查看其他格式的文档（Yolo Darknet、Yolo v8 等）。