tensorflow - 如何将保存的模型加载器与 tf.estimator 一起用于 VGG16

Question

我希望按照本教程实现 FCN8： https ://medium.com/nanonets/how-to-do-image-segmentation-using-deep-learning-c673cc5862ef

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

我的问题是：我该如何做这段代码在 Estimator 类中所做的事情？我可以使用 saved_model.loader.load 在我的 Estimator 的 model_fn 中加载 VGG16 吗？还是有其他方法可以做到这一点？

谢谢！

标签： tensorflowmachine-learningneural-networkconv-neural-networktensorflow-estimator