#!/usr/bin/env python

import argparse
import tensorflow as tf

parser = argparse.ArgumentParser()
parser.add_argument('--meta', help='metagraph file')
parser.add_argument('--model', help='model file')
args = parser.parse_args()

assert args.meta and args.model, "Metagraph/model needed"

with tf.Session() as sess:
  new_saver = tf.train.import_meta_graph(args.meta)
  new_saver.restore(sess, args.model)
  new_saver.export_meta_graph('tmp.pb')