python-3.x - 如何使用 MS-COCO 的类并使用 Mask RCNN 添加 2 个新类？

Question

我希望我的模型只检测 3 个类，即人、轮胎和道路。我正在使用已经在“人”类上训练过的预训练 MS COCO 权重。

如何使用这个已经存在的“人”类并进一步训练我的模型以适应 2 个新类，即“轮胎”和“道路”？

我已经设置NUM_CLASSES = 1 + 3（背景+对象）。

然而，我的数据集只有“轮胎”和“道路”的注释，因此这些是我添加使用的唯一类self.add_class(source_name, class_id, class_name)。