json - 如何注册要与detectron2 一起使用的数据集？我们有 COCO JSON 格式的图像及其注释

Question

我正在尝试使用 Detectron2 训练模型。我正在使用杂货店图像数据，并且有 COCO 格式的注释。我在加载模型时遇到问题。模型不带注释。我指的是这个博客https://gilberttanner.com/blog/detectron2-train-a-instance-segmentation-model。

注册数据集时面临问题。

from detectron2.data.datasets import register_coco_instances

for d in ["train", "test"]:
    register_coco_instances(f"microcontroller_{d}", {}, f"Microcontroller Segmentation/{d}.json", f"Microcontroller Segmentation/{d}")

这段代码有问题吗？

Answer 1

我认为这可能会帮助你

from detectron2.data.datasets import register_coco_instances
register_coco_instances("YourTrainDatasetName", {},"path to train.json", "path to train image folder")
register_coco_instances("YourTestDatasetName", {}, "path to test.json", "path to test image folder")

让我知道它是否适合你。我已经用这个训练了detectron2 :)