yolov5
-
Labelling Tool) labelImg 사용법deep learning/프로젝트 2021. 12. 21. 01:09
우리가 지도학습을 하기 위해서는 이미지와 이에 해당하는 label들이 필요하다. 최근에는 labelling도 딥러닝으로 해주는 auto labelling이나 active learning에 관한 연구들도 많이 진행되고 있지만, 아직 나는 그정도 수준이 안되서 이번에 라벨링하는 툴의 사용법을 알아보고 추후에 auto labelling을 해볼 수 있도록 해봐야 겠다! 그래서 labelimg 프로그램은 https://github.com/tzutalin/labelImg 여기 github에 사용법이 업로드 되어 있다. Installation pip3 install labelImg labelImg Ubuntu Linux Python 3 + Qt5 git clone https://github.com/heartexlab..
-
전처리) 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..