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 安装 coco 助手

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

1 个答案: