TFLite 경량화 모델을 이용한 쓰레기 분리수거 로봇

Waste Sorting Robot Based on a Lightweight TFLite Model

  • 김가은 (경기대학교 전자공학과 ) ;
  • 김유진 (경기대학교 전자공학과 ) ;
  • 신은지 (경기대학교 전자공학과 ) ;
  • 신승민 (한국공학대학교 전자공학과 )
  • Ga-Eun Kim (Dept. of Electronic Engineering, Kyong-gi University) ;
  • Yu-Jin Kim (Dept. of Electronic Engineering, Kyong-gi University) ;
  • Eun-Ji Shin (Dept. of Electronic Engineering, Kyong-gi University) ;
  • Seung-Min Shin (Dept. of Embedded Systems Engineering, Tech University of Korea)
  • 발행 : 2024.10.31

초록

본 연구는 TFLite 경량화 AI 모델을 활용한 자동 쓰레기 분리수거 시스템을 제안한다. 라즈베리파이 환경에서 아두이노를 통해 두 개의 로봇팔을 제어하여, 기존 방식보다 작업 효율성을 높였다. 쓰레기 종류별 데이터 분포를 일정하게 맞추고, 고품질의 데이터를 사용함으로써 모델의 인식 성능을 크게 개선하였다. 실시간 객체 인식 테스트 결과 평균 정확도는 90% 이상에 달했다. 이러한 과정을 통해 자동 분리수거 시스템의 정확성과 효율성을 한층 더 높일 수 있을 것으로 기대된다.

키워드

과제정보

본 논문은 과학기술정보통신부 대학디지털교육역량강화사업의 지원을 통해 수행한 ICT멘토링 프로젝트 결과물입니다

참고문헌

  1. Alsing, Oscar, "Mobile object detection using Tensor Flow Lite and transfer learning' , KTH, School of Electrical Engineering and Computer Science, p.1-78, 2018