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

问题描述

我正在尝试使用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}") */

此代码是否有问题？如果您还有其他建议，请帮助我。

解决方法

我觉得这对你有帮助

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 :)