如何使用 MS-COCO 的一个类并使用 Mask RCNN 添加 2 个新类？

Question

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

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

我已经设置了

NUM_CLASSES = 1 + 3

（背景+对象）。

然而，我的数据集只有“轮胎”和“道路”的注释，因此这些是我使用

self.add_class(source_name, class_id, class_name)

添加的唯一类。

Answer 1

我也面临同样的问题，请问您解决了吗