我可以将任何保存的模型(* .pb)与Estimators一起使用吗?

时间:2019-02-14 18:24:42

标签: python tensorflow tensorflow-estimator transfer-learning

我正在执行从图像分类模型到分割模型的转移学习,因此我需要加载经过训练的模型,访问某些图层输出,并向图形添加一些已知的操作。

为此,我采用了任何一种训练有素的模型,而不仅仅是训练为Estimator模型。

我能够在会话范围内处理使用tf.saved_model.load的预训练模型,但是,该方法不能使用Estimators(没有明确引用该会话)。因此,有没有办法加载任何模型并使用Estimators添加一些操作?

0 个答案:

没有答案