python - 创建MS COCO样式数据集

如何创建与TensorFlow一起使用的MS COCO style数据集？有人对此有经验吗？我有图像，注释和地面真相蒙版。我需要将它们转换为与MS COCO兼容，并感谢您的帮助。我找不到任何开源工具来创建COCO风格的JSON注释。

TensorFlow MS COCO读取我不太有经验的JSON文件。

最佳答案

我正在使用一个python库，该库具有许多有用的类和函数来执行此操作。它称为Image Semantics。

这是添加 mask 并将其以COCO格式导出的示例:

from imantics import Mask, Image, Category

image = Image.from_path('path/to/image.png')
mask = Mask(mask_array)
image.add(mask, category=Category("Category Name"))

# dict of coco
coco_json = image.export(style='coco')
# Saves to file
image.save('coco/annotation.json', style='coco')