Tensorflow对象检测API是否训练了特征提取器?

时间:2018-07-31 01:14:20

标签: python tensorflow conv-neural-network object-detection object-detection-api

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

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

1 个答案:

答案 0 :(得分:4)

是的,默认情况下经过培训。您可以使用火车配置中的 freeze_variables 字段将其关闭。但是,TF人员不鼓励这样做,因为他们发现这样做会产生较差的结果,并且训练时间相同。有关更多详细信息,请参见this thread on github