argparse 用法

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获取参数值。