CN111832383B - 姿态关键点识别模型的训练方法、姿态识别方法及装置 - Google Patents
姿态关键点识别模型的训练方法、姿态识别方法及装置 Download PDFInfo
- Publication number
- CN111832383B CN111832383B CN202010383035.3A CN202010383035A CN111832383B CN 111832383 B CN111832383 B CN 111832383B CN 202010383035 A CN202010383035 A CN 202010383035A CN 111832383 B CN111832383 B CN 111832383B
- Authority
- CN
- China
- Prior art keywords
- feature
- feature map
- loss
- key point
- thermodynamic diagram
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
- 238000012549 training Methods 0.000 title claims abstract description 100
- 238000000034 method Methods 0.000 title claims abstract description 79
- 238000010586 diagram Methods 0.000 claims abstract description 227
- 238000012545 processing Methods 0.000 claims abstract description 112
- 238000012937 correction Methods 0.000 claims abstract description 39
- 238000001514 detection method Methods 0.000 claims abstract description 19
- 208000037170 Delayed Emergence from Anesthesia Diseases 0.000 claims description 61
- 230000006870 function Effects 0.000 claims description 42
- 238000004364 calculation method Methods 0.000 claims description 27
- 230000008569 process Effects 0.000 claims description 19
- 238000004590 computer program Methods 0.000 claims description 12
- 238000002372 labelling Methods 0.000 claims description 11
- 238000013507 mapping Methods 0.000 claims description 8
- 238000003860 storage Methods 0.000 claims description 8
- 238000005070 sampling Methods 0.000 claims description 7
- 230000004044 response Effects 0.000 claims description 6
- 238000012163 sequencing technique Methods 0.000 claims description 2
- 238000005520 cutting process Methods 0.000 abstract description 2
- 230000036544 posture Effects 0.000 description 7
- 230000006399 behavior Effects 0.000 description 4
- 210000003423 ankle Anatomy 0.000 description 3
- 238000013528 artificial neural network Methods 0.000 description 3
- 238000013527 convolutional neural network Methods 0.000 description 3
- 230000000007 visual effect Effects 0.000 description 3
- 230000004913 activation Effects 0.000 description 2
- 210000003414 extremity Anatomy 0.000 description 2
- 210000003127 knee Anatomy 0.000 description 2
- 230000004048 modification Effects 0.000 description 2
- 238000012986 modification Methods 0.000 description 2
- 238000010606 normalization Methods 0.000 description 2
- 230000003287 optical effect Effects 0.000 description 2
- 238000011176 pooling Methods 0.000 description 2
- 238000007781 pre-processing Methods 0.000 description 2
- 210000000707 wrist Anatomy 0.000 description 2
- 230000005540 biological transmission Effects 0.000 description 1
- 210000000988 bone and bone Anatomy 0.000 description 1
- 238000004422 calculation algorithm Methods 0.000 description 1
- 230000008859 change Effects 0.000 description 1
- 230000007423 decrease Effects 0.000 description 1
- 238000009826 distribution Methods 0.000 description 1
- 230000000694 effects Effects 0.000 description 1
- 238000005516 engineering process Methods 0.000 description 1
- 238000000605 extraction Methods 0.000 description 1
- 238000005286 illumination Methods 0.000 description 1
- 230000006872 improvement Effects 0.000 description 1
- 238000004519 manufacturing process Methods 0.000 description 1
- 238000004904 shortening Methods 0.000 description 1
- 230000003238 somatosensory effect Effects 0.000 description 1
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V40/00—Recognition of biometric, human-related or animal-related patterns in image or video data
- G06V40/20—Movements or behaviour, e.g. gesture recognition
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/40—Extraction of image or video features
- G06V10/46—Descriptors for shape, contour or point-related descriptors, e.g. scale invariant feature transform [SIFT] or bags of words [BoW]; Salient regional features
- G06V10/462—Salient features, e.g. scale invariant feature transforms [SIFT]
- G06V10/464—Salient features, e.g. scale invariant feature transforms [SIFT] using a plurality of salient features, e.g. bag-of-words [BoW] representations
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V40/00—Recognition of biometric, human-related or animal-related patterns in image or video data
- G06V40/10—Human or animal bodies, e.g. vehicle occupants or pedestrians; Body parts, e.g. hands
-
- Y—GENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
- Y02—TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
- Y02T—CLIMATE CHANGE MITIGATION TECHNOLOGIES RELATED TO TRANSPORTATION
- Y02T10/00—Road transport of goods or passengers
- Y02T10/10—Internal combustion engine [ICE] based vehicles
- Y02T10/40—Engine management systems
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- General Health & Medical Sciences (AREA)
- Health & Medical Sciences (AREA)
- Multimedia (AREA)
- Computing Systems (AREA)
- Software Systems (AREA)
- Evolutionary Computation (AREA)
- Computational Linguistics (AREA)
- Molecular Biology (AREA)
- Biophysics (AREA)
- General Engineering & Computer Science (AREA)
- Biomedical Technology (AREA)
- Mathematical Physics (AREA)
- Data Mining & Analysis (AREA)
- Artificial Intelligence (AREA)
- Human Computer Interaction (AREA)
- Life Sciences & Earth Sciences (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Psychiatry (AREA)
- Social Psychology (AREA)
- Image Analysis (AREA)
Abstract
本发明实施例公开了一种姿态关键点识别模型的训练方法、姿态识别方法及装置,其中,基于第一损失函数和第二损失函数,根据训练图像训练姿态关键点识别模型的特征金字塔网络和矫正网络,并在第一损失和第二损失满足预定条件时,确定姿态关键点识别模型,将经过目标检测后裁剪获取的目标区域图像输入至训练好的姿态关键点识别模型中进行处理,得到各关键点对应的热力图,根据各关键点对应的热力图确定目标对象的姿态,由此,本实施例通过采用包括特征金字塔网络和矫正网络的姿态关键识别模型,提高了姿态关键点识别的准确性。
Description
技术领域
本发明涉及计算机技术领域,更具体地,涉及一种姿态关键点识别模型的训练方法、姿态识别方法及装置。
背景技术
关键点识别是指从一张含有目标对象的图像中对该目标对象的关键点进行建模识别。例如,人体关键点一般定义为:踝关节,左膝关节,左臀部,右臀部,左膝关节,左踝关节,右踝关节,上颈部,头顶,右手腕,左肘,左肩,右肩,右肘,左手腕等。人体具有柔性特征,当人体处于不同的动作时,会出现各种姿态和形状,人体任何一个部位的微小变化都会产生一种新的姿态,同时其关键点的可见性受穿着、姿态、视角等影响非常大,而且还可能面临着遮挡、光照、雾等环境的影响,除此之外,2D人体关键点和3D人体关键点在视觉上会有明显的差异,身体不同部位都会有视觉上缩短的效果(foreshortening),使得人体骨骼关键点检测成为计算机视觉领域中一个极具挑战性的问题。
发明内容
有鉴于此,本发明实施例提供一种姿态关键点识别模型的训练方法、姿态识别方法及装置,以提高姿态关键点识别的准确性。
第一方面,本发明实施例提供一种姿态关键点识别模型的训练方法,所述姿态关键点识别模型包括特征金字塔网络和矫正网络,所述方法包括:
获取训练样本,所述训练样本包括多组含有目标对象的训练图像以及对应的姿态关键点标注图像;
将所述训练图像输入至所述特征金字塔网络进行处理,获取所述训练图像对应的多个不同尺度的第一特征图;
对所述多个不同尺度的第一特征图依次进行加和处理,生成至少一个第二特征图;
基于第一损失函数,根据所述至少一个第二特征图与所述训练图像对应的标准关键点热力图计算第一损失,所述标准关键点热力图根据所述训练图像对应的姿态关键点标注图像获取;
根据所述第一损失调节所述特征金字塔网络的参数;
将所述至少一个第二特征图输入至所述矫正网络进行处理,生成第三特征图;
基于第二损失函数,根据所述第三特征图与所述训练图像对应的标准关键点热力图计算第二损失;
根据所述第二损失调节所述特征金字塔网络的参数和所述矫正网络的参数;
响应于所述第一损失和第二损失满足预定条件,确定所述姿态关键点识别模型。
第二方面,本发明实施例提供一种姿态识别方法,所述方法包括:
获取待处理图像;
对所述待处理图像进行目标检测,获取目标对象区域图像;
将所述目标区域图像输入至姿态关键点识别模型的特征金字塔网络中进行处理,获取所述目标区域图像对应的多个不同尺度的第一特征图;
对所述多个不同尺度的第一特征图依次进行加和处理,生成至少一个第二特征图;
根据所述至少一个第二特征图确定所述目标区域图像对应的初始热力图;
将所述至少一个第二特征图和所述初始热力图输入至所述姿态关键点识别模型的矫正网络中进行处理,以对所述初始热力图进行矫正,确定目标热力图;
将所述目标热力图映射至所述待处理图像中,确定目标对象的姿态。
第三方面,本发明实施例提供一种姿态关键点识别模型的训练装置,所述姿态关键点识别模型包括特征金字塔网络和矫正网络,所述装置包括:
样本获取单元,被配置为获取训练样本,所述训练样本包括多组含有目标对象的训练图像以及对应的姿态关键点标注图像;
第一处理单元,被配置为将所述训练图像输入至所述特征金字塔网络进行处理,获取所述训练图像对应的多个不同尺度的第一特征图;
第二处理单元,被配置为对所述多个不同尺度的第一特征图依次进行加和处理,生成至少一个第二特征图;
第一损失计算单元,被配置为基于第一损失函数,根据所述至少一个第二特征图与所述训练图像对应的标准关键点热力图计算第一损失,所述标准关键点热力图根据所述训练图像对应的姿态关键点标注图像获取;
第一参数调节单元,被配置为根据所述第一损失调节所述特征金字塔网络的参数;
第三处理单元,被配置为将所述至少一个第二特征图输入至所述矫正网络进行处理,生成第三特征图;
第二损失计算单元,被配置为基于第二损失函数,根据所述第三特征图与所述训练图像对应的标准关键点热力图计算第二损失;
第二参数调节单元,被配置为根据所述第二损失调节所述特征金字塔网络的参数和所述矫正网络的参数;
模型确定单元,被配置为响应于所述第一损失和第二损失满足预定条件,确定所述姿态关键点识别模型。
第四方面,本发明实施例提供一种姿态识别装置,所述装置包括:
待处理图像获取单元,被配置为获取待处理图像;
目标检测单元,被配置为对所述待处理图像进行目标检测,获取目标对象区域图像;
第一特征处理单元,被配置为将所述目标区域图像输入至姿态关键点识别模型的特征金字塔网络中进行处理,获取所述目标区域图像对应的多个不同尺度的第一特征图;
第二特征处理单元,被配置为对所述多个不同尺度的第一特征图依次进行加和处理,生成至少一个第二特征图;
初始热力图确定单元,被配置为根据所述至少一个第二特征图确定所述目标区域图像对应的初始热力图;
目标热力图确定单元,被配置为将所述至少一个第二特征图和所述初始热力图输入至所述姿态关键点识别模型的矫正网络中进行处理,以对所述初始热力图进行矫正,确定目标热力图;
映射单元,被配置为将所述目标热力图映射至所述待处理图像中,确定目标对象的姿态。
第五方面,本发明实施例提供一种电子设备,包括存储器和处理器,所述存储器用于存储一条或多条计算机程序指令,其中,所述一条或多条计算机程序指令被所述处理器执行以实现如本发明实施例第一方面所述的方法和/或本发明实施例第二方面所述的方法。
第六方面,本发明实施例提供一种计算机可读存储介质,其上存储计算机程序指令,所述计算机程序指令在被处理器执行时以实现如本发明实施例第一方面所述的方法和/或本发明实施例第二方面所述的方法。
本发明实施例基于第一损失函数和第二损失函数,根据训练图像训练姿态关键点识别模型的特征金字塔网络和矫正网络,并在第一损失和第二损失满足预定条件时,确定姿态关键点识别模型,将经过目标检测后裁剪获取的目标区域图像输入至训练好的姿态关键点识别模型中进行处理,得到各关键点对应的热力图,根据各关键点对应的热力图确定目标对象的姿态,由此,本实施例通过采用包括特征金字塔网络和矫正网络的姿态关键识别模型,提高了姿态关键点识别的准确性。
本发明的各种可行实施例及其技术优势将在下文中详细说明。
附图说明
通过以下参照附图对本发明实施例的描述,本发明的上述以及其它目的、特征和优点将更为清楚,在附图中:
图1是本发明实施例的姿态关键点识别模型的训练方法的流程图;
图2是本发明实施例的姿态关键点识别模型的训练过程示意图;
图3是本发明实施例的Bottleneck的示意图;
图4是本发明实施例的Dilated Bottleneck的示意图;
图5是本发明实施例的姿态识别方法的流程图;
图6是本发明实施例的姿态识别过程示意图;
图7是相关技术的目标对象姿态识别结果的示意图;
图8是本发明实施例的目标对象识别结果的示意图;
图9是本发明实施例的姿态关键点识别模型的训练装置的示意图;
图10是本发明实施例的姿态识别装置的示意图;
图11是本发明实施例的电子装置的示意图。
具体实施方式
以下基于实施例对本发明进行描述,但是本发明并不仅仅限于这些实施例。在下文对本发明的细节描述中,详尽描述了一些特定的细节部分。对本领域技术人员来说没有这些细节部分的描述也可以完全理解本发明。为了避免混淆本发明的实质,公知的方法、过程、流程、元件和电路并没有详细叙述。
此外,本领域普通技术人员应当理解,在此提供的附图都是为了说明的目的,并且附图不一定是按比例绘制的。
除非上下文明确要求,否则在说明书的“包括”、“包含”等类似词语应当解释为包含的含义而不是排他或穷举的含义;也就是说,是“包括但不限于”的含义。
在本发明的描述中,需要理解的是,术语“第一”、“第二”等仅用于描述目的,而不能理解为指示或暗示相对重要性。此外,在本发明的描述中,除非另有说明,“多个”的含义是两个或两个以上。
现有解决人体图像关键点识别的人体关键点检测算法基本上是在几何先验的基础上基于模版匹配的思路来进行,核心就在于如何去用模版表示整个人体结构,包括关键点的表示、肢体结构的表示以及不同肢体结构之间的关系的表示。由于人体的姿态变化性较大,采用模版匹配的方法的关键点识别准确性较低。由此,本发明实施例提供一种基于神经网络的姿态关键点识别模型的训练方法,以及基于该姿态关键点识别模型的姿态识别方法,以提高姿态关键点识别的准确性,进而提供姿态识别的准确性。
图1是本发明实施例的姿态关键点识别模型的训练方法的流程图。如图1所示,本实施例的姿态关键点识别模型的训练方法包括以下步骤:
步骤S110,获取训练样本,所述训练样本包括多组含有目标对象的训练图像以及对应的姿态关键点标注图像。在一种可选的实现方式中,本实施例的训练样本可以采用MSCOCO数据集。应理解,其他具有目标标注的数据集也可应用于本实施例中,本实施例并不对此进行限制。以下训练步骤以一个训练图像的处理过程为例进行说明。
步骤S120,将训练图像输入至特征金字塔网络进行处理,获取该训练图像对应的多个不同尺度的第一特征图。在本实施例中,姿态关键点识别模型包括特征金字塔网络和矫正网络,特征金字塔网络用于生成表征关键点的初始热力图,矫正网络用于对初始热力图进行矫正,获取目标热力图。
在神经网络中,随着网络的加深,空间分辨率降低,输出的特征图具有更多的图像局部语义信息,为了有效权衡空间分辨率和图像语义信息的平衡,本实施例采用特征金字塔网络(Feature Pyramid Networks,FPN)。特征金字塔网络使得各分辨率的特征图与其后续分辨率的特征图进行加和操作,以使得相应层输出的特征图融合了不同分辨率、不同语义强度的特征,由此,可以在不增加额外计算时间和计算量的情况下,有效权衡空间分辨率和图像语义信息的平衡。在一种可选的实现方式中,姿态关键点识别模型以残差网络ResNet-50或ResNet-101为骨干网络。可选的,残差网络ResNet-50或ResNet-101经过ImageNet数据集预训练。
ResNet(Residual Neural Network)的主要思想是在网络中增加了直连通道,允许原始输入信息直接传输到后面的网络层中,使得这一网络层可以不用学习前一层的整个输出,而是学习上一个网络层输出的残差。ResNet通过将输出信息直接绕道传输到输出,保护了信息的完整性,并且整个网络只需要学习输入、输出的差别部分,简化了学习目标和难度。ResNet有不同的层数,例如,ResNet-50(50层的ResNet)、ResNet-101(101层的ResNet)、ResNet-152(152层的ResNet)等,其均可应用于本实施例中。
可选的,特征金字塔网络包括多个残差块,每个残差块输出对应的第一特征图。特征金字塔网络中残差块的数量可以根据实际情况进行设置,本实施例并不对此进行限制。其中,残差块的结构为卷积神经网络结构,包括归一化层、激活层和卷积层,其中,卷积层的层数和卷积层中滤波器的个数可以根据实际情况进行设置。可选的,卷积层滤波器的大小、补偿和填充分别为3*3、1、1。
步骤S130,对各第一特征图依次进行加和处理,生成多个第二特征图。在一种可选的实现方式中,步骤S130具体可以包括:
将当前层残差块输出的第一特征图与加和特征图进行加和处理,获取对应的第二特征图,所述加和特征图为当前层之后的各层残差块输出的第一特征图的加和处理结果、或最后一层残差块输出的第一特征图,以及根据最后一层残差块输出的第一特征图确定对应的第二特征图。
步骤S140,基于第一损失函数,根据至少一个第二特征图与该训练图像对应的标准关键点热力图计算第一损失。其中,标准关键点热力图根据所述训练图像对应的姿态关键点标注图像的热力图预先处理获取。
在一种可选的实现方式中,步骤S140具体可以包括:
A1:对于各第二特征图,对该第二特征图进行卷积处理,获取该训练图像的各关键点的第一热力图。
A2:对于各关键点,根据该关键点的至少一个第一热力图分别与对应的标准关键点热力图计算关键点的第一关键点损失。
A3:根据各关键点的第一关键点损失确定第一损失。
在本实施例中,对一个第二特征图进行卷积处理,可以获取一组关键点的第一热力图。可选的,假设步骤S130获取n个(n大于等于1)第二特征,步骤S140可以获取n组关键点的第二热力图,也即每个关键点具有对应的n个第一热力图,计算关键点的n个第一热力图分别与该关键点的标准关键点热力图的n个损失,计算n个损失的和或平均值以确定该关键点的第一关键点损失,根据各关键点的第一关键点损失确定第一损失。可选的,第一损失可以为包括各关键点损失的向量或数组。
步骤S150,根据第一损失调节特征金字塔网络的参数。
在一种可选的实现方式中,第一损失函数为L2损失函数:
L2=Ex,y~P(X,Y)||Fgenerate(x)-y||2
其中,x为输入的训练图像,y为该训练图像的各关键点对应的标准关键点热力图,Fgenerate(x)为特征金字塔网络输出的各第二特征图对应的第一热力图,Ex,y~P(X,Y)为各第二特征图对应的关键点的概率分布。
图2是本发明实施例的姿态关键点识别模型的训练过程示意图。如图2所示,本实施例的特征金字塔网络2包括第一卷积层21、第一残差块22、第二残差块23、第三残差块24和第四残差块25。将训练图像输入至特征金字塔网络2中,经第一卷积层21处理后输出初始特征图C1。例如输入的训练图像为经过裁剪后的人体图像,大小为384*288,第一卷积层21对训练图像进行卷积处理和最大池化处理后输出96*72*64的初始特征图C1,其中,在第一卷积层21中,卷积核的大小为7*7、通道数为64、填充为3、步长为2,池化核的大小为3*3、填充为0、步长为2。将初始特征图C1依次通过第一残差块22、第二残差块23、第三残差块24和第四残差块25,分别输出不同尺度的特征图C2、C3、C4和C5。以上述训练为例,特征图C2、C3、C4和C5的尺度大小分别为96*72*256,48*36*512,24*18*1024,12*9*2014。
进一步地,对特征图C2、C3、C4和C5进行1*1的卷积处理得到第一特征图C2'、C3'、C4'和C5',对第一特征图C2'、C3'、C4'和C5'依次进行加和处理,生成多个第二特征图C2”、C3”、C4”和C5”。可选的,对Ci′(i=5,4,3)进行1*1的卷积处理后进行上采样处理,将上采样后的特征图与上一层残差块对应的第一特征图Ci-1′依次进行加和操作,得到第二特征图Ci-1″。其中,将第一特征图C5'和C4'进行加和处理后得到C4”,将第一特征图C5'、C4'和C3'进行加和处理后得到C3”(也即将C4”上采样后与C3'加和处理得到C3”),将第一特征图C5'、C4'、C3'和C2'进行加和处理后得到C3”(也即将C3”上采样后与C2'加和处理得到C2”)。上采样处理(unsample)使得第一特征图Ci′的分辨率与第一特征图Ci-1′相同,以可以进行加和操作。其中,对第一特征图C5'进行卷积处理等操作后获取第二特征图C5”。根据第二特征图Cj″(j=5,4,3,2),获取第二特征图Cj″对应的多个关键点的热力图,基于L2损失函数,根据各关键点的热力图与训练图像的各关键点的标准关键点热力图计算第一损失,根据第一损失调节特征金字塔网络2的参数。其中,标准关键点热力图根据所述训练图像对应的姿态关键点标注图像预先处理获取。也即,对于每个第二特征图,计算一个关键点对应的热力图与该关键点的标准关键点热力图的损失,根据各关键点对应的损失计算第一损失。
在MSCOCO数据集中,人体图像一般取17个关键点,本实施例以人体图像的17个关键点为例,在一种可选的实现方式中,对第二特征图Cj″(j=5,4,3,2)分别进行1*1的卷积处理后,再进行3*3的卷积处理,获取第二特征图Cj″对应的17个热力图,每个热力图对应1个人体关键点,然后计算第二特征图Cj″对应的17个热力图分别与17个标准关键点热力图的损失,获得第二特征图Cj″对应的各关键点损失,根据各第二特征图对应的关键点损失获得第一损失。
容易理解,图2中各特征图的大小仅仅是示例性的,其并不与实际大小相对应。应理解,在本实施例中,各深层残差块输出的第一特征图与其前一层残差块输出的第一特征图(或与之前的加和特征图)加和后输出一个第二特征图,因此,本实施例输出多个第二特征图(例如C2”-C5”),应理解,也可以将各第一特征图全部加和后输出一个第二特征图(例如C2”),也可以间隔输出多个第二特征图(例如C2”、C4”),本实施例并不对此进行限制。
步骤S160,将至少一个第二特征图输入至矫正网络进行处理,生成第三特征图。在一种可选的实现方式中,矫正网络由多个瓶颈层和上采样层组成。瓶颈层可以为Bottleneck或Dilated Bottleneck(膨胀型的瓶颈层)。
可选的,步骤S160具体包括:将至少一个第二特征图输入对应的瓶颈层和上采样层进行处理,获取对应的第四特征图,将各第四特征图进行合并连接(concat)处理,生成所述对应的第三特征图。由此,本实施例通过加入瓶颈层减少参数的数量,以减小计算量,提高识别效率。
应理解,在同一训练迭代周期内,步骤S160和步骤S140-S150没有前后的执行顺序。
步骤S170,基于第二损失函数,根据第三特征图与该训练图像对应的标准关键点热力图计算第二损失。可选的,第二损失函数为L2损失函数。
在一种可选的实现方式中,步骤S170包括:
B1:对第三特征图进行卷积处理,获取训练图像的各关键点的第二热力图。
B2:对于各所述关键点,根据所述关键点的第二热力图与对应的标准关键点热力图计算所述关键点的第二关键点损失。
B3:对各第二关键点损失进行排序。
B4:根据关键点损失排序结果选择预定个第二关键点损失确定所述第二损失。
可选的,在本实施例的矫正网络中,选择关键点损失较大的预定个关键点损失计入矫正网络的第二损失函数,由此,使得矫正网络针对误差较大的关键点进行矫正,在提高关键点识别准确性的同时提高了识别效率。
步骤S180,根据第二损失调节特征金字塔网络的参数和所述矫正网络的参数。
步骤S190,响应于所述第一损失和第二损失满足预定条件,确定所述姿态关键点识别模型。在一种可选的实现方式中,所述预定条件为所述第一损失小于第一预定值,且所述第二损失小于第二预定值。在另一种可选的实现方式中,预定条件还可以包括训练过程的迭代次数达到预定次数等。例如所述预定条件为训练迭代次数达到预定值,所述第一损失小于第一预定值,且所述第二损失小于第二预定值。
如图2所示,矫正网络3包括Bottleneck层31、Dilated Bottleneck层32、DilatedBottleneck层33以及上采样层34、上采样层35和上采样层36。
图3是本发明实施例的Bottleneck层的示意图。图4是本发明实施例的DilatedBottleneck层的示意图。如图3所示,Bottleneck层包括路径1和路径2,将路径1和路径2的处理结果经过线性整流后输出。其中,路径1包括1*1卷积层、3*3卷积层和1*1卷积层。路径2不对输入特征图进行处理。其中,Bottleneck层中的1*1卷积层用于降低和升高特征维度,以减少参数的数量,从而可以减小计算量,并且,在降低维度后可以更加有效地进行数据的训练和特征提取。
如图4所示,Dilated Bottleneck层包括路径3和路径4,将路径3和路径4的处理结果经过线性整流后输出。其中,路径3包括1*1卷积层、3*3的膨胀卷积层和1*1卷积层。路径4包括1*1卷积层。由此,本实施例通过引入Dilated Bottleneck层可以在可以减小计算量的同时,有效地扩大感受野,提高了识别准确性。
在一种可选的实现方式中,根据第二特征图C2”获取第四特征图F2,第二特征图C3”经过Bottleneck层31和上采样层34处理,获取第四特征图F3,第二特征图C4”经过Dilated Bottleneck层32和上采样层35处理,获取第四特征图F4,第二特征图C5”经过Dilated Bottleneck 33和上采样层36处理,获取第四特征图F5。容易理解,在特征图的尺度较小,感受野较小时,可以采用Dilated Bottleneck层,由例如第二特征图C4”和第二特征图C5”。其中,通过上采样层34-36使得第四特征图F3-F5的分辨率大小与F2相同。
应理解,由于第二特征图C2”-C5”的分辨率大小不同,因此,采用的瓶颈层和上采样层的数量不同,具体数量可以根据实际情况设置,本实施例并不对此进行限制。在本实施例中,第二特征图C2”采用0个瓶颈层和0个上采样层。
在本实施例中,如图2所示,对第四特征图F2-F5进行concat连接处理,并将concat连接处理后的特征图经过Bottleneck层37处理得到第三特征图。将第三特征图进行1*1的卷积处理和3*3的卷积处理,获取第三特征图对应的各关键点的热力图,基于第二损失函数(例如L2损失函数),根据第三特征图对应的各关键点的热力图和各标准关键点热力图计算各关键点对应的第二关键点损失,对各第二关键点损失进行排序,例如按照第二关键点损失从大到小进行排序,根据关键点损失排序结果选择预定个第二关键点损失确定所述第二损失。例如,以人体的17个关键点为例,根据关键点损失排序结果选择9个最大的第二关键点损失计入第二损失计算。应理解,第二关键点损失的选取个数根据实际情况进行设置,本实施例并不对此进行限制。在本实施例中,通过在矫正网络中,选择关键点损失较大的预定个第二关键点损失计入矫正网络的第二损失函数,使得矫正网络针对误差较大的关键点进行矫正,在提高关键点识别准确性的同时提高了识别效率。
本发明实施例基于第一损失函数和第二损失函数,根据训练图像训练姿态关键点识别模型的特征金字塔网络和矫正网络,并在第一损失和第二损失满足预定条件时,确定姿态关键点识别模型,本实施例通过采用包括特征金字塔网络和矫正网络的姿态关键识别模型,提高了姿态关键点识别的准确性。
图5是本发明实施例的姿态识别方法的流程图。如图5所示,本发明实施例的姿态识别方法包括以下步骤:
步骤S210,获取待处理图像。其中,待处理图像包括至少一个目标对象,例如,目标对象为行人,则待处理图像中包括至少一个行人。
步骤S220,对待处理图像进行目标检测,获取目标对象区域图像。在一种可选的实现方式中,采用预定的目标检测模型对待处理图像进行目标检测,获取目标对象区域图像。其中,目标检测模型可以为Faster R-CNN模型、SSD模型或YOLO模型等,本实施例并不对此进行限制。
步骤S230,将目标区域图像输入至姿态关键点识别模型的特征金字塔网络中进行处理,获取所述目标区域图像对应的多个不同尺度的第一特征图。在一种可选的实现方式中,姿态关键点识别模型以Resnet-101或Resnet-50为骨干网络。可选的,特征金字塔网络根据L2损失函数预先训练。
特征金字塔网络包括多个残差块,每个残差块对应输出一个尺度的第一特征图。特征金字塔网络中残差块的数量可以根据实际情况进行设置,本实施例并不对此进行限制。其中,残差块的结构为卷积神经网络结构,包括归一化层、激活层和卷积层,其中,卷积层的层数和卷积层中滤波器的个数可以根据实际情况进行设置。在本实施例中,姿态关键点识别模型通过上述实施例中的训练方法进行训练,在此不再赘述。
步骤S240,对多个不同尺度的第一特征图依次进行加和处理,生成至少一个第二特征图。
将当前层残差块输出的第一特征图与加和特征图进行加和处理,获取对应的第二特征图,所述加和特征图为当前层之后的各层残差块输出的第一特征图的加和处理结果、或最后一层残差块输出的第一特征图,以及根据最后一层残差块输出的第一特征图确定对应的第二特征图,具体方法可参照图2中关于加和处理的说明,在此不再赘述。
本实施例采用特征金字塔网络(Feature Pyramid Networks,FPN)。特征金字塔网络使得各分辨率的特征图与后续分辨率的特征图进行加和操作,以使得相应层输出的特征图融合了不同分辨率、不同语义强度的特征,由此,可以在不增加额外计算时间和计算量的情况下,有效权衡空间分辨率和图像语义信息的平衡。
步骤S250,根据至少一个第二特征图确定所述目标区域图像对应的初始热力图。可选的,将第二特征图经过1*1卷积处理和3*3卷积处理后,获取各关键点的关键点热力图。由于每个第二特征图对应一组各关键点的关键点热力图,也即,每个关键点对应多个关键点热力图,因此可以根据关键点对应的多个关键点热力图确定该关键点对应的初始热力图。也即,目标区域图像对应的初始热力图包括目标区域图像对应的多个关键点的初始热力图,例如若目标对象为人体,预置有17个关键点,则可根据各第二特征图确定这17个关键点分别对应的初始热力图。
步骤S260,将至少一个第二特征图和初始热力图输入至姿态关键点识别模型的矫正网络中进行处理,以对初始热力图进行矫正,确定目标热力图。在一种可选的实现方式中,矫正网络包括多个瓶颈层和上采样层。多个瓶颈层可以包括Bottleneck层和DilatedBottleneck层,瓶颈层的结构图如图3和图4所示,其中,瓶颈层中的1*1卷积层用于降低和升高特征维度,以减少参数的数量,从而可以减小计算量,并且,在降低维度后可以更加有效地进行数据的训练和特征提取。同时,通过引入Dilated Bottleneck层可以在可以减小计算量的同时,有效地扩大感受野,提高了识别准确性。可选的,矫正网络根据L2损失函数预先训练。
在一种可选的实现方式中,步骤S260包括:
C1:将至少一个第二特征图输入至对应的瓶颈层和上采样层进行处理,确定所述目标区域图像对应的第三特征图。可选的,将至少一个第二特征图输入对应的瓶颈层和上采样层进行处理,获取对应的第四特征图,将各第四特征图进行合并连接(concat)处理,生成目标区域图像对应的第三特征图。
C2:根据第三特征图确定预定个关键点对应的矫正热力图。可选的,将第三特征图经过1*1的卷积处理和3*3的卷积处理,获取第三特征图对应的预定个关键点的矫正热力图。容易理解,这里关键点的选取个数根据矫正网络的训练过程中设置的关键点损失选取个数确定。
C3:根据初始热力图和矫正热力图确定目标热力图。具体地,根据各关键点对应的矫正热力图对对应的初始热力图进行矫正,确定各关键点的目标热力图。在本实施例中,通过对部分误差较大的关键点的初始热力图进行矫正,在提高关键点识别准确性的同时提高了识别效率。
步骤S270,将目标热力图映射至待处理图像中,确定目标对象的姿态。在本实施例中,将识别的目标区域图像的各关键点的目标热力图映射至待处理图像中,连接各关键点以识别目标对象的姿态。由此,本实施例可以较为准确地识别出的关键点,进而较为准确地确定目标对象的姿态。可选的,在一些应用场景中,可以根据目标对象的姿态确定用户的行为,进而判断用户行为是否符合当前要求,例如,可以通过对司机姿态的识别确定司机行为,判断司机在开车过程中是否存在不合规(玩手机等)行为等。
图6是本发明实施例的姿态识别过程示意图。如图6所示,获取待处理图像61,通过目标检测模型62对待处理图像61进行目标检测,获得目标区域图像63和64,将目标区域图像63和64输入至预先训练的姿态关键点识别模型65中,经特征金字塔网络651和矫正网络652处理后,输出目标区域图像63对应的各关键点的目标热力图66、以及目标区域图像64对应的各关键点的目标热力图67,将识别的目标区域图像63和64的各关键点的目标热力图映射至待处理图像61中,根据映射获得的图像68确定目标对象的姿态。其中,特征金字塔网络651和矫正网络652的具体处理过程与图2中的训练过程类似,在此不再赘述。
图7是相关技术的目标对象姿态识别结果的示意图。图8是本发明实施例的目标对象识别结果的示意图。如图7和图8所示,相比与相关技术,本发明实施例能够更准确的识别人体的关键点。
本实施例通过对待处理图像进行目标检测,获取目标对象区域图像,将目标区域图像输入至姿态关键点识别模型的特征金字塔网络中进行处理,获取所述目标区域图像对应的多个不同尺度的第一特征图,对各所述第一特征图依次进行加和处理,生成多个第二特征图,根据多个所述第二特征图确定所述目标区域图像对应的初始热力图,将多个所述第二特征图和所述初始热力图输入至所述姿态关键点识别模型的矫正网络中进行处理,以对所述初始热力图进行矫正,确定目标热力图,将所述目标热力图映射至所述待处理图像中,确定目标对象的姿态,由此,本实施例通过采用包括特征金字塔网络和矫正网络的姿态关键识别模型,提高了姿态关键点识别的准确性,从而提高目标对象姿态识别的准确性。
图9是本发明实施例的姿态关键点识别模型的训练装置的示意图。在本实施例中,所述姿态关键点识别模型包括特征金字塔网络和矫正网络。可选的,所述姿态关键点识别模型以Resnet-101或Resnet-50为骨干网络。
本实施例的训练装置9包括样本获取单元91、第一处理单元92、第二处理单元93、第一损失计算单元94、第一参数调节单元95、第三处理单元96、第二损失计算单元97、第二参数调节单元98和模型确定单元99。
样本获取单元91被配置为获取训练样本,所述训练样本包括多组含有目标对象的训练图像以及对应的姿态关键点标注图像。第一处理单元92被配置为将所述训练图像输入至所述特征金字塔网络进行处理,获取所述训练图像对应的多个不同尺度的第一特征图。第二处理单元93被配置为对各所述第一特征图依次进行加和处理,生成至少一个第二特征图。
在一种可选的实现方式中,所述特征金字塔网络包括多层残差块,各层残差块输出有对应的第一特征图。第二处理单元93包括第二特征图确定子单元931。第二特征图确定子单元931被配置为将当前层残差块输出的第一特征图与加和特征图进行加和处理,获取对应的第二特征图,所述加和特征图为当前层之后的各层残差块输出的第一特征图的加和处理结果、或最后一层残差块输出的第一特征图。第一特征图确定子单元931还被配置根据最后一层残差块输出的第一特征图确定对应的第二特征图。
第一损失计算单元94被配置为被配置为基于第一损失函数,根据所述至少一个第二特征图与所述训练图像对应的标准关键点热力图计算第一损失,所述标准关键点热力图根据所述训练图像对应的姿态关键点标注图像获取。在一种可选的实现方式中,第一损失计算单元包括第一热力图获取子单元941、第一关键损失计算子单元942以及第一损失计算子单元943。第一热力图获取子单元941被配置为对于各所述第二特征图,对所述第二特征图进行卷积处理,获取所述训练图像的各关键点的第一热力图。第一关键损失计算子单元942被配置为对于各所述关键点,根据所述关键点的至少一个第一热力图分别与对应的标准关键点热力图计算所述关键点的第一关键点损失。第一损失计算子单元943被配置为根据各所述关键点的第一关键点损失确定所述第一损失。
第一参数调节单元95被配置为根据所述第一损失调节所述特征金字塔网络的参数。在一种可选的实现方式中,所述第一损失函数为L2损失函数。
第三处理单元96被配置为将所述至少一个第二特征图输入至所述矫正网络进行处理,生成第三特征图。在一种可选的实现方式中,第三处理单元96包括第四特征图获取子单元961和第三特征图确定子单元962。第四特征图获取子单元961被配置为将所述至少一个第二特征图输入对应的瓶颈层和上采样层进行处理,获取对应的第四特征图。第三特征图确定子单元962被配置为将各第四特征图进行连接处理,生成所述对应的第三特征图。
第二损失计算单元97被配置为被配置为基于第二损失函数,根据所述第三特征图与所述训练图像对应的标准关键点热力图计算第二损失。可选的,第二损失函数为L2损失函数。在一种可选的实现方式中,第二损失计算单元97包括第二热力图获取子单元971、第二关键点损失计算子单元972、排序子单元973和第二损失确定子单元974。第二热力图获取子单元971被配置为对所述第三特征图进行卷积处理,获取所述训练图像的各关键点的第二热力图。第二关键点损失计算子单元972被配置为对于各所述关键点,根据所述关键点的第二热力图与对应的标准关键点热力图计算所述关键点的第二关键点损失。排序子单元973被配置为对各所述第二关键点损失进行排序。第二损失确定子单元974被配置为根据关键点损失排序结果选择预定个第二关键点损失确定所述第二损失。
第二参数调节单元98被配置为根据所述第二损失调节所述特征金字塔网络的参数和所述矫正网络的参数。模型确定单元99被配置为响应于所述第一损失和第二损失满足预定条件,确定所述姿态关键点识别模型。在一种可选的实现方式中,所述预定条件为所述第一损失小于第一预定值,所述第二损失小于第二预定值。
本发明实施例基于第一损失函数和第二损失函数,根据训练图像训练姿态关键点识别模型的特征金字塔网络和矫正网络,并在第一损失和第二损失满足预定条件时,确定姿态关键点识别模型,本实施例通过采用包括特征金字塔网络和矫正网络的姿态关键识别模型,提高了姿态关键点识别的准确性。
图10是本发明实施例的姿态识别装置的示意图。如图10所示,本发明实施例的姿态识别装置10包括待处理图像获取单元101、目标检测单元102、第一特征处理单元103、第二特征处理单元104、初始热力图确定单元105、目标热力图确定单元106以及映射单元107。
待处理图像获取单元101被配置为获取待处理图像。目标检测单元102被配置为对所述待处理图像进行目标检测,获取目标对象区域图像。
第一特征处理单元103被配置为将所述目标区域图像输入至姿态关键点识别模型的特征金字塔网络中进行处理,获取所述目标区域图像对应的多个不同尺度的第一特征图。可选的,所述姿态关键点识别模型以Resnet-101或Resnet-50为骨干网络。可选的,所述特征金字塔网络的损失函数为L2损失函数。
第二特征处理单元104被配置为对所述多个不同尺度的第一特征图依次进行加和处理,生成至少一个第二特征图。在一种可选的实现方式中,所述特征金字塔网络包括多层残差块,各层残差块输出有对应的第一特征图。第二特征处理单元104包括第二特征处理子单元1041。第二特征处理子单元1041被配置为将当前层残差块输出的第一特征图与加和特征图进行加和处理,获取对应的第二特征图,所述加和特征图为当前层之后的各层残差块输出的第一特征图的加和处理结果、或最后一层残差块输出的第一特征图。第二特征处理子单元1041还被配置为根据最后一层残差块输出的第一特征图确定对应的第二特征图。
初始热力图确定单元105被配置为根据所述至少一个第二特征图确定所述目标区域图像对应的初始热力图。目标热力图确定单元106被配置为将所述至少一个第二特征图和所述初始热力图输入至所述姿态关键点识别模型的矫正网络中进行处理,以对所述初始热力图进行矫正,确定目标热力图。可选的,矫正网络的损失函数为L2损失函数。
在一种可选的实现方式中,目标热力图确定单元106包括第三特征图确定子单元1061、矫正热力图确定子单元1062和目标热力图确定子单元1063。第三特征图确定子单元1061被配置为将所述至少一个第二特征图输入至对应的瓶颈层和上采样层进行处理,确定所述目标区域图像对应的第三特征图。矫正热力图确定子单元1062被配置为根据所述第三特征图确定预定个关键点对应的矫正热力图。目标热力图确定子单元1063被配置为根据所述初始热力图和所述矫正热力图确定目标热力图。
映射单元107被配置为将所述目标热力图映射至所述待处理图像中,确定目标对象的姿态。
本实施例通过对待处理图像进行目标检测,获取目标对象区域图像,将目标区域图像输入至姿态关键点识别模型的特征金字塔网络中进行处理,获取所述目标区域图像对应的多个不同尺度的第一特征图,对各所述第一特征图依次进行加和处理,生成多个第二特征图,根据多个所述第二特征图确定所述目标区域图像对应的初始热力图,将多个所述第二特征图和所述初始热力图输入至所述姿态关键点识别模型的矫正网络中进行处理,以对所述初始热力图进行矫正,确定目标热力图,将所述目标热力图映射至所述待处理图像中,确定目标对象的姿态,由此,本实施例通过采用包括特征金字塔网络和矫正网络的姿态关键识别模型,提高了姿态关键点识别的准确性,从而提高目标对象姿态识别的准确性。
图11是本发明实施例的电子设备的示意图。如图11所示,图11所示的电子设备为通用地址查询装置,其包括通用的计算机硬件结构,其至少包括处理器111和存储器112。处理器111和存储器112通过总线113连接。存储器112适于存储处理器111可执行的指令或程序。处理器111可以是独立的微处理器,也可以是一个或者多个微处理器集合。由此,处理器111通过执行存储器112所存储的指令,从而执行如上所述的本发明实施例的方法流程实现对于数据的处理和对于其它装置的控制。总线113将上述多个组件连接在一起,同时将上述组件连接到显示控制器114和显示装置以及输入/输出(I/O)装置115。输入/输出(I/O)装置115可以是鼠标、键盘、调制解调器、网络接口、触控输入装置、体感输入装置、打印机以及本领域公知的其他装置。典型地,输入/输出装置115通过输入/输出(I/O)控制器1111与系统相连。
本领域的技术人员应明白,本申请的实施例可提供为方法、装置(设备)或计算机程序产品。因此,本申请可采用完全硬件实施例、完全软件实施例或结合软件和硬件方面的实施例的形式。而且,本申请可采用在一个或多个其中包含有计算机可用程序代码的计算机可读存储介质(包括但不限于磁盘存储器、CD-ROM、光学存储器等)上实施的计算机程序产品。
本申请是参照根据本申请实施例的方法、装置(设备)和计算机程序产品的流程图来描述的。应理解可由计算机程序指令实现流程图中的每一流程。
这些计算机程序指令可以存储在能引导计算机或其他可编程数据处理设备以特定方式工作的计算机可读存储器中,使得存储在该计算机可读存储器中的指令产生包括指令装置的制造品,该指令装置实现流程图一个流程或多个流程中指定的功能。
也可提供这些计算机程序指令到通用计算机、专用计算机、嵌入式处理机或其他可编程数据处理设备的处理器以产生一个机器,使得通过计算机或其他可编程数据处理设备的处理器执行的指令产生用于实现在流程图一个流程或多个流程中指定的功能的装置。
本发明的另一实施例涉及一种非易失性存储介质,用于存储计算机可读程序,所述计算机可读程序用于供计算机执行上述部分或全部的方法实施例。
即,本领域技术人员可以理解,实现上述实施例方法中的全部或部分步骤是可以通过程序来指令相关的硬件来完成,该程序存储在一个存储介质中,包括若干指令用以使得一个设备(可以是单片机,芯片等)或处理器(processor)执行本申请各实施例所述方法的全部或部分步骤。而前述的存储介质包括:U盘、移动硬盘、只读存储器(ROM,Read-OnlyMemory)、随机存取存储器(RAM,Random Access Memory)、磁碟或者光盘等各种可以存储程序代码的介质。
以上所述仅为本发明的优选实施例,并不用于限制本发明,对于本领域技术人员而言,本发明可以有各种改动和变化。凡在本发明的精神和原理之内所作的任何修改、等同替换、改进等,均应包含在本发明的保护范围之内。
Claims (26)
1.一种姿态关键点识别模型的训练方法,其特征在于,所述姿态关键点识别模型包括特征金字塔网络和矫正网络,所述方法包括:
获取训练样本,所述训练样本包括多组含有目标对象的训练图像以及对应的姿态关键点标注图像;
将所述训练图像输入至所述特征金字塔网络进行处理,获取所述训练图像对应的多个不同尺度的第一特征图;
对所述多个不同尺度的第一特征图依次进行加和处理,生成至少一个第二特征图;
基于第一损失函数,根据所述至少一个第二特征图与所述训练图像对应的标准关键点热力图计算第一损失,所述标准关键点热力图根据所述训练图像对应的姿态关键点标注图像获取;
根据所述第一损失调节所述特征金字塔网络的参数;
将所述至少一个第二特征图输入至所述矫正网络进行处理,生成第三特征图;
基于第二损失函数,根据所述第三特征图与所述训练图像对应的标准关键点热力图计算第二损失;
根据所述第二损失调节所述特征金字塔网络的参数和所述矫正网络的参数;
响应于所述第一损失和第二损失满足预定条件,确定所述姿态关键点识别模型。
2.根据权利要求1所述的方法,其特征在于,基于第一损失函数,根据所述至少一个第二特征图与所述训练图像对应的标准关键点热力图计算第一损失包括:
对于各所述第二特征图,对所述第二特征图进行卷积处理,获取所述训练图像的各关键点的第一热力图;
对于各所述关键点,根据所述关键点的至少一个第一热力图分别与对应的标准关键点热力图计算所述关键点的第一关键点损失;
根据各所述关键点的第一关键点损失确定所述第一损失。
3.根据权利要求1所述的方法,其特征在于,基于第二损失函数根据所述第三特征图与所述训练图像对应的标准关键点热力图计算第二损失包括:
对所述第三特征图进行卷积处理,获取所述训练图像的各关键点的第二热力图;
对于各所述关键点,根据所述关键点的第二热力图与对应的标准关键点热力图计算所述关键点的第二关键点损失;
对各所述第二关键点损失进行排序;
根据关键点损失排序结果选择预定个第二关键点损失确定所述第二损失。
4.根据权利要求1所述的方法,其特征在于,所述特征金字塔网络包括多层残差块,各层残差块输出有对应的第一特征图;
对所述多个不同尺度的第一特征图进行依次加和处理,生成至少一个第二特征图包括:
将当前层残差块输出的第一特征图与加和特征图进行加和处理,获取对应的第二特征图,所述加和特征图为当前层之后的各层残差块输出的第一特征图的加和处理结果、或最后一层残差块输出的第一特征图;以及
根据所述最后一层残差块输出的第一特征图确定对应的第二特征图。
5.根据权利要求1所述的方法,其特征在于,将所述至少一个第二特征图输入至所述矫正网络进行处理,生成第三特征图包括:
将所述至少一个第二特征图输入对应的瓶颈层和上采样层进行处理,获取对应的第四特征图;
将各所述第四特征图进行连接处理,生成所述第三特征图。
6.根据权利要求1所述的方法,其特征在于,所述预定条件为所述第一损失小于第一预定值,且所述第二损失小于第二预定值。
7.根据权利要求1所述的方法,其特征在于,所述预定条件为训练迭代次数达到预定值,所述第一损失小于第一预定值,且所述第二损失小于第二预定值。
8.根据权利要求1所述的方法,其特征在于,所述姿态关键点识别模型以Resnet-101或Resnet-50为骨干网络。
9.一种姿态识别方法,其特征在于,所述方法包括:
获取待处理图像;
对所述待处理图像进行目标检测,获取目标对象区域图像;
将所述目标区域图像输入至姿态关键点识别模型的特征金字塔网络中进行处理,获取所述目标区域图像对应的多个不同尺度的第一特征图;
对所述多个不同尺度的第一特征图依次进行加和处理,生成至少一个第二特征图;
根据所述至少一个第二特征图确定所述目标区域图像对应的初始热力图;
将所述至少一个第二特征图和所述初始热力图输入至所述姿态关键点识别模型的矫正网络中进行处理,以对所述初始热力图进行矫正,确定目标热力图;
将所述目标热力图映射至所述待处理图像中,确定目标对象的姿态。
10.根据权利要求9所述的方法,其特征在于,所述特征金字塔网络包括多层残差块,各层残差块输出有对应的第一特征图;
对所述多个不同尺度的第一特征图进行加和处理,生成多个第二特征图包括:
将当前层残差块输出的第一特征图与加和特征图进行加和处理,获取对应的第二特征图,所述加和特征图为当前层之后的各层残差块输出的第一特征图的加和处理结果、或最后一层残差块输出的第一特征图;
根据所述最后一层残差块输出的第一特征图确定对应的第二特征图。
11.根据权利要求9所述的方法,其特征在于,将多个所述第二特征图和所述初始热力图输入至所述姿态关键点识别模型的矫正网络中进行处理,以对所述初始热力图进行矫正,确定目标热力图包括:
将所述至少一个第二特征图输入至对应的瓶颈层和上采样层进行处理,确定所述目标区域图像对应的第三特征图;
根据所述第三特征图确定预定个关键点对应的矫正热力图;
根据所述初始热力图和所述矫正热力图确定目标热力图。
12.根据权利要求8所述的方法,其特征在于,所述姿态关键点识别模型以Resnet-101或Resnet-50为骨干网络。
13.一种姿态关键点识别模型的训练装置,其特征在于,所述姿态关键点识别模型包括特征金字塔网络和矫正网络,所述装置包括:
样本获取单元,被配置为获取训练样本,所述训练样本包括多组含有目标对象的训练图像以及对应的姿态关键点标注图像;
第一处理单元,被配置为将所述训练图像输入至所述特征金字塔网络进行处理,获取所述训练图像对应的多个不同尺度的第一特征图;
第二处理单元,被配置为对所述多个不同尺度的第一特征图依次进行加和处理,生成至少一个第二特征图;
第一损失计算单元,被配置为基于第一损失函数,根据所述至少一个第二特征图与所述训练图像对应的标准关键点热力图计算第一损失,所述标准关键点热力图根据所述训练图像对应的姿态关键点标注图像获取;
第一参数调节单元,被配置为根据所述第一损失调节所述特征金字塔网络的参数;
第三处理单元,被配置为将所述至少一个第二特征图输入至所述矫正网络进行处理,生成第三特征图;
第二损失计算单元,被配置为基于第二损失函数,根据所述第三特征图与所述训练图像对应的标准关键点热力图计算第二损失;
第二参数调节单元,被配置为根据所述第二损失调节所述特征金字塔网络的参数和所述矫正网络的参数;
模型确定单元,被配置为响应于所述第一损失和第二损失满足预定条件,确定所述姿态关键点识别模型。
14.根据权利要求13所述的装置,其特征在于,第一损失计算单元包括:
第一热力图获取子单元,被配置为对于各所述第二特征图,对所述第二特征图进行卷积处理,获取所述训练图像的各关键点的第一热力图;
第一关键损失计算子单元,被配置为对于各所述关键点,根据所述关键点的至少一个第一热力图分别与对应的标准关键点热力图计算所述关键点的第一关键点损失;
第一损失计算子单元,被配置为根据各所述关键点的第一关键点损失确定所述第一损失。
15.根据权利要求13所述的装置,其特征在于,第二损失计算单元包括:
第二热力图获取子单元,被配置为对所述第三特征图进行卷积处理,获取所述训练图像的各关键点的第二热力图;
第二关键点损失计算子单元,被配置为对于各所述关键点,根据所述关键点的第二热力图与对应的标准关键点热力图计算所述关键点的第二关键点损失;
排序子单元,被配置为对各所述第二关键点损失进行排序;
第二损失确定子单元,被配置为根据关键点损失排序结果选择预定个第二关键点损失确定所述第二损失。
16.根据权利要求13所述的装置,其特征在于,所述特征金字塔网络包括多层残差块,各层残差块输出有对应的第一特征图;
第二处理单元包括:
第二特征图确定子单元,被配置为将当前层残差块输出的第一特征图与加和特征图进行加和处理,获取对应的第二特征图,所述加和特征图为当前层之后的各层残差块输出的第一特征图的加和处理结果、或最后一层残差块输出的第一特征图;
第一特征图确定子单元还被配置根据所述最后一层残差块输出的第一特征图确定对应的第二特征图。
17.根据权利要求13所述的装置,其特征在于,第三处理单元包括:
第四特征图获取子单元,被配置为将所述至少一个第二特征图输入对应的瓶颈层和上采样层进行处理,获取对应的第四特征图;
第三特征图确定子单元,被配置为将各第四特征图进行连接处理,生成所述对应的第三特征图。
18.根据权利要求13所述的装置,其特征在于,所述预定条件为所述第一损失小于第一预定值,所述第二损失小于第二预定值。
19.根据权利要求13所述的装置,其特征在于,所述预定条件为训练迭代次数达到预定值,所述第一损失小于第一预定值,且所述第二损失小于第二预定值。
20.根据权利要求13所述的装置,其特征在于,所述姿态关键点识别模型以Resnet-101或Resnet-50为骨干网络。
21.一种姿态识别装置,其特征在于,所述装置包括:
待处理图像获取单元,被配置为获取待处理图像;
目标检测单元,被配置为对所述待处理图像进行目标检测,获取目标对象区域图像;
第一特征处理单元,被配置为将所述目标区域图像输入至姿态关键点识别模型的特征金字塔网络中进行处理,获取所述目标区域图像对应的多个不同尺度的第一特征图;
第二特征处理单元,被配置为对所述多个不同尺度的第一特征图依次进行加和处理,生成至少一个第二特征图;
初始热力图确定单元,被配置为根据所述至少一个第二特征图确定所述目标区域图像对应的初始热力图;
目标热力图确定单元,被配置为将所述至少一个第二特征图和所述初始热力图输入至所述姿态关键点识别模型的矫正网络中进行处理,以对所述初始热力图进行矫正,确定目标热力图;
映射单元,被配置为将所述目标热力图映射至所述待处理图像中,确定目标对象的姿态。
22.根据权利要求21所述的装置,其特征在于,所述特征金字塔网络包括多层残差块,各层残差块输出有对应的第一特征图;
第二特征处理单元包括:
第二特征处理子单元,被配置为将当前层残差块输出的第一特征图与加和特征图进行加和处理,获取对应的第二特征图,所述加和特征图为当前层之后的各层残差块输出的第一特征图的加和处理结果、或最后一层残差块输出的第一特征图;
第二特征处理子单元还被配置为根据所述最后一层残差块输出的第一特征图确定对应的第二特征图。
23.根据权利要求21所述的装置,其特征在于,目标热力图确定单元包括:
第三特征图确定子单元,被配置为将所述至少一个第二特征图输入至对应的瓶颈层和上采样层进行处理,确定所述目标区域图像对应的第三特征图;
矫正热力图确定子单元,被配置为根据所述第三特征图确定预定个关键点对应的矫正热力图;
目标热力图确定子单元,被配置为根据所述初始热力图和所述矫正热力图确定目标热力图。
24.根据权利要求21所述的装置,其特征在于,所述姿态关键点识别模型以Resnet-101或Resnet-50为骨干网络。
25.一种电子设备,包括存储器和处理器,其特征在于,所述存储器用于存储一条或多条计算机程序指令,其中,所述一条或多条计算机程序指令被所述处理器执行以实现如权利要求1-12中任一项所述的方法。
26.一种计算机可读存储介质,其上存储计算机程序指令,其特征在于,所述计算机程序指令在被处理器执行时以实现如权利要求1-12中任一项所述的方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010383035.3A CN111832383B (zh) | 2020-05-08 | 2020-05-08 | 姿态关键点识别模型的训练方法、姿态识别方法及装置 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010383035.3A CN111832383B (zh) | 2020-05-08 | 2020-05-08 | 姿态关键点识别模型的训练方法、姿态识别方法及装置 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN111832383A CN111832383A (zh) | 2020-10-27 |
CN111832383B true CN111832383B (zh) | 2023-12-08 |
Family
ID=72913957
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202010383035.3A Active CN111832383B (zh) | 2020-05-08 | 2020-05-08 | 姿态关键点识别模型的训练方法、姿态识别方法及装置 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN111832383B (zh) |
Families Citing this family (10)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN112418046B (zh) * | 2020-11-17 | 2023-06-23 | 武汉云极智能科技有限公司 | 一种基于云机器人的健身指导方法、存储介质及系统 |
CN112434612A (zh) * | 2020-11-25 | 2021-03-02 | 创新奇智(上海)科技有限公司 | 吸烟检测方法、装置、电子设备及计算机可读存储介质 |
CN112580488B (zh) * | 2020-12-15 | 2023-12-22 | 深圳大学 | 一种基于自启发的人体姿态估计模型训练方法及装置 |
CN112489129A (zh) * | 2020-12-18 | 2021-03-12 | 深圳市优必选科技股份有限公司 | 位姿识别模型训练方法、装置、位姿识别方法和终端设备 |
CN112597896B (zh) * | 2020-12-23 | 2022-09-09 | 中国科学技术大学 | 基于无线信号的人体姿势成像方法及装置 |
CN112597955B (zh) * | 2020-12-30 | 2023-06-02 | 华侨大学 | 一种基于特征金字塔网络的单阶段多人姿态估计方法 |
CN113095129B (zh) * | 2021-03-01 | 2024-04-26 | 北京迈格威科技有限公司 | 姿态估计模型训练方法、姿态估计方法、装置和电子设备 |
CN113076891B (zh) * | 2021-04-09 | 2023-08-22 | 华南理工大学 | 基于改进高分辨率网络的人体姿态预测方法及系统 |
CN113297995B (zh) * | 2021-05-31 | 2024-01-16 | 深圳市优必选科技股份有限公司 | 一种人体姿态估计方法及终端设备 |
CN113870215B (zh) * | 2021-09-26 | 2023-04-07 | 推想医疗科技股份有限公司 | 中线提取方法及装置 |
Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108038409A (zh) * | 2017-10-27 | 2018-05-15 | 江西高创保安服务技术有限公司 | 一种行人检测方法 |
CN110276316A (zh) * | 2019-06-26 | 2019-09-24 | 电子科技大学 | 一种基于深度学习的人体关键点检测方法 |
CN110443144A (zh) * | 2019-07-09 | 2019-11-12 | 天津中科智能识别产业技术研究院有限公司 | 一种人体图像关键点姿态估计方法 |
WO2019233244A1 (zh) * | 2018-06-08 | 2019-12-12 | 腾讯科技(深圳)有限公司 | 图像处理方法、装置、计算机可读介质及电子设备 |
WO2020010979A1 (zh) * | 2018-07-10 | 2020-01-16 | 腾讯科技(深圳)有限公司 | 手部关键点的识别模型训练方法、识别方法及设备 |
-
2020
- 2020-05-08 CN CN202010383035.3A patent/CN111832383B/zh active Active
Patent Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108038409A (zh) * | 2017-10-27 | 2018-05-15 | 江西高创保安服务技术有限公司 | 一种行人检测方法 |
WO2019233244A1 (zh) * | 2018-06-08 | 2019-12-12 | 腾讯科技(深圳)有限公司 | 图像处理方法、装置、计算机可读介质及电子设备 |
WO2020010979A1 (zh) * | 2018-07-10 | 2020-01-16 | 腾讯科技(深圳)有限公司 | 手部关键点的识别模型训练方法、识别方法及设备 |
CN110276316A (zh) * | 2019-06-26 | 2019-09-24 | 电子科技大学 | 一种基于深度学习的人体关键点检测方法 |
CN110443144A (zh) * | 2019-07-09 | 2019-11-12 | 天津中科智能识别产业技术研究院有限公司 | 一种人体图像关键点姿态估计方法 |
Non-Patent Citations (2)
Title |
---|
基于级联卷积神经网络的服饰关键点定位算法;李锵;姚麟倩;关欣;;天津大学学报(自然科学与工程技术版)(第03期);全文 * |
基于视频的实时多人姿态估计方法;闫芬婷;王鹏;吕志刚;丁哲;乔梦雨;;激光与光电子学进展(02);全文 * |
Also Published As
Publication number | Publication date |
---|---|
CN111832383A (zh) | 2020-10-27 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN111832383B (zh) | 姿态关键点识别模型的训练方法、姿态识别方法及装置 | |
CN109657631B (zh) | 人体姿态识别方法及装置 | |
US10078796B2 (en) | Apparatus and method of hand gesture recognition based on depth image | |
CN111160085A (zh) | 一种人体图像关键点姿态估计方法 | |
CN110110600B (zh) | 眼部oct图像病灶识别方法、装置及存储介质 | |
JP6392478B1 (ja) | 情報処理装置、情報処理プログラム、及び、情報処理方法 | |
CN107784321B (zh) | 数字绘本快速识别方法、系统及计算机可读存储介质 | |
CN111401318B (zh) | 动作识别方法及装置 | |
CN111507134A (zh) | 人形姿态检测方法、装置、计算机设备及存储介质 | |
CN110135508B (zh) | 模型训练方法、装置、电子设备及计算机可读存储介质 | |
US20230334893A1 (en) | Method for optimizing human body posture recognition model, device and computer-readable storage medium | |
CN112651389B (zh) | 非正视虹膜图像的矫正模型训练、矫正、识别方法及装置 | |
CN112287865B (zh) | 一种人体姿态识别的方法及装置 | |
CN109711374B (zh) | 人体骨骼点识别方法及装置 | |
JP6381368B2 (ja) | 画像処理装置、画像処理方法、およびプログラム | |
CN108268840B (zh) | 一种人脸跟踪方法及装置 | |
CN112115790A (zh) | 人脸识别方法、装置、可读存储介质和电子设备 | |
CN114519401A (zh) | 一种图像分类方法及装置、电子设备、存储介质 | |
CN111316283B (zh) | 姿势识别方法及装置 | |
CN112329571A (zh) | 一种基于姿态质量评估的自适应人体姿态优化方法 | |
JPWO2015068417A1 (ja) | 画像照合システム、画像照合方法およびプログラム | |
CN101256627B (zh) | 一种基于不变矩的图形畸变分析方法 | |
CN116310976A (zh) | 学习习惯养成方法、装置、电子设备及存储介质 | |
CN112801045B (zh) | 一种文本区域检测方法、电子设备及计算机存储介质 | |
CN112861678B (zh) | 一种图像识别方法及装置 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |