Google Colab으로 Yolov5를 활용한 드론 데이터셋 학습

데이터셋 만들기

https://app.roboflow.com/login

이 사이트에서는 사용자 데이터를 업로드하여 bounding box로 라벨링을 할 수 있음.

다양한 드론 사진을 준비하여 드론 데이터셋을 만들기

• Training(학습) Set모델 학습에 사용되는 데이터. 이를 이용해 서로 다른 epoch로 학습시킴.
• Validation(검증) Set학습이 이미 완료된 모델을 검증하기 위한 데이터셋
• Test(평가) Set검증이 완료된 모델의 성능을 평가하기 위한 데이터셋

6(TrainDataSet):2(ValidationDataSet):2(TestDataSet) 비율로 설정

모델을 학습하는데에는 오직 유일하게 Train dataset만 이용하며

Validation, Test DataSet은 학습한 모델을 검증하는데 사용 한다.

무료 drone dataset을 사용하였다. (드론은 생소하지않은 객체라 이미 많은 데이터셋이 구현되어있음)