IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v10y2022i9p1541-d808213.html
   My bibliography  Save this article

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

Author

Listed:
  • 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)

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%.

Suggested Citation

  • Subhajit Chatterjee & Debapriya Hazra & Yung-Cheol Byun & Yong-Woon Kim, 2022. "Enhancement of Image Classification Using Transfer Learning and GAN-Based Synthetic Data Augmentation," Mathematics, MDPI, vol. 10(9), pages 1-16, May.
  • Handle: RePEc:gam:jmathe:v:10:y:2022:i:9:p:1541-:d:808213
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/10/9/1541/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/10/9/1541/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Tachwali, Y. & Al-Assaf, Y. & Al-Ali, A.R., 2007. "Automatic multistage classification system for plastic bottles recycling," Resources, Conservation & Recycling, Elsevier, vol. 52(2), pages 266-285.
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Florin Leon & Mircea Hulea & Marius Gavrilescu, 2022. "Preface to the Special Issue on “Advances in Artificial Intelligence: Models, Optimization, and Machine Learning”," Mathematics, MDPI, vol. 10(10), pages 1-4, May.

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.

      Corrections

      All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:10:y:2022:i:9:p:1541-:d:808213. See general information about how to correct material in RePEc.

      If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

      If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

      If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

      For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

      Please note that corrections may take a couple of weeks to filter through the various RePEc services.

      IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.