【tensorflow】命令行参数解析

169 阅读 0 评论 112 点赞

我是靠谱客的博主清脆煎蛋，这篇文章主要介绍【tensorflow】命令行参数解析，现在分享给大家，希望可以做个参考。

1. tf.app.flags，用于支持接受命令行传递参数

复制代码

1
2
3
4
5
6
7
8
9
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
调用参数：FLAGS.int_name = 14
python tt.py --str_name test_str --int_name 99 --bool_name True

使用argparse包

复制代码

1
2
3
4
5
6
7
parser = argparse.ArgumentParser()
parser.add_argument("--input_dir", help="path to folder containing images")
parser.add_argument("--mode", required=True, choices=["train", "test", "export"])
parser.add_argument("--output_dir", required=True, help="where to put output files")
parser.add_argument("--seed", type=int)
a = parser.parse_args()
调用参数a.mode = "train"