test.txt
-
전처리) train_list.와 val_list 파일 나누기deep learning/코딩 2021. 12. 2. 21:15
그동안 Yolo와 같은 모델을 통해 학습하기 위해 정말 많은 전처리를 하게 되었다. 그 중엔 xml 파일을 yolo 스타일의 txt 파일로 변경하거나 json 파일을 똑같이 yolo txt 스타일로 변경하는 일을 자주 하게 되었다. 그동안 한번 코드를 짜둔 후 계속 복사해서 사용하다 보니 꽤나 불편하고 오랜만에 보면 기억도 잘 안나서 블로그에 정리해 두고 깃허브에 코드를 올려둘 계획이다. 아직 yolov5 이하의 모델을 사용해본적이 없어서 다른 모델은 모르겠지만 학습 중에 학습을 하고 그 것을 검증할 데이터 셋을 나눠야 한다. 아래 코드는 그 코드이다. from glob import glob path = "학습을 돌릴 path" #학습을 돌릴 path => train.txt와 val.txt를 저장할 pa..