argparse 库
基本用法
使用 ArgumentParser 创建一个对象,使用 add_argument 添加命令行参数,然后最后用 .parse_args() 方法返回一个参数的字典。
parser = argparse.ArgumentParser(
prog="Image Format Convert",
usage="python convert.py -i INPUT_DIR -f OUTPUT_FORMAT [-C]"
)
parser.add_argument("-i", "--input_dir", required=True, type=str,
help="The directory of input images")
parser.add_argument("-f", "--fmt", type=str, default="png",
help="Output format (default png)")
parser.add_argument("-C", "--clip", action="store_true",
help="Whether to clip")
args = parser.parse_args()
main(args.input_dir, args.fmt, args.clip)
然后在命令行中传入参数:
python test.py -i ./img -f png -C
最终所有的参数都储存在 args 变量中,使用对应的参数名(如 args.input_dir)就可以获得参数值。
store_const
使用 store_const 的两个变种 store_true 和 store_false,可以在参数出现时,将该参数的值记录为 true 或 false。例如:
parser.add_argument("--nnunet", action="store_true")
则在命令行出输入 python example.py --nnunet 时,parser 返回的结果会将 --nnunet 记录为 True,如果不出现则记为 False。
BEYOND THE ARTICLE
读完了,有什么想法?
阅读已保存的问答示例,并循着引用回到原文。
— 本文完 — 返回文章列表