1、argparse
argparse是python的命令行解析库
import argparse
def parse_arguments():
# 创建解析器
parser = argparse.ArgumentParser()
# 第一个是参数名称,第二个参数是类型,第三个参数是默认值,第四个是参数描述
parser.add_argument('--logs_base_dir', type=str,help='Directory where to write event logs.', default='~/logs/detailed_graph')
return parser.parse_args()
def main(args):
path = args.logs_base_dir
if __name__ == '__main__':
main(parse_arguments())
2、tf.app.flags.FLAGS-接受命令行传递参数
此方法的底层函数就是Python argparse模块
import tensorflow as tf
# 第一个是参数名称,第二个参数是默认值,第三个是参数描述
tf.app.flags.DEFINE_string('str_name', 'def_v_1', "descrip1")
tf.app.flags.DEFINE_integer('int_name', 10, "descript2")
tf.app.flags.DEFINE_boolean('bool_name', False, "descript3")
FLAGS = tf.app.flags.FLAGS
def main(args):
print(FLAGS.str_name)
print(FLAGS.int_name)
print(FLAGS.bool_name)
if __name__ == '__main__':
tf.app.run() # 执行main函数