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