article Open access

Solving a classification problem using Transfer learning on small image datasets

  • Gradus
Research footprint

At a glance

Citations
1
References
15
Comments
0
Paper overview

Abstract

In this study, we explore the efficiency of transfer learning in training convolutional neural networks (CNNs) for image classification tasks, focusing on a dataset with limited size. Utilizing the VGG16 model, we investigate the impact of varying the number of trainable layers on classification accuracy. Our specific classification challenge involves distinguishing between images of cars with open and closed hoods. The study demonstrates that employing transfer learning enables the pretrained VGG16 model to achieve over 97% accuracy in this binary classification task, even with a training set of just 1000 image samples. This research was conducted solely by the undersigned, who also represents the sole author of this paper.

Record transparency

Publication details

DOI
10.47833/2024.1.csc.011
OpenAlex
W4399026509
Document type
article
Language
EN
Source
Gradus
Last metadata update
Community

Comments

Log in to join the discussion.

  1. No comments yet. Start the discussion.