[YOLO] 커스텀 데이터셋 학습시키기 - roboflow에서 데이터셋 만들기

 

봉식이를 인식하는 모델을 만들어보기로 했다.

 

다른 고양이들은 cat으로 인식하고 봉식이는 bongsik으로 인식했으면 좋겠다는 뭐 그런 목표로...

 

일단 학습시킬 데이터셋이 필요하다. 당연하게도... 사진만 있으면 안되고 바운딩 박스를 직접 그려주어야하는데

 

https://app.roboflow.com/

Sign in to Roboflow

 

위 사이트를 이용했다.

 

※ workspace를 만들 때 public plan으로 해야 나중에 export가 가능하다... 나도 이걸 알고싶지 않았다 ㅎ

 

 

 

귀여운 우리 봉시기

 

 

정말이지 🐶노가다 작업인데 봉식이 사진이라 그나마 괜찮았다...

 

 

아무리 집사라도 매일 사진을 찍는건 아님 + 바운딩 박스 그리는거 좀 귀찮음 이라는 이유로 사진은 400개 가량 업로드 하였고 증강 기능을 이용하여 이미지 갯수를 늘려보았다.

 

 

 

무료 옵션도 많다.

 

 

이렇게 예시도 보여준다. 

 

 

 

봉식이 이미지를 972개로 만들어보쟈

 

쨘

 

 

 

이렇게 다양한 포맷으로 export가 가능하다.

 

 

 

코랩에서 돌리기 위해 zip형식으로 받아서 구글 드라이브에 넣었다..

 

 

이제 yolo를 학습시켜야한다..... ㅜ