[发明专利]服饰检测方法、装置、电子设备、程序和介质

说明书

本发明实施例公开了一种服饰检测方法、装置、电子设备、程序和介质，其中，方法包括：提取待检测图像中的人体预测关键点；提取待检测图像中的服饰特征；根据服饰特征和人体预测关键点进行服饰检测，获得服饰检测结果。本发明实施例可以通过从待检测图像中提取人体预测关键点和服饰特征，结合人体预测关键点和服饰特征来获得服饰检测结果，由于人体关键点可以为服饰检测提供更好的上下文信息，因此使用人体关键点来辅助服饰的检测，可以提高检测服饰的准确度。

技术领域

本发明涉及计算机视觉技术，尤其是一种服饰检测方法、装置、电子设备、程序和介质。

背景技术

智能手机和移动互联网的发展产生了海量的图片数据，基于图片数据的计算机视觉相关技术也取得到了飞速的发展。在计算机视觉中，图片中物体的检测是一项重要的任务，它是物体识别的基础。目前大数据驱动的深度学习方法是人工智能领域的一大热点，它也普遍应用到计算机视觉中物体的检测。

利用物体识别进行图片中人体服饰的检测可以为用户进行个性化推荐，提高用户体验。目前，在进行包含服饰的图片的图像识别之前，会预先收集相关的服饰特征，根据预先收集的服饰特征对整体图片进行相似度比对识别。由于针对已收集到的服饰特征对整体图片中的所有特征进行对比识别，导致识别特征的准确度较低。

发明内容

本发明实施例提供一种用于服饰检测的技术方案。

根据本发明实施例的一个方面，提供的一种服饰检测方法，包括：

提取待检测图像中的人体预测关键点；

提取所述待检测图像中的服饰特征；

根据所述服饰特征和所述人体预测关键点进行服饰检测，获得服饰检测结果。

在本发明的一种实施方式中，所述提取待检测图像中的人体预测关键点，包括：基于人体关键点网络，获取所述待检测图像中的人体预测关键点；

和/或，

所述提取所述待检测图像中的服饰特征，包括：基于服饰检测网络，获取所述待检测图像中的服饰特征。

在本发明的另一种实施方式中，所述人体关键点网络和所述服饰检测网络共享第1至N个卷积层，所述第1至N个卷积层为共享网络层，所述N为大于或等于2的整数。

在本发明的又一种实施方式中，所述人体关键点网络包括：所述共享网络层和关键点检测分支层；

所述共享网络层输出的特征为共享层特征，以所述共享层特征作为人体关键点特征；

所述关键点检测分支层包括M个卷积层，所述M为大于或等于1的整数；

所述关键点检测分支层用于根据所述共享网络层输出的人体关键点特征回归人体关键点的位置，得到所述人体预测关键点。

在本发明的再一种实施方式中，所述服饰检测网络包括：所述共享网络层和服饰检测分支层；

所述共享网络层输出的特征为共享层特征，以所述共享层特征作为所述服饰特征；

所述服饰检测分支层包括：池化层和全连接层；

所述服饰检测分支层用于基于所述人体预测关键点以及所述服饰特征检测所述待检测图像中服饰的位置信息。

在本发明的再一种实施方式中，所述根据所述服饰特征和所述人体预测关键点进行服饰检测，包括：

基于所述人体预测关键点生成至少一个备选框，所述备选框用于框定所述待检测图像中的各个人体预测关键点的区域；

根据所述服饰特征和所述至少一个备选框进行服饰检测，输出所述服饰检测结果；所述服饰检测结果包括：服饰位置信息和/或服饰分类信息。