Solving a classification problem using Transfer learning on small image datasets

Drenyovszki, Rajmund (2024) Solving a classification problem using Transfer learning on small image datasets. GRADUS, 11 (1). pp. 1-9. ISSN 2064-8014

Preview

Text 2024_1_CSC_011_Drenyovszki.pdf - Published Version Available under License Creative Commons Attribution. Download (705kB) | Preview

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.

Actions (login required)

Edit Item