1

我想通过注册多个数据集来训练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 实例！

请提供您的意见

4

1 回答 1

2

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

安装的简单步骤是：

!pip install coco-assistant
于 2021-05-19T11:08:28.683 回答