[YOLOv5] 커스텀 데이터 학습 - 데이터셋 만들기 (2)

Roboflow

직접 데이터셋을 만들기 위해 무료로 바운딩박스툴을 제공해 주는 roboflow를 사용하였다. roboflow 링크: https://roboflow.com/

---

데이터셋 사진 준비하기

나는 횡단보도 데이터셋을 만들고 싶었기 때문에 횡단보도 사진을 준비했다. 주변에 있는 횡단보도를 찍었다.

Roboflow에서 이미지들을 업로드하고 바운딩박스를 만들어 준다.

완료했다면 데이터셋으로 만들 수 있다.

만들기 전에 여러 증강 옵션을 넣으면 총 이미지 개수가 늘어난다. 하지만 항목을 모두 체크한다고 좋은 게 아니다... 처음에 이미지 개수를 늘이려고 모두 체크했다가 이미지들이 이상하게 변해버려서 YOLOv5 학습이 제대로 되지 않았다... 아래 사진에 나와있는 옵션 정도만 추가하는 것이 좋다.

이제 데이터셋을 만들어주면 저번 [YOLOv5 커스텀 데이터 학습 - Google Colab (1)] 포스팅에서 사용한 권총 데이터셋처럼 사용할 수 있게 된다. 여담으로 처음에 증강 옵션을 너무 많이 선택하는 바람에 v3를 다시 만들었다;;

---

Roboflow 사이트를 활용하여 나만의 횡단보도 데이터셋을 만들어 보았다. 처음부터 끝까지 거의 다 알아서 해줘서 편했다. 다만 public으로 설정을 해줘야 export가 가능하기 때문에 데이터셋을 다른 사람도 사용할 수 있는 것 같다. 다음 포스팅에서는 직접 만든 데이터셋을 로컬 환경에서 학습시켜 볼 예정이다.