[发明专利]基于图像识别的通用着装识别方法、系统及处理设备

权利要求书

1.一种基于图像识别的通用着装识别方法，其特征在于，包括以下步骤：

S100：通过公共数据集训练基础模型，得到通用着装识别的第一特征提取器；

S200：从待识别图像中获取第一行人着装图像；

S300：调用所述第一特征提取器，提取所述第一行人着装图像的第一特征向量；

S400：将所述第一特征向量与特征库进行比对，根据比对结果输出所述第一行人着装图像的着装识别结果。

2.根据权利要求1所述的一种基于图像识别的通用着装识别方法，其特征在于，所述S400步骤中，所述特征库通过以下方法得到：

S410：获取多张不同类型的第二行人着装图像，处理后得到样本着装图像；

S420：通过第二特征提取器提取全部所述样本着装图像的第二特征向量；

S430：将每张所述第二行人着装图像与所述第二特征向量相互关联，得到所述特征库。

3.根据权利要求2所述的一种基于图像识别的通用着装识别方法，其特征在于，所述S400步骤具体包括：

S410’：通过余弦相似度将所述第一特征向量与全部的所述第二特征向量的进行比对；

S420’：当所述余弦相似度大于或等于设定阈值时，选出所述余弦相似度最大值所对应的第二特征向量作为目标特征向量，将所述目标特征向量关联的所述第二行人着装图像所对应的着装类型作为着装识别结果；

S430’：当所述余弦相似度小于所述设定阈值时，输出未知着装类型作为识别结果。

4.根据权利要求3所述的一种基于图像识别的通用着装识别方法，其特征在于，所述S430’之后还包括：

S440’：获取所述未知着装类型所对应的所述第一行人着装图像，对所述特征库进行更新。

5.根据权利要求1所述的一种基于图像识别的通用着装识别方法，其特征在于，所述S200步骤具体包括：

S210：从所述待识别第一行人着装图像中获取至少一个行人的行人patch；

S220：通过所述行人patch对人体图像进行分割，得到行人mask；

S230：根据所述行人mask去除背景干扰，得到所述第一行人着装图像。

6.根据权利要求5所述的一种基于图像识别的通用着装识别方法，其特征在于，所述S220步骤中，通过DeepLabv3plus、FCN、PSPNet、DDRNet、BiseNet或Bisenetv2算法进行人体图像分割。

7.根据权利要求1所述的一种基于图像识别的通用着装识别方法，其特征在于，所述S100步骤中，通过通用的行人重识别方法训练所述基础模型，所述基础模型采用开源训练框架fast-reid进行训练。

8.根据权利要求1所述的一种基于图像识别的通用着装识别方法，其特征在于，所述S100步骤中，所述公共数据集为Market1501、DukeMTMC、MSMT17_C、CUHK03中的一个或多个。

9.一种基于图像识别的通用着装识别系统，其特征在于，通过权利要求1-8中任一项所述的一种基于图像识别的通用着装识别方法运行，包括第一特征提取器和特征库；所述第一特征提取器通过公共数据集训练得到，能够得到所述第一行人着装图像的第一特征向量；所述特征库通过所述第一特征提取器获取第二行人着装图像的第一特征向量，并与相应的第二行人着装图像进行关联。

10.一种处理设备，其特征在于，包括：

一个或多个处理器；

存储器，用于存储一个或多个计算机程序，一个或多个所述处理器用于执行所述存储器存储的一个或多个计算机程序，以使一个或多个所述处理器执行如权利要求1-8中任一项所述的一种基于图像识别的通用着装识别方法。