如何注册要与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 :)