重用一个模型，对19个类进行预培训，只用于其中的一个类

Question

我有一个预先训练过的语义分割网络，它已经在城市景观数据集及其19个类(人、车、交通标志、…)上接受了培训。。其中之一就是“人”。我只对这门课感兴趣。

对同一数据集而不是19个类重新训练模型是否有益？

如果是，什么是好策略？保留特征提取器的权重，重新训练最后的分类层(S)？还是调好所有的重量？

提前感谢！

Answer 1

如果您已经为多个类训练了一个语义分割模型(在您的例子中是19个类)，并且您只想对一个类( Person类)进行预测。在这种情况下，基本上是以(B，C，H，W)的形式得到4D输出，其中B是批大小，C是类的数目，H和W分别是高度/宽度。在您的情况下，您应该考虑C到2，也就是说，除Person类以外的值应该替换为0，而person类应该用1替换，这样您就可以只对person类进行预测，甚至不保留模型。