Python argparse 사용하기! (터미널에서 python을 실행할 때, 옵션을 쉽게주자)

import argparse

def arg_test(opt):
    p1 = opt.test_a
    p2 = opt.test_b
    p3 = opt.test_c

    print(p1,p2,p3)

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--test-a', type=str, default='good1', help='argument 1')
    parser.add_argument('--test-b', type=str, default='good2', help='argument 2')
    parser.add_argument('--test-c', type=str, default='good3', help='argument 3')


    opt = parser.parse_args()

    print(opt)

    arg_test(opt)

간단하게 코드로 설명하겠습니다.

 

parser라는 변수에 argparse.ArgumentParser()를 정의해주고

parser.add_argument를 이용해서 어떠한 이름으로 argument를 받을지 정의하고, default 값도 정의할 수 있습니다.

help를 이용해서 어떠한 값을 넣어야하는지도 설명을 할 수 있습니다.

 

pycharm에 terminal을 이용해서 실행을 한 결과

argument에 아무것도 안줬어도 default 값을 이미 정의해놨기 때문에 오류없이 출력을 했습니다.

 

 

이번엔 argument를 던져주겠습니다.

argument가 잘 입력됐습니다.

yolov5에서 이런식으로 학습을 진행해서 궁금해서 찾아봤습니다.

 

요새 회사에서 Data pipe line을 만들고 있어서, 관심있게 보고있네요.

감사합니다.