融合迁移学习和集成学习的服装风格图像分类方法  

A fashion style image classification method integrating transfer learning and ensemble learning

作　　者：游小荣[1] 李淑芳[2] YOU Xiaorong;LI Shufang(School of Intelligent Manufacturing,Changzhou Vocational Institute of Textile and Garment,Changzhou 213164,China;School of Intelligent Textiles and Materials,Changzhou Vocational Institute of Textile and Garment,Changzhou 213164,China)

机构地区：[1]常州纺织服装职业技术学院智能制造学院,江苏常州213164 [2]常州纺织服装职业技术学院智能纺织与材料学院,江苏常州213164

出　　处：《现代纺织技术》2024年第9期127-134,共8页Advanced Textile Technology

基　　金：江苏省碳纤维先进材料智能制造工程技术研究开发中心项目(苏教科[2023]11号)。

摘　　要：针对服装风格人工分类受主观性、地域等因素影响而造成的分类错误问题,研究了一种基于人工智能的服装风格图像分类方法。首先,在FashionStyle14数据集基础上筛除重复或无效图像,构建服装风格图像数据集;然后,采用迁移学习方法,对EfficientNet V2、RegNet Y 16GF和ViT Large 16等模型进行微调训练,生成新模型,实现基于单个深度学习的服装风格图像分类;最后,为进一步提高图像分类的准确性、可靠性和鲁棒性,分别采用基于投票、加权平均和堆叠的集成学习方法对上述单个模型进行组合预测。迁移学习实验结果表明,基于ViT Large 16的深度学习模型在测试集上表现最佳,平均准确率为77.024%;集成学习方法实验结果显示,基于投票的集成学习方法在相同测试集上平均准确率可达78.833%。研究结果为解决服装风格分类问题提供了新的思路。The fashion industry plays an important role in the global economy,and its GDP scale shows a growing trend,currently accounting for approximately 2%of the global GDP.Fashion style classification can help consumers better understand and choose fashion products that suit their preferences,making it easier for them to find the styles and brands they are interested in.At the same time,fashion style classification also plays an important role in fashion research and design,providing inspiration and reference for designers.In addition,for fashion brand companies and retailers,the precise classification of fashion styles helps to better understand market demand and consumer preferences,thereby adjusting product categories and promotion strategies.Manual fashion style classification has the characteristics of subjectivity,diversity,variability and regionality,which can easily lead to errors in classification results.Therefore,it is important to improve the objectivity and accuracy of classification by using technical means,such as artificial intelligence.In response to the above problems,this paper studied a fashion style image classification method that combines transfer learning and ensemble learning.Firstly,based on the FashionStyle14 data set,duplicate or invalid images were filtered out to construct a fashion style image data set.Secondly,pre-trained models such as EfficientNet V2,RegNet Y 16GF and ViT Large 16 were used for fine-tuning training to generate new models to achieve fashion style image classification with a single deep learning model.Thirdly,the new model was tested and its classification performance was evaluated according to the evaluation indicators,and a good deep learning model was selected.Fourthly,an ensemble learning method based on voting,weighted integration and stacking was built to perform combined predictions on the above models,so as to improve the generalization ability and stability of the model.Finally,the ensemble learning method with the best performance was selected to classify fashi

关 键 词：服装风格 迁移学习 集成学习 ViT模型 图像分类 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术] TS102.3[自动化与计算机技术—计算机科学与技术]