如何创建TensorFlow模型的检查点文件?

时间:2019-05-14 16:46:35

标签: python tensorflow neural-network deep-learning

我正在尝试使用TensorFlow实施神经样式转换,并希望将其集成到Android应用程序中以进行图像过滤。任何人都可以帮助我保存此TensorFlow模型,以便在应用程序中使用该模型时将其还原吗?

我尝试创建此TensorFlow模型的.ckpt文件,但是无法创建一个。

我没有在这里复制代码,因为它包含数十行。这是模型https://colab.research.google.com/github/tensorflow/models/blob/master/research/nst_blogpost/4_Neural_Style_Transfer_with_Eager_Execution.ipynb

的链接

我希望保存此模型以极低的延迟生成新图像。目前,我必须为每个图像运行数百次迭代,而这是我在应用程序中运行时不希望的实时结果。

0 个答案:

没有答案