[发明专利]人体属性检测模型的训练方法、装置、电子设备及介质

说明书

本公开提供了人体属性检测模型的训练方法、装置、电子设备及介质，涉及人工智能技术领域，具体涉及计算机视觉、深度学习等技术领域，可应用于智能云和安全巡检场景下。具体实现方案：获取与多种人体属性类别分别对应的正样本子图像和负样本子图像；确定与多个正样本子图像分别对应的多个第一标注属性；和与多个负样本子图像分别对应的多个第二标注属性；及根据多个正样本子图像、多个负样本子图像、多个第一标注属性及多个第二标注属性训练人工智能模型，得到人体属性检测模型，使得训练得到的人体属性检测模型能够有效建模人体的细粒度的属性，能够提升人体属性检测模型对人体图像的特征表达能力，有效提升人体属性检测的准确性和检测效率。

技术领域

本公开涉及人工智能技术领域，具体涉及计算机视觉、深度学习等技术领域，可应用于智能云和安全巡检场景下，尤其涉及人体属性检测模型的训练方法、装置、电子设备及介质。

背景技术

人工智能是研究使计算机来模拟人的某些思维过程和智能行为(如学习、推理、思考、规划等)的学科，既有硬件层面的技术也有软件层面的技术。人工智能硬件技术一般包括如传感器、专用人工智能芯片、云计算、分布式存储、大数据处理等技术；人工智能软件技术主要包括计算机视觉技术、语音识别技术、自然语言处理技术以及机器学习/深度学习、大数据处理技术、知识图谱技术等几大方向。

相关技术中用于人体属性检测的模型，对用于识别的人体图像的特征表达能力不佳，从而影响人体属性检测的准确性。

发明内容

提供了一种人体属性检测模型的训练方法、人体属性识别方法、装置、电子设备、存储介质及计算机程序产品。

根据第一方面，提供了一种人体属性检测模型的训练方法，包括：获取与多种人体属性类别分别对应的多个样本图像；对所述多个样本图像分别进行检测，以得到与所述多种人体属性类别分别对应的多个正样本子图像和多个负样本子图像；根据所述多种人体属性类别，确定与所述多个正样本子图像分别对应的多个第一标注属性；根据所述多种人体属性类别，确定与所述多个负样本子图像分别对应的多个第二标注属性；以及根据所述多个正样本子图像、所述多个负样本子图像、所述多个第一标注属性以及所述多个第二标注属性训练初始的人工智能模型，以得到人体属性检测模型。

根据第二方面，提供了一种人体属性识别方法，包括：获取待测人体图像；将所述待测人体图像输入至如上述人体属性检测模型的训练方法训练得到的人体属性检测模型之中，以得到所述人体属性检测模型输出的目标人体属性。

根据第三方面，提供了一种人体属性检测模型的训练装置，包括：第一获取模块，用于获取与多种人体属性类别分别对应的多个样本图像；检测模块，用于对所述多个样本图像分别进行检测，以得到与所述多种人体属性类别分别对应的多个正样本子图像和多个负样本子图像；第一确定模块，用于根据所述多种人体属性类别，确定与所述多个正样本子图像分别对应的多个第一标注属性；第二确定模块，用于根据所述多种人体属性类别，确定与所述多个负样本子图像分别对应的多个第二标注属性；以及训练模块，用于根据所述多个正样本子图像、所述多个负样本子图像、所述多个第一标注属性以及所述多个第二标注属性训练初始的人工智能模型，以得到人体属性检测模型。

根据第四方面，提供了一种人体属性识别装置，包括：第二获取模块，用于获取待测人体图像；识别模块，用于将所述待测人体图像输入至如上述人体属性检测模型的训练装置训练得到的人体属性检测模型之中，以得到所述人体属性检测模型输出的目标人体属性。

根据第五方面，提供了一种电子设备，包括：至少一个处理器；以及与所述至少一个处理器通信连接的存储器；其中，所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行本公开实施例的人体属性检测模型的训练方法，或者执行本公开实施例的人体属性识别方法。