Enhancement of Image Classification Using Transfer Learning and GAN-Based Synthetic Data Augmentation

Subhajit Chatterjee, Debapriya Hazra, Yung-Cheol Byun and Yong-Woon Kim
Additional contact information
Subhajit Chatterjee: Department of Computer Engineering, Jeju National University, Jeju 63243, Korea
Debapriya Hazra: Department of Computer Engineering, Jeju National University, Jeju 63243, Korea
Yung-Cheol Byun: Department of Computer Engineering, Jeju National University, Jeju 63243, Korea
Yong-Woon Kim: Centre for Digital Innovation, CHRIST University (Deemed to be University), Bengaluru 560029, Karnataka, India

Mathematics, 2022, vol. 10, issue 9, 1-16

Abstract: Plastic bottle recycling has a crucial role in environmental degradation and protection. Position and background should be the same to classify plastic bottles on a conveyor belt. The manual detection of plastic bottles is time consuming and leads to human error. Hence, the automatic classification of plastic bottles using deep learning techniques can assist with the more accurate results and reduce cost. To achieve a considerably good result using the DL model, we need a large volume of data to train. We propose a GAN-based model to generate synthetic images similar to the original. To improve the image synthesis quality with less training time and decrease the chances of mode collapse, we propose a modified lightweight-GAN model, which consists of a generator and a discriminator with an auto-encoding feature to capture essential parts of the input image and to encourage the generator to produce a wide range of real data. Then a newly designed weighted average ensemble model based on two pre-trained models, inceptionV3 and xception, to classify transparent plastic bottles obtains an improved classification accuracy of 99.06%.

Keywords: deep learning; generative adversarial networks; image classification; transfer learning; plastic bottle (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2022
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (1)

Downloads: (external link)
https://www.mdpi.com/2227-7390/10/9/1541/pdf (application/pdf)
https://www.mdpi.com/2227-7390/10/9/1541/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:10:y:2022:i:9:p:1541-:d:808213

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().