[YOLOv8] 전동 킥보드 헬멧 객체 탐지, helmet detection project1

팀 주제 : AI 기술을 이용한 스마트 전동킥보드 안전 시스템

 

3월 초에 주제를 정한 후, 3월 4월에 걸쳐 한창 문서 작업을 하다가 팀원들 각자 하나의 기능을 맡아 구현하기로 했다.

내가 맡은 기능은 전동 킥보드 이용자가 헬멧을 착용하였는지 탐지하고, 2인 이상 탑승하였는지 감지하는 객체 탐지이다.

우선 헬멧 착용을 감지하는 ai 학습을 진행할 예정이다.

객체 탐지와 관련된 논문을 읽어본 결과 YOLO가 가장 객체 탐지 정확도를 높이는 모델이라 하여서 YOLO를 AI 학습 모델로 사용하기로 했다. YOLO도 버전이 많은데, 그 중 가장 최신에 나오기도 하고, 객체 탐지가 잘된다고 하는 YOLOv8를 사용할거다.

 

DAY1. 2024/05/15

Target : contain the dataset of 'bike helmet' and educate yolo basic the helmet dataset.

목표 : 전동 킥보드는 법률적으로 헬멧이 필수이다. 전동 킥보드에 캠을 달아 사용자가 전동킥보드를 이용하는 동안 헬멧을 착용하고 있는지 파악하기 위해 YOLOv8를 이용해 헬멧 감지를 할 예정이다. 일단 자전거 헬멧 데이터셋을 찾아보고자 한다.

 

 

- using dataset : kaggle, robotflow

AI학습을 위한 대량 데이터셋을 찾기 위해 무료 데이터셋을 제공하는 사이트를 찾았다.

일단 유명한 오픈 데이터셋인 AI Hub와 google에서 제공하는 데이터셋을 찾아보았다.

하지만 아쉽게도 헬멧 관련한 데이터셋을 찾을 수 없었고, 구글로 서칭을 통해 데이터셋을 찾았다.

구글 검색어에 'bike helmet detection dataset', 'bicycle helmet dataset' 등 헬멧과 관련된 키워드를 검색했다.

그렇게 찾은 사이트는 kaggle과 roboflow라는 사이트이다.

인공지능 학습으로 프로젝트를 하는 것은 처음이라 kaggle과 roboflow는 처음 알게 되어,

해당 사이트가 오픈 데이터셋을 제공하는지, 이용 방법은 어떻게 되는지 찾아보았다.

- Kaggle

Kaggle is a platform for data science competitions, where data scientists and machine learning engineers can compete with each other to create the best models for solving specific problems or analyzing certain data sets.

 

- Roboflow

Roboflow empowers developers to build their own computer vision applications, no matter their skillset or experience. We streamline the process between labeling your data and training your model. After building our own applications, we learned firsthand how tedious it can be to train and deploy a computer vision model.

 

찾아본 결과, 위의 설명과 같이 모두 무료로 데이터셋을 제공한다는 것을 알았다.

roboflow와 kaggle에서 자전거 헬멧 관련 데이터셋을 찾았고,

Roboflow의 데이터의 양질이 더 좋은 것 같다고 판단하여 Roboflow 데이터셋을 먼저 이용하기로 했다.

 

- robotflow dataset link (first I used)

https://universe.roboflow.com/bike-helmets/bike-helmet-detection-2vdjo

 

- kaggle dataset link

https://www.kaggle.com/datasets/moazzimalibhatti/helmet-wearing-persons

https://www.kaggle.com/datasets/aneesarom/rider-with-helmet-without-helmet-number-plate

https://www.kaggle.com/datasets/brendan45774/bike-helmets-detection

 

 

아래의 사이트는 YOLOv8 학습 방법이 나와있다.

- YOLO v8 (Follow the explaination)

https://docs.ultralytics.com/#where-to-start

https://blog.roboflow.com/how-to-train-yolov8-on-a-custom-dataset/