如何使用保存的模型加载器和tf.estimator for VGG16

时间:2018-05-30 22:36:47

标签: tensorflow machine-learning neural-network conv-neural-network tensorflow-estimator

我正在寻求实施FCN8,遵循本教程: https://medium.com/nanonets/how-to-do-image-segmentation-using-deep-learning-c673cc5862ef

然而,我的代码结构不同,因为它是更大项目的一部分。我正在使用Estimator类。

我的问题是:我如何在Estimator类中执行此代码的操作?我可以使用saved_model.loader.load在我的model_fn中为我的Estimator加载VGG16吗?或者还有其他方法可以做到这一点吗?

谢谢!

0 个答案:

没有答案