Enhanced Convolutional Neural Network for Fashion Classification

• Published in: Engineering, technology & applied science research technology & applied science research, 2024-10, Vol.14 (5), p.16534-16538
• Main Authors: Haji, Lailan M., Mustafa, Omar M., Abdullah, Sherwan A., Ahmed, Omar M.
• Format: Article
• Language: English
• ISSN: 2241-4487; 1792-8036
• DOI: 10.48084/etasr.8147

Fashion items are hard to classify since there are a million variations in style, texture, and pattern. Image classification is among the noted strengths of convolutional neural networks. This research introduces an improved CNN architecture for fashion classification, utilizing image augmentation and batch normalization to improve model performance and generalization. To make the model more robust, image augmentation techniques like rotation, width and height shift, zoom, and flips were employed. In addition, a Batch Normalization layer is added in the middle, which can help on stabilizing the learning process and accelerating convergence. The proposed model was trained on an augmented dataset, achieving a satisfactory improvement in test accuracy of 91.97% compared to a baseline CNN model, which obtained 88.5% accuracy. According to the results, the image augmentation with the application of Batch Normalization improves the CNN architecture for better effectiveness in fashion classification tasks.