View gist:5aec5b55e5965feb20bf5ecd1feb6c65
import argparse
import os
parser = argparse.ArgumentParser(description='Generate a saved model.')
parser.add_argument('--export_model_dir', type=str, default='./saved_model/dense-lite', help='export model directory')
parser.add_argument('--model_version', type=int, default=1, help='model version')
parser.add_argument('--model', type=str, default='dense.pb', help='model pb file')
parser.add_argument("--input_tensor", default="image:0", help="input tensor", type=str)
parser.add_argument("--output_tensor", default="Openpose/concat_stage7:0", help="output tensor", type=str)
args = parser.parse_args()