计算机

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

读完了,有什么想法?

阅读已保存的问答示例,并循着引用回到原文。

查看公开问答示例
— 本文完 — 返回文章列表