💡 YOLOv5

• Yolo란 "You Only Look Once"의 약자로, 실시간 객체 감지를 위한 딥러닝 기반 알고리즘이다. YOLO는 이미지를 그리드로 분할하고,각 그리드 셀마다 Bounding Box를 예측하여 그 안에 포함된 객체를 인식한다. YOLO는 주로 자율 주행 자동차, 보안 시스템, 객체 추적 등 다양한 응용 분야에서 활용된다. Yolo 알고리즘을 지원하는 대표적인 프레임워크로는 TensorFlow, Pytorch, Darknet 등이 있다.

❓문제 상황

• 나는 육군 본부 주관으로 진행했던 AI 교육과정에서 프로젝트를 진행했다. 프로젝트는 Yolo를 이용한 개발이었고, 군에서 필요한 문제를 과제로 선정하였다. 우리 팀은 위병소 출입체계에 대한 솔루션을 주제로 삼았다.

📌문제해결 알고리즘

★Smart 위병소 출입체계 개발

현상황은 위병소를 출입할 때, 얼굴, 공무원증, 휴대폰 국방 모바일 보안 화면을 제시해야 위병소 출입이 가능하다. 이를 자동화 시키는게 프로젝트의 주제였다.

1. 데이터 수집 및 전처리

프로젝트는 팀원 5명의 얼굴을 학습시키고, "팀원"의 얼굴, 공무원증, 보안화면 라벨링 후 데이터 증강 진행

2. YOLO모델 설계 및 학습

Colab Pro 사용하요 학습 시킨 후, pt 파일() Local 에서 셋팅

--> Colab에 webcam 연동제한

3. Serial 통신 (아두이노 활용)

얼굴, 공무원증, 보안화면 인식되는 것 확인 후, 결과값으로 아두이노에 신호보내기

4. UI 개발 및 배포

사용자 인터페이스 개발 및 배포
-->Flask 이용해서 구현 (동일 네트워크를 넘어 외부 호스팅 확인)

P.S. 본 프로젝트는 23.07.10 ~ 23.7.21에 진행되었고, 까먹기전에 기록을 해두려고 한다.

#사용 기술 : Pytorch, Yolo, Gamma, Arduino, Flask