Advanced Textile Technology ›› 2024, Vol. 32 ›› Issue (1): 45-53.

Previous Articles     Next Articles

A Method for Identifying Women's Sleeves Based on Improved YOLOv5 and ResNet50

  

  1. 1. Academy of Fine Arts, Chongqing University of Education, Chongqing 400067,China; 2. Institute of Vehicle Engineering, Chongqing University of Technology, Chongqing 400054, China
  • Online:2024-01-10 Published:2024-01-30

基于改进YOLOv5和ResNet50的女装袖型识别方法

  

  1. 1.重庆第二师范学院美术学院,重庆 400065;2.重庆理工大学车辆工程学院,重庆 400054
  • 通讯作者: 妥吉英,E-mail:tjy@cqut.edu.cn
  • 作者简介:曹涵颖(1990― ),女,重庆人,硕士研究生,研究方向为服装与服饰设计。
  • 基金资助:
    重庆市教科委“十四五”规划项目(2021-JZ-030);重庆市教委科学技术研究计划项目(KJQN202201621);重庆市自然科学基金面上项目(cstc2020jcyj-msxmX0331);重庆第二师范学院校级科研项目(KY202127C)

Abstract: In response to the problems of numerous classifications of women's clothing sleeve shapes, difficulty in feature recognition, and unsatisfactory detection results, on the basis of fully utilizing the correlation information between different women's clothing sleeve shapes, an improved YOLOv5 object detection and ResNet50 object classification deep learning method was used to achieve automatic recognition of women's clothing sleeve shapes.
Firstly, two methods of sleeve type classification were combined based on the length and shape of women's clothing sleeves. The sleeves were divided into four primary classifications based on the length of women's clothing: sleeveless, ultra short, short, and long sleeves. On the basis of the primary classifications, they were further divided into 15 secondary classifications based on their morphological characteristics, including bra, neck hanging, raglan sleeves, bubble sleeves, and leg sleeves. Secondly, clothing sample images were collected from e-commerce platforms, taking into account factors such as different angles, lighting, and backgrounds. On the basis of balancing different sleeve types, 3600 sleeve type images of women's clothing were collected, screened, and labeled. A clothing sleeve type dataset containing approximately 6500 sleeve type samples was obtained, and sleeve type labeling was performed using Labelimg software. Once again, based on the analysis of the YOLOv5 object detection network, CBAM attention mechanism, ResNet50 residual network principle, and network features, an improved YOLOv5 and ResNet50 combined deep learning method based on CBAM attention mechanism is proposed for women's sleeve automatic recognition. Among them, YOLOv5 model gradually adjusts the parameters of the network model through the back propagation and gradient descent characteristics of the Convolutional neural network on the self labeled garment sleeve shape data set to obtain the network parameters suitable for the detection of women's sleeve shape, thus realizing the target detection of women's sleeve shape in the primary level classification. The convolutional attention module CBAM, which combines channel attention mechanism and spatial attention mechanism, is beneficial for solving the problem of no attention preference in the original network, thereby enhancing the effectiveness of sleeve detection. Four independent ResNet50 residual networks were used to carry out sleeve type secondary classification recognition based on the improved YOLOv5 network detection of four sleeve types: sleeveless, ultra short sleeved, short sleeved, and long sleeved, respectively, in order to obtain the final results of women's sleeve type recognition. Finally, based on the Python language and Pytorch framework, the proposed deep learning algorithm for women's clothing sleeve recognition was designed and implemented, and the model was trained on the sleeve dataset to verify the effectiveness of sleeve recognition through experiments.
The results indicate that ① compared to the YOLOv5 method and the CBAM improved YOLOv5 method, the CBAM improved YOLOv5 and ResNet50 combined method, which introduces the correlation information between women's sleeve shapes, has more advantages in the accuracy of women's sleeve shape recognition. The overall recognition accuracy is about 93.3, and its overall accuracy is 12.2 and 8 percentage points higher than the YOLOv5 model improved by YOLOv5 and CBAM, respectively; ② in the task of identifying women's sleeve type by YOLOv5, improved YOLOv5, YOLOv5 and ResNet50 combined methods, compared with sleeveless and long sleeves, the identification of ultra-short sleeves and short sleeves is more difficult, and the overall accuracy is more difficult to improve.

Key words: women's sleeve shape, deep learning, YOLOv5;attention mechanism, ResNet50

摘要: 针对女装袖型分类繁多、特征识别困难、检测效果不理想等问题,根据不同女装袖型的关联信息,结合注意力机制改进的YOLOv5目标检测网络和ResNet50残差网络,提出了一种女装袖子造型的自动识别方法。首先,从电商平台收集服装样本图像,按照长短大类和形态小类标记对女装袖型进行归类,建立了包含3600张图像的袖型数据集;其次,结合注意力机制改进的YOLOv5目标检测网络和ResNet50残差网络,设计了女装袖型识别方法;最后,在袖型数据集上开展模型训练,并通过实验验证袖型识别的效果。结果表明:改进YOLOv5和ResNet50相结合的深度学习方法可以有效地对女装袖型进行识别,整体识别准确率约93.3%。该女装袖型识别方法准确、便捷,可以实现大量服装款式的分类快速检测,提高服装设计效率,促进人工智能技术在服装设计领域的应用,助力我国智能制造和电子商务的发展。

关键词: 女装袖型, 深度学习, YOLOv5, 注意力机制, ResNet50

CLC Number: