이미지 기반의 쓰레기 종류 예측

프로젝트 소개

• 딥러닝(CNN)을 사용하여 이미지를 입력으로 했을 때, 쓰레기 종류를 분류하여 출력한다.

Data

• dataset.zip : https://github.com/garythung/trashnet/blob/master/data/dataset-resized.zip 또는 https://www.kaggle.com/asdasdasasdas/garbage-classification
• multi-trash.zip : dataset.zip에서 임의의 2가지 쓰레기 이미지를 그림판으로 합성한 이미지 10개.

파일 설명

• 1_oneClass_classification.ipynb : 한 가지 class에 대한 분류 예측
• 2_multiClass_classification.ipynb : multi-class에 대한 분류 예측

실행 화면

• Google Colab 환경에서 실행.
• (.ipynb 파일 참고)

참고자료

• https://towardsdatascience.com/how-to-build-an-image-classifier-for-waste-sorting-6d11d3c9c478 (https://github.com/collindching/Waste-Sorter)
• CNN에 대한 이론 설명 : https://bcho.tistory.com/tag/VGG