技术特征：
1.一种着装分类模型的训练方法，其特征在于，包括以下步骤：获取训练样本，其中，所述训练样本包括样本图像和对应的着装分类标签；将所述样本图像输入着装分类模型，所述着装分类模型用于提取所述样本图像的第一特征，并基于注意力机制对所述第一特征进行处理得到第二特征，以及基于残差注意力机制对所述第二特征进行处理得到着装分类结果；基于mask机制对所述样本图像中的人像区域进行语义分割，得到目标图像；基于所述第二特征、目标特征、所述着装分类结果以及对应的着装分类标签计算目标损失，并根据所述目标损失调整所述着装分类模型的参数，直至满足收敛条件；其中，所述目标特征基于所述目标图像和所述第一特征得到。2.根据权利要求1所述的着装分类模型的训练方法，其特征在于，所述基于所述第二特征、目标特征、所述着装分类结果以及对应的着装分类标签计算目标损失的步骤具体包括：基于所述第二特征和所述目标特征计算第一损失；基于所述着装分类结果和对应的着装分类标签计算第二损失；根据所述第一损失和所述第二损失确定目标损失。3.根据权利要求2所述的着装分类模型的训练方法，其特征在于，所述基于所述第二特征和所述目标特征计算第一损失的步骤具体包括：将所述第二特征和所述目标特征代入第一损失函数得到第一损失；所述第一损失函数用于迫使所述第二特征不断逼近所述目标特征。4.根据权利要求2所述的着装分类模型的训练方法，其特征在于，所述基于所述第二特征、所述目标特征、所述着装分类结果以及对应的着装分类标签计算目标损失的步骤具体包括：基于所述第二特征和所述目标特征计算第一损失；将所述着装分类结果和对应的着装分类标签代入第二损失函数计算第二损失；将所述着装分类结果和对应的着装分类标签代入第三损失函数计算第三损失；所述第三损失函数为所述第二损失函数的正则项；根据所述第一损失、所述第二损失以及所述第三损失确定目标损失。5.根据权利要求1所述的着装分类模型的训练方法，其特征在于，所述基于mask机制对所述样本图像中的人像区域进行语义分割，得到目标图像的步骤包括：基于mask机制对所述样本图像中的人像区域进行语义分割，得到中间图像；对所述中间图像中的非人脸区域进行语义分割，得到目标图像。6.一种着装分类模型的训练装置，其特征在于，包括：获取模块，用于获取训练样本，其中，所述训练样本包括样本图像和对应的着装分类标签；分类模块，用于将所述样本图像输入着装分类模型，所述着装分类模型用于提取所述样本图像的第一特征，并基于注意力机制对所述第一特征进行处理得到第二特征，以及基于残差注意力机制对所述第二特征进行处理得到着装分类结果；分割模块，用于基于mask机制对所述样本图像中的人像区域进行语义分割，得到目标图像；训练模块，用于基于所述第二特征、目标特征、所述着装分类结果以及对应的着装分类
标签计算目标损失，并根据所述目标损失调整所述着装分类模型的参数，直至满足收敛条件；其中，所述目标特征基于所述目标图像和所述第一特征得到。7.一种着装分类方法，其特征在于，包括以下步骤：获取待分类图像，所述待分类图像包含着装信息；通过着装分类模型对所述待分类图像进行着装分类处理，得到所述待分类图像的着装分类结果，所述着装分类模型通过如权利要求1-5任一项所述的着装分类模型的训练方法得到。8.一种着装分类装置，其特征在于，包括：待分类图像获取模块，用于获取待分类图像，所述待分类图像包含着装信息；分类处理模块，用于通过着装分类模型对所述待分类图像进行着装分类处理，得到所述待分类图像的着装分类结果，所述着装分类模型通过如权利要求1-5任一项所述的着装分类模型的训练方法得到。9.一种电子设备，其特征在于，包括存储器以及与所述存储器连接的处理器，所述处理器执行存储在所述存储器上的计算机程序时实现权利要求1-5中任一项所述的着装分类模型的训练方法或者如权利要求7所述的着装分类方法。10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现权利要求1-5中任一项所述的着装分类模型的训练方法或者如权利要求7所述的着装分类方法。
技术总结
本发明公开了一种着装分类模型的训练方法及装置、着装分类方法及装置，该模型训练方法包括：获取训练样本；将样本图像输入着装分类模型；对样本图像中的人像区域进行语义分割；基于第二特征、目标特征、着装分类结果以及对应的着装分类标签计算目标损失，并根据目标损失调整着装分类模型的参数，直至满足收敛条件。本发明通过MASK机制进行语义分割，辅助模型的注意力机制学习区分人像区域和背景，降低了着装模型训练的难度，模型的注意力机制能更好地表征各类着装属性本身的特征，从而准确高效地实现着装分类任务，另外通过设计多种损失函数联合优化模型，使模型收敛至最优，提高了着装分类的准确率。着装分类的准确率。着装分类的准确率。
技术研发人员：吕心铋 罗超 梁贤朋 邹宇
受保护的技术使用者：携程旅游信息技术（上海）有限公司
技术研发日：2021.12.17
技术公布日：2022/3/25