[객체탐지] YOLOv5- 전동킥보드 객체 탐지 해보기

YOLO를 이용해서 계속해서 프로젝트를 진행 중이다. 

 

진행 상황을 정리할 겸 원본 영상과, 학습시킨 모델로 전동 킥보드를 인식시켰을 때 어떻게 인식되는지 보여주겠다! 

 

(사실 지금은 12월이지만 11월에 써놨던 글을 포스팅 버튼을 안 눌렀어서 지금 올린다...ㅎㅎ 지금은 더 진행되긴 했다^^) 

 

---

 

🚀 내가 진행 중인 프로젝트 

나는 지금 도로 위에 방치된 전동킥보드들을 자동적으로 위치를 파악해서 도로 안전에 위협이 되는 전동킥보드를 회수하는 프로젝트 설계를 진행 중이다. 

그러기 위해서 

• 자동차 혹은 버스의 블랙박스 영상 만으로 전동킥보드를 자동으로 탐지해야 한다. 

YOLO 모델은 이미 학습을 시켰고, 내가 직접 촬영해본 영상을 이 모델로 탐지해봐서 전동킥보드를 찾아내는지 확인해봐야겠다. 

---

🚀 원본 영상 

먼저 원본 영상이다. 학교 앞 인도에서 전동 킥보드를 쉽게 볼 수 있다. 

걸어가면서 전동킥보드들이 담기게 영상을 찍어보았다. 

학교 앞 전동킥보드

 

 

 

 

🚀 전동킥보드 탐지 영상 

YOLO 모델을 통해서 객체 탐지를 진행하면 다음과 같이 탐지 결과 영상이 나온다.

많이 시도해보지 않은 상태에서 테스트 삼아 해본 것 치고는 꽤 잘 전동킥보드를 인식하는 것을 확인할 수 있다. 헤헤헤 (천만다행) 

학교 앞 전동킥보드 객체 탐지 YOLO

 

---

✅ 결론 

• 생각보다 YOLO 모델은 똑똑하다.
• 생각보다 YOLO 모델은 더더욱 유저 친화적이고, 초보자도 쉽게 접근할 수 있다.
• 생각보다 내가 학습시킨 데이터와 설정한 모델 사이즈, epochs 수 등이 좋았나보다.
• 이렇게 잘 인식할 줄은 몰랐다. 그래도 오탐지 하는 부분들이 있다. 계속 학습시키면서 줄여 나가야겠다. 
• 학습 데이터를 더 늘릴 수 있으면 늘려야한다. 걸어가면서 찍었으니 이정도 탐지되는거지 차 타고 빠르게 지나가면서 블랙박스 영상만으로 탐지해보면 거의 탐지 안 될 수도 있다.