Deakin University
Browse

File(s) under permanent embargo

A combination model based on transfer learning for waste classification

journal contribution
posted on 2020-10-10, 00:00 authored by Belinda HuangBelinda Huang, Jing He, Zenglin Xu, Guangyan HuangGuangyan Huang
The increasing amount of solid waste is becoming a significant problem that needs to be addressed urgently. The reliable and accurate classification method is a crucial step in waste disposal because different types of wastes have different disposal ways. The existing waste classification models driven by deep learning are not easy to achieve accurate results and still need to be improved due to the various architecture networks adopted. Their performance on different datasets is varied, and there is also a lack of specific large‐scale datasets for training. We propose a new combination classification model based on three pretrained CNN models (VGG19, DenseNet169, and NASNetLarge) for processing the ImageNet database and achieve high classification accuracy. In our proposed model, the transfer learning model based on each pretrained model is constructed as a candidate classifier, and the optimal output of three candidate classifiers is selected as the final classification result. The experiments based on two waste image datasets demonstrate that the proposed model achieves 96.5% and 94% classification accuracy and outperforms several counterpart methods.

History

Journal

Concurrency and Computation: Practice & Experience

Volume

32

Issue

19

Article number

e5751

Pagination

1 - 12

Publisher

Wiley

Location

Hoboken, N.J.

ISSN

1532-0626

eISSN

1532-0634

Language

eng

Publication classification

C1 Refereed article in a scholarly journal

Copyright notice

2020, John Wiley & Sons, Ltd.