在终端运行python脚本的时候经常需要用到argparse进行参数交互,通过argparse在终端输入脚本所需参数运行程序,尤其在服务器上进行训练的时候,只能通过命令进行操作,用的会更多,目前我主要用到了文件地址这个参数,记录一下,后面用到其他的再来补充:
1.首先建立函数
导入argparse
库,parser.add_argument
添加参数,第一个引号是参数名,type是参数类型,help是参数提示,要多少个参数就写多少次.
import argparse
def get_parser():
parser = argparse.ArgumentParser(description='bag path input')
parser.add_argument('--base_path1', type=str, help='first file path')
parser.add_argument('--base_path2', type=str, help='second file path')
parser.add_argument('--base_path3', type=str, help='third file path')
parser.add_argument('--base_path4', type=str, help='tranform file path')
parser.add_argument('--save_path', type=str, help='csv save path')
args = parser.parse_args()
return args
2.主函数调用
if __name__=='__main__':
args=get_parser()
base_list = [args.base_path1,args.base_path2,args.base_path3,args.base_path4]#
save_path=args.save_path
3.终端输入命令运行程序:
python ./label_create.py --base_path1 (文件路径) --base_path2 (文件路径) --base_path3 (文件路径) --base_path4 (文件路径) --save_path (文件路径)