Review of Deep Transfer Learning Models for Image Classification

Authors

  • Neelesh Karthikeyan College of Engineering Guindy, Anna University

DOI:

https://doi.org/10.3991/ijes.v10i01.29783

Keywords:

Deep Learning, Transfer Learning, Waste Classification, CNN, Hyperparameter, TrashNet

Abstract


With the rapid rise in urbanization, solid waste generation has increased exceedingly. This study aims to develop a Convolutional Neural Network (CNN) to classify trash into biodegradable and non-biodegradable wastes. The TrashNet dataset was utilized in this study, and image augmentation was employed to make the model more robust against translation invariance. Transfer Learning methods based on CNN have shown promising outcomes on diverse image classification problems. This paper reviews the deep learning models available with pre-trained weights in the Keras library. The performance of the models was compared, and the model based on NASNetMobile had the highest accuracy of 97%. Further, the model’s hyper-parameters were tuned, and the significance of a hyper-parameter on the model’s accuracy was studied.

Cover iJES Vol. 10 No. 01 (2022)

Downloads

Published

2022-03-16

How to Cite

Karthikeyan, N. (2022). Review of Deep Transfer Learning Models for Image Classification. International Journal of Recent Contributions from Engineering, Science & IT (iJES), 10(01), pp. 17–28. https://doi.org/10.3991/ijes.v10i01.29783

Issue

Vol. 10 No. 01 (2022)

Section

Papers

License

Copyright (c) 2022 Neelesh K

Creative Commons License

This work is licensed under a Creative Commons Attribution 4.0 International License.

The submitting author warrants that the submission is original and that she/he is the author of the submission together with the named co-authors; to the extend the submission incorporates text passages, figures, data or other material from the work of others, the submitting author has obtained any necessary permission.
Articles in this journal are published under the Creative Commons Attribution Licence (CC-BY What does this mean?). This is to get more legal certainty about what readers can do with published articles, and thus a wider dissemination and archiving, which in turn makes publishing with this journal more valuable for you, the authors.
By submitting an article the author grants to this journal the non-exclusive right to publish it. The author retains the copyright and the publishing rights for his article without any restrictions.
This journal has been awarded the SPARC Europe Seal for Open Access Journals (What's this?)