[TIL] YOLO dataloader 구조 변경

YOLO v7 MS COCO 구조로 dataloader 변경하기

배경(2022.08.30)

• YOLO논문에서 bbox의 위치정보는 bbox의 중심 좌표(x, y)를 기준으로 너비와 높이를 상대 좌표로 변환 후 정규화 -> 처리 속도를 빠르게 하였음
• 현재 회사에서는 COCO 포멧으로 데이터 라벨링 작업을 함
• 기존 사용했던 YOLOX은 COCO 포멧을 그대로 사용하여 데이터 로딩하고 있음 참고
• YOLO v7이 7월에 출시되고, YOLOX에 비하여 성능이 확연하게 다르기 때문에 개발하고 있는 모델을 YOLO v7로 고도화가 필요함
  
• YOLO v7은 기존 YOLO 포멧을 따름
  • Helpful URL : https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data
• 대부분 COCO 포멧을 covert2Yolo를 활용하여 YOLO 포멧으로 변경후 학습을 진행함
• 앞으로 데이터셋은 계속해서 추가될 예정이고, 그 때마다 covert2Yolo를 활용하기엔 귀찮기 때문에 YOLO v7 dataloader를 변경할 예정