학습용 데이터셋 만들기

YOLOv8 모델을 개발하려고하는데 내가 의도한 사물을 인식하기 위해서는 커스텀 트레이닝을 수행해야 한다.

기존 모델에 추가 학습을 시키는 전이학습? 같은걸 해야하는데, 먼저 학습시킬 데이터가 필요하다.

그런데 기존 데이터셋을 보면 유명한 사이트에서 다운로드 받는 방식으로 만들어 내는것 같다.

이렇게 되면 내가 원하는 객체에 대한 학습을 하려니 내가 원하는 객체를 갖고있는 데이터셋이 없기 때문에 결국 내가 직접 만들어야 하는 상황이 생긴다.

 

그렇다면, 내가 데이터셋을 만들면 되지 않을까? 

YOLO에서 활용되고 있는 데이터셋을 다운로드 받고 그 구조를 분석해보자.

 

그리고 데이터셋을 만들어주는 어플리케이션앱을 개발하자.