Journal of Jilin University Science Edition ›› 2023, Vol. 61 ›› Issue (5): 1195-1201.

Previous Articles     Next Articles

Clothing Classification Algorithm Based onConvolution and Transformer Fusion

ZHU Shuchang1, LI Wenhui2   

  1. 1. School of Art and Design, Jilin Engineering Normal University, Changchun 130052, China;
    2. College of Computer Science and Technology, Jilin University, Changchun 130012, China
  • Received:2022-07-19 Online:2023-09-26 Published:2023-09-26

Abstract: Aming at  the problem that traditional clothing classification algorithms based on convolutional neural networks could not meet the needs of massive and diverse clothing classification, we  proposed a clothing classification network based on convolutional attention fusion.  The network adopted a parallel structure, including a ResNet branch and a Transformer branch, and  fullly utilizing  the local features extracted by the convolution operation and the global features extracted by the self-attention mechanism to enhance the representation learning ability of the network, thereby improving the performance and generalization ability of the clothing classification algorithm.  In order to verify the effectiveness of the method, we conducted comparative experiments on the Fashion-MNIST and DeepFashion datasets.   The results show that on the Fashion-MNIST dataset, the method achieves an accuracy rate of 93.58%, and on the DeepFashion dataset, the method  achieves an accuracy rate of 71.1%, which is superior to the  experimental results of other methods.

Key words: clothing category classification, convolutional neural network,  , feature fusion

CLC Number: 

  • TP391.41