着装分类模型的训练方法及装置、着装分类方法及装置

摘要

本发明公开了一种着装分类模型的训练方法及装置、着装分类方法及装置，该模型训练方法包括：获取训练样本；将样本图像输入着装分类模型；对样本图像中的人像区域进行语义分割；基于第二特征、目标特征、着装分类结果以及对应的着装分类标签计算目标损失，并根据目标损失调整着装分类模型的参数，直至满足收敛条件。本发明通过MASK机制进行语义分割，辅助模型的注意力机制学习区分人像区域和背景，降低了着装模型训练的难度，模型的注意力机制能更好地表征各类着装属性本身的特征，从而准确高效地实现着装分类任务，另外通过设计多种损失函数联合优化模型，使模型收敛至最优，提高了着装分类的准确率。