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분리수거장으로 활용할 수 있다.