Improved Convolutional Neural Network and Transfer Learning with VGG16 Approach for Image Classification
Muzafar Shah Habibullah, Md. Oli Ullah, Minhaz Kamal, Md. Sabbir Hasan Sohag, Md. Saifur Rahman · 2023
Transfer learning may boost modeling speed. This research unifies the improved VGG16 model. Skipping VGG16’s entirely connected layer and tying it to the layer following it improves CNN’s architecture and reduces its processing burden. CNNs are classifying cat and dog photos. With a small dataset, CNN training may take time and resources. Transfer learning uses ImageNet models to handle similar challenges in new contexts. Transfer learning enhances CNN cat-dog classification. The proposed technique selects a suitable pre-trained model, freezes its layers to retain acquired properties, adds layers to learn task-specific features, and tunes it using the cat and dog dataset. Data augmentation improves model performance and prevents overfitting. Hyperparameter optimization and validation enhance model accuracy and speed. This method enhances cat and dog photo classification even without adequate data to train CNNs with an accuracy of 97%.