import argparse

def get_parser():

    # argparse.ArgumentParser生成argparse對(duì)象 description為描述信息，當(dāng)在命令行輸入需要顯示幫助信息時(shí)，會(huì)顯示

    parser = argparse.ArgumentParser(description="pytorch unet training")

    # 路徑參數(shù)設(shè)置 help為參數(shù)的幫助信息

    parser.add_argument("--data_path", default="./", help="DRIVE root")

    # 預(yù)測(cè)類別數(shù)量 type如果不指定需要輸入的是str類型

    parser.add_argument("--num_classes", default=1, type=int)

    # 指定設(shè)備使用

    parser.add_argument("--device", default="cuda", help="training device")

    # 指定batch size大小 "-b", "--batch_size"表示兩個(gè)都可以在命令行使用

    parser.add_argument("-b", "--batch_size", default=4, type=int)

    return parser

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print(args)

import argparse

def get_parser():

    # argparse.ArgumentParser生成argparse對(duì)象 description為描述信息，當(dāng)在命令行輸入需要顯示幫助信息時(shí)，會(huì)顯示

    parser = argparse.ArgumentParser(description="pytorch unet training")

    # 路徑參數(shù)設(shè)置 help為參數(shù)的幫助信息 default為默認(rèn)參數(shù)

    parser.add_argument("--data_path", default="./", help="DRIVE root")

    # 預(yù)測(cè)類別數(shù)量 type如果不指定需要輸入的是str類型

    parser.add_argument("--num_classes", default=1, type=int)

    # 指定設(shè)備使用

    parser.add_argument("--device", default="cuda", help="training device")

    # 指定batch size大小 "-b", "--batch_size"表示兩個(gè)都可以在命令行使用

    parser.add_argument("-b", "--batch_size", default=4, type=int)

    return parser

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("data_path: ",args.data_path)

    print("num_classes: ", args.num_classes)

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("-b: ",args.b)

    print("--batch_size: ", args.batch_size)

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("--batch_size: ",args.batch_size)

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("--batch_size type: ",type(args.batch_size))

parser.add_argument("--num_classes", default=1, type=int, required=True)

parser.add_argument('-arch', required=True, choices=['alexnet', 'vgg'])