Commit e8f6eb8b authored by Oleg Dzhimiev's avatar Oleg Dzhimiev


parent 56f982a2
......@@ -222,14 +222,14 @@ with tf.Session() as sess:
image_data[nimg] = None
# clean
Save MetaGraph to Saved_Model in *.pb (protocol buffer) format to
be able to use from Java
# force clean
shutil.rmtree(dirs['exportdir'], ignore_errors=True)
# save MetaGraph to Saved_Model as *.pb
builder = tf.saved_model.builder.SavedModelBuilder(dirs['exportdir'])
#builder.add_meta_graph_and_variables(sess,[tf.saved_model.tag_constants.SERVING]) # True = *.pbtxt, False = *.pb
if lf:
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment