A Fine-grained Attributes Recognition Model for Clothing Based on Improved the CSPDarknet and PAFPN Network

Abstract

An efficient and accurate recognition model for fine-grained attributes of clothing has significant business prospects and social implications. However, the inherent diversity and complexity of clothing makes acquiring datasets with fine-grained attributes a costly endeavor. To address these challenges, we propose a lightweight clothing fine-grained attributes recognition model. First, the Ghost module is introduced into the CSPDarknet network to enhance the depth and expressiveness of feature learning while reducing the parameters and computational complexity. Then, the Conv module is replaced with the GSConv module in the PAFPN network to further reduce the network computational load, and the SE attention mechanism is also added to enhance the perception of key features. Finally, the Detect module is utilized to achieve effective recognition of fine-grained attributes of clothing. To evaluate the model performance, we construct a clothing dataset containing 20 fine-grained attributes. The experimental results show that the model achieves precision, recall and mAP of 76.2%, 78.9% and 81.7%. Compared to the original model, the parameters are reduced by 26.2%, and the FPS is improved by 25.4%. Our proposed model performs well on the small-scale dataset and improves its performance in resource-constrained environments, which has practical applications in clothing recommendation, virtual fitting, and personalization.