Yolo 기반 쓰레기 탐지 APP

date
6월 21, 2024

Team BIO : 재활용품 분리배출 도우미#

프로젝트 소개#

모바일 환경에서 YOLOv8기반 Object Detection으로 재활용 쓰레기를 탐지하여 사용자들에게 올바른 재활용 가이드 라인을 제공한다.

사용한 Dataset - AI-HUB 생활폐기물 데이터

https://noticon-static.tammolo.com/dgggcrkxq/image/upload/v1709963083/noticon/v7lunl6p1dta5mrkuugl.png

구현 기능#

기능 1 : 재활용품 탐지 - 모바일 서비스#

사용자는 모바일 어플을 통해 재활용 쓰레기의 정확한 품목을 알 수 있다.

기능 2 : 재활용 가이드 제공 - 모바일 서비스#

사용자에게 탐지된 재활용품의 처리 가이드 라인을 제공한다.

기능 3 : 데이터 시각화 - 개발자 도구#

모델의 파이프라인을 수정하여 탐지한 물체의 이미지 벡터를 추출할 수 있다.

추출한 이미지 벡터로 T-SNE 및 PCA 시각화를 진행 후, 내부적으로 정성 평가를 위해 각 이미지를 비교할 수 있도록 도구를 개발하였다.

개발 과정 간략도#

디렉토리 설명#

Garbage-Classification<br> ┣ Mobile_Develop                # 어플리케이션 개발 폴더<br> ┃ ┗ app                               # 어플리케이션 코드<br> ┣ Model_Develop                 # 모델 및 데이터셋 관련 폴더<br> ┃ ┣ Custom_Model             # 기존 YOLO 패키지 수정 <br> ┃ ┣ Dataset_Sampling        # 데이터셋 샘플링 관련 코드<br> ┃ ┣ Dataset_Visualization   # 데이터셋 시각화 관련 코드<br> ┃ ┗ Model_Train                  # 모델 가중치 및 결과<br><br><br><br>

현재 및 향후 기대 효과#

[배경] 대한민국 가정에서 버린 쓰레기는 모두 각 지역구 선별장으로 이동하게 되며 일부 자동화 선별장을 제외한 대부분은 사람이 직접 수작업 분류를 진행하고 있다. <br>

<첫째>

현재는 사용자가 직접 쓰레기를 비추며 안내문을 출력하지만 IOT서비스와 연결하여 SMART쓰레기통 , SMART주방 , SMART분리수거장으로 활용할 수 있다.

Be Your Own Lamp