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

Question

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

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

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

我的问题是我们是否可以用多个数据集训练 model。 我们可以注册多个 coco 实例进行训练吗？ 我想在 Task1 Task2 Task3 上训练我的 model 并在 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

Answer 2

实际上它可以使用多个集合进行训练并将它们合并，如GitHub 问题 #2544中所述

register_coco_instances("my_trainsetA", {}, "train/_annotations.coco.json", "trainA")
...

cfg.DATASETS.TRAIN = ("my_trainsetA", "my_trainsetB", ... )
cfg.DATASETS.TEST = ("my_testsetA", ...)