之前用过两种方式获取终端启动项目的参数
1、sys.argv以列表的形式存在
import sys
files = sys.argv[1:]
2、argparse库
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--trainroot', required=True, help='path to dataset')
parser.add_argument('--valroot', required=True, help='path to dataset')
parser.add_argument('--cuda', action='store_true', help='enables cuda')
opt = parser.parse_args()
root=opt.valroot # 获取--valroot后输入的内容
执行命令:
mac,windows:
python tmp.py --trainroot a\train --valroot a\val --cuda
linux:
source python tmp.py --trainroot a\train --valroot a\val --cuda
有其他方法欢迎评论区交流~
网友评论