GarbageClassification

KUCC x YCC AI 부트캠프 23.02.26

1등 1조 - Garbage Classification

재활용 쓰레기 분류 AI 개발

문제 인식

분리 수거의 효율을 높이기 위해, 사진을 찍으면 쓰레기를 분류해주는 AI 제작

Dataset

Kaggle 대회 'Garbage Classification (12 classes)'의 약 15,000개 이미지

https://www.kaggle.com/datasets/mostafaabla/garbage-classification

프로젝트 목표

프로젝트 세부 목표

1. 직접 로더 및 모델 구현 2. 전이학습, 파인튜닝으로 모델 성능 향상 3. 여러 튜닝 기법 활용: WandB를 이용한 AutoML, optimization, regularization

튜닝 과정

최종 모델

Accuracy: 95.0999%

Custom Dataset으로 Test

white glass, carboard, plastic, white glass로 분류 마지막 사진은 plastic이지만, 사람의 눈으로도 white glass인지 plastic인지 구분이 어려울 수 있는 사진이므로 (Accuracy를 평가하는 기준: Human Level Performance) 예측이 틀렸을 것. 데이터셋을 늘리면 해결이 될 여지가 있긴 함.

아쉬운 점 및 보충할 부분