如何使用多个自定义数据集训练 Detectron2 模型

Question

我想通过注册多个数据集来训练 detectron2 模型

我已经从不同的任务中提取了我的注释，现在我有多个数据集，我们需要一起训练。

Datasets Folder
Task1 -- annotations.json
      -- image dir
Task2 -- annotations.json
      -- image dir
Task3 -- annotations.json
      -- image dir
Task4 -- annotations.json
      -- image dir

我的问题是我们是否可以使用多个数据集训练模型。我们可以注册多个 coco 实例进行训练吗？我想在 Task1 Task2 Task3 上训练我的模型并在 task4

上测试

from detectron2.data.datasets import register_coco_instances
register_coco_instances("train", {}," ./Task1/annotations.json", "./Task1/imagedir")
register_coco_instances("Test", {}, "./Task4/annotations.json", "./Task4/imagedir")

或者我需要合并所有的coco实例吗？

请提供您的意见

Answer 1

回答我自己的问题。显然，没有这种方法可以尝试多个数据集。如果您想合并不同的数据集，请使用 COCO Assitant 库。

简单的安装步骤是：

!pip install coco-assistant

如何使用多个自定义数据集训练 Detectron2 模型

How to train Detectron2 model with multiple custom dataset

python

object-detection

image-segmentation

multiple-instances

coco