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

Question

我正在尝试使用 Detectron2 训练模型。我正在使用 Grocery 图像数据，并且有 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")

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