argparse 是python自带的命令行参数解析包,可以用来方便地读取命令行参数。
import argparse def parse_args(): parser = argparse.ArgumentParser() parser.add_argument('-t', '--tensorflow-data-dir', default='pic/') parser.add_argument('--train-shards', default=2, type=int) parser.add_argument('--validation-shards', default=2, type=int) parser.add_argument('--num-threads', default=2, type=int) parser.add_argument('--dataset-name', default='satellite', type=str) args = parser.parse_args() return args
import argparse :导入包。
parser = argparse.ArgumentParser() :生成一个参数解析器。
parser.add_argument() :增加参数,'-'、'--' 指定参数名称,default 是未指定该参数时的默认值。
args = parser.parse_args() :用来解析参数,可以使用args.train-shards获取参数值。