sonfire186

sonfire186 / export_tf_model.py

Created September 24, 2019 04:10 — forked from zhanwenchen/export_tf_model.py

Minimal code to load a trained TensorFlow model from a checkpoint and export it with SavedModelBuilder

	import os
	import tensorflow as tf

	trained_checkpoint_prefix = 'checkpoints/dev'
	export_dir = os.path.join('models', '0') # IMPORTANT: each model folder must be named '0', '1', ... Otherwise it will fail!

	loaded_graph = tf.Graph()
	with tf.Session(graph=loaded_graph) as sess:
	# Restore from checkpoint
	loader = tf.train.import_meta_graph(trained_checkpoint_prefix + '.meta')