使用新类对预训练模型进行训练

Question

我可以在maskrcnn预训练模型的基础上进行训练吗？如果预训练模型的类别与我自己的数据集的类别不同，预训练模型是否仍然有效？

我尝试使用新的班级训练数据集。但似乎收敛得很慢。

Answer 1

通常，在您自己的数据集上微调预训练模型是一种行之有效的做法。通常将模型的头部替换为您自己的头部，该头部将与数据集中的类数量相匹配，并且首先具有随机初始化的权重。