python - 特征提取器是否在 Tensorflow 对象检测 API 中进行过训练？

Question

在 Tensorflow 对象检测 API 中，一个典型的神经网络将有 2 个组件：一个特征提取器，然后是一个神经网络，它使用特征提取的输出对图像进行进一步分类。

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

标签： pythontensorflowconv-neural-networkobject-detectionobject-detection-api

Answer 1

是的，它是默认训练的。您可以使用火车配置中的freeze_variables字段将其关闭。然而，TF 的人不鼓励这样做，因为他们发现这样做会产生更差的结果和相同的训练时间。有关详细信息，请参阅github 上的此线程。