Pychrm带参启动两种方式

if __name__ == '__main__':
    '''
    main function
    '''
    import argparse

    parser = argparse.ArgumentParser()
    parser.add_argument("--input_dir", type=str)
    args = parser.parse_args()

    get_model()
    print(args.input_dir)
    pred = multi_lung_disease_detector(args.input_dir)

    print("case: {}, pred: {}".format(os.path.basename(args.input_dir), pred))

 

以上代码中有参数，直接赋值是不可以的

那么运行可以通过以下方式：

方式一：

python test.py --input_dir .\demo_data\xxxxxxx.png

方式二：