Common garbage classification using mobilenet

College

Gokongwei College of Engineering

Department/Unit

Electronics And Communications Engg

Document Type

Conference Proceeding

Source Title

2018 IEEE 10th International Conference on Humanoid, Nanotechnology, Information Technology, Communication and Control, Environment and Management, HNICEM 2018

Publication Date

7-2-2018

Abstract

Garbage classification is the first step in waste segregation, recycling, or reuse. MobileNet was used to generate a model that classifies common trash according to the following categories: glass, paper, cardboard, plastic, metal, and other trash. A dataset of 2527 trash images in.jpg extension was used for the training. The model used transfer learning from a model trained on the ImageNet Large Visual Recognition Challenge dataset. The TensorFlow for Poets git repository was cloned as a working directory to retrain the MobileNet model in 500 steps. The resulting baseline model, with a final test accuracy of 87.2% was optimized and quantized. In the Andoid app development, the optimized model (with 89.34% confidence) is preferred over the quantized model (with 1.47% confidence) based on the test using a plastic image. The model app was successfully installed in a Samsung Galaxy S6 Edge}+textbf{{mobile phone. The installed mobile app successfully identified a cardboard material in an image with a}{cardboard container. It is recommended to rerun the training using more steps as this may improve the quantized model performance since a quantized model is fit for mobile devices than models with no quantization. © 2018 IEEE.

html

Digitial Object Identifier (DOI)

10.1109/HNICEM.2018.8666300

Disciplines

Electrical and Electronics | Systems and Communications

Keywords

Refuse and refuse disposal; Androids; Smartphones; Image processing; Samsung Galaxy S6 Edge+

Upload File

wf_yes

This document is currently not available here.

Share

COinS