인공지능 기초 팀과제

데이터 수집

garythung, trashnet, (2014), GitHub repository, https://github.com/garythung/trashnet.git

데이터 가공

쓰레기 이미지를 종류별로 하위 폴더로 나뉘어 사용함. 학습용 데이터와 검증용 데이터로 분류

데이터 분석

the dataset consists of 2527 images:

501 glass

594 paper

403 cardboard

482 plastic

410 metal

137 trash

데이터 시각화

코드 실행 결과 참조

모델 학습

코드 실행 결과 참조

Trash code는 batch 32, epoch 50으로 학습 진행한 결과

Trash code2는 batch 128, epoch 50으로 학습 진행한 결과

참고문헌

TensorFlow, 「이미지 분류」, TensorFlow Core, https://www.tensorflow.org/tutorials/images/classification?hl=ko

tykwon97, ChuckChuckBakSa, (2021), GitHub repository, https://github.com/tykwon97/ChuckChuckBakSa.git

DS2.ai, 「이미지 분류 인공지능 생성하기」, DS2.ai Docs(kr), https://krdocs.ds2.ai/recipes/undefined-2

Code AI Blogs, 「Classifying Waste Images with Machine Learning」, Medium, https://medium.com/geekculture/classifying-waste-images-with-machine-learning-14b249f11544