Tensorflow:将预训练的模型权重加载到不同的架构中

时间:2019-07-09 09:43:21

标签: tensorflow pre-trained-model

我有一个Tensorflow模型,可以很好地用于检测图像中的对象并生成边界矩形。输出包括一个Softmax和四个用于定位的模拟值。我需要再添加一个模拟输出来预测对象方向。如何导入预训练的模型权重并冻结它们,以便仅对处理最后一层方向的零件进行训练。

0 个答案:

没有答案