Tensorflow目标检测API中是否训练特征提取器？

Question

3

在Tensorflow目标检测API中，典型的神经网络包括两个组成部分：特征提取器和利用特征提取器输出进一步对图像进行分类的神经网络。

如果您在对象检测API中自定义训练更快的resnet50神经网络以检测额外的2个对象，在训练过程中，特征提取器是否也会被训练？即特征提取器组件的权重是否会改变？

- Tristan Berry

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- gdelab · Accepted Answer

是的，它默认进行训练。您可以在训练配置中使用freeze_variables字段关闭它。但是，TensorFlow的开发人员不建议这样做，因为他们发现这会导致更差的结果和相同的训练时间。有关更多详细信息，请参见Github上的此线程。