CN113435431B - 姿态检测方法、神经网络模型的训练方法、装置及设备 - Google Patents
姿态检测方法、神经网络模型的训练方法、装置及设备 Download PDFInfo
- Publication number
- CN113435431B CN113435431B CN202110993216.2A CN202110993216A CN113435431B CN 113435431 B CN113435431 B CN 113435431B CN 202110993216 A CN202110993216 A CN 202110993216A CN 113435431 B CN113435431 B CN 113435431B
- Authority
- CN
- China
- Prior art keywords
- limb
- target
- image
- initial training
- training
- 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 441
- 238000001514 detection method Methods 0.000 title claims abstract description 229
- 238000003062 neural network model Methods 0.000 title claims abstract description 30
- 238000000034 method Methods 0.000 claims abstract description 85
- 230000009471 action Effects 0.000 claims abstract description 23
- 238000012545 processing Methods 0.000 claims description 76
- 230000006870 function Effects 0.000 claims description 25
- 238000012937 correction Methods 0.000 claims description 24
- 238000004590 computer program Methods 0.000 claims description 12
- 238000012986 modification Methods 0.000 claims description 9
- 230000004048 modification Effects 0.000 claims description 9
- 238000009432 framing Methods 0.000 claims description 5
- 210000003414 extremity Anatomy 0.000 description 338
- 230000008569 process Effects 0.000 description 31
- 238000013528 artificial neural network Methods 0.000 description 13
- 230000001960 triggered effect Effects 0.000 description 12
- 230000000694 effects Effects 0.000 description 11
- 230000002829 reductive effect Effects 0.000 description 11
- 238000002372 labelling Methods 0.000 description 9
- 238000010586 diagram Methods 0.000 description 8
- 230000003993 interaction Effects 0.000 description 7
- 238000012544 monitoring process Methods 0.000 description 5
- 230000036961 partial effect Effects 0.000 description 5
- 210000004247 hand Anatomy 0.000 description 4
- 230000000717 retained effect Effects 0.000 description 4
- 230000008901 benefit Effects 0.000 description 3
- 238000004891 communication Methods 0.000 description 3
- 230000008878 coupling Effects 0.000 description 3
- 238000010168 coupling process Methods 0.000 description 3
- 238000005859 coupling reaction Methods 0.000 description 3
- 238000005516 engineering process Methods 0.000 description 3
- 230000037237 body shape Effects 0.000 description 2
- 239000011521 glass Substances 0.000 description 2
- 230000002452 interceptive effect Effects 0.000 description 2
- 230000000670 limiting effect Effects 0.000 description 2
- 238000011160 research Methods 0.000 description 2
- 210000000707 wrist Anatomy 0.000 description 2
- 230000003190 augmentative effect Effects 0.000 description 1
- 239000003086 colorant Substances 0.000 description 1
- 238000013527 convolutional neural network Methods 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 230000008921 facial expression Effects 0.000 description 1
- 238000005429 filling process Methods 0.000 description 1
- 230000003287 optical effect Effects 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
- 210000001364 upper extremity Anatomy 0.000 description 1
- 230000000007 visual effect Effects 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/22—Matching criteria, e.g. proximity measures
-
- 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
-
- 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
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Physics & Mathematics (AREA)
- General Engineering & Computer Science (AREA)
- Evolutionary Computation (AREA)
- Bioinformatics & Computational Biology (AREA)
- Computational Linguistics (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Health & Medical Sciences (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Evolutionary Biology (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Image Analysis (AREA)
Abstract
本公开提供了一种姿态检测方法、神经网络模型的训练方法、装置及设备,其中,该方法包括:获取包含目标主播的肢体部位的视频图像帧;通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,其中,所述姿态检测模型为利用目标样本图像及对应的训练标签进行监督训练得到,所述目标样本图像包含待检测的肢体部位样本,所述训练标签基于所述肢体部位样本的肢体特征点的监督类型及所述肢体特征点对应的初始训练标签进行调整得到;在得到所述目标主播的姿态检测结果之后,根据所述姿态检测结果生成所述目标主播所对应的虚拟主播的姿态触发信号;根据所述姿态触发信号控制所述虚拟主播执行相应的触发动作。
Description
技术领域
本公开涉及计算机的技术领域,具体而言,涉及一种姿态检测方法、神经网络模型的训练方法、装置及设备。
背景技术
在现有的直播技术中,主播可以通过触发外部设备(例如,鼠标或者键盘)在直播界面展示特效动画;或者,通过检测主播的面部表情触发展示相应的特效动画。因此,一方面,在现有的直播技术中触发展示的特效比较单一化,且特效效果展示较差;且该方案中仅仅只是单纯地演示某个互动动作,无法与主播的肢体动作产生动作关联,导致实际互动效果不佳,进而难以适应于当前用户对于直播的多元化需求。因此,亟需一种能够对主播进行姿态估计的技术方案。另一方面,在现有的姿态估计方法中,在通过姿态估计模型进行姿态估计时,首先需要采集包含特定对象的肢体部位的图像,然后,需要对采集到的图像中肢体部位的关键点进行标注,从而根据标注之后的图像来训练姿态估计模型。此时,可以通过标注人员对每个肢体部位的关键点进行标注。在图像数量较多的情况下,将产生大量的标注任务。此时,若选择通过标注人员对关键点进行标注,在消耗了大量标注时间的同时,还将浪费大量的人力成本。因此,采用上述所描述的方式确定用于进行姿态检测的姿态检测模型时,将增大相应的开发成本。
发明内容
本公开实施例至少提供一种姿态检测方法、神经网络模型的训练方法、装置及设备。
第一方面,本公开实施例提供了一种姿态检测方法,包括:获取包含目标主播的肢体部位的视频图像帧;通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,其中,所述姿态检测模型为利用目标样本图像及对应的训练标签进行监督训练得到,所述目标样本图像包含待检测的肢体部位样本,所述训练标签基于所述肢体部位样本的肢体特征点的监督类型及所述肢体特征点对应的初始训练标签进行调整得到;在得到所述目标主播的姿态检测结果之后,根据所述姿态检测结果生成所述目标主播所对应的虚拟主播的姿态触发信号;根据所述姿态触发信号控制所述虚拟主播执行相应的触发动作。
在本公开实施例中,在获取到视频图像帧之后,通过姿态检测模型对目标主播进行姿态检测,进而根据姿态检测结果控制虚拟主播执行相应触发动作的方式,可以实现自动的通过目标主播的姿态检测结果在视频直播界面上触发展示虚拟主播的相应部位执行相应触发操作,进而实现触发虚拟主播执行与主播的肢体产生动作关联的动作,从而提高实际互动效果。同时,通过确定待检测的肢体部位的姿态检测结果对虚拟主播进行触发,还可以实现虚拟主播中对应触发部位的精准触发,从而满足用户的丰富触发需求。同时,在本公开技术方案中,通过基于肢体部位样本的肢体特征点的监督类型及肢体特征点对应的初始训练标签进行调整得到训练标签,可以通过对初始训练标签的复用来训练得到姿态检测模型,通过该方式可以节省数据标注成本,并加快数据标注的速度。
一种可选的实施方式中,所述通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,包括:在所述视频图像帧中检测目标框体,其中,所述目标框体用于框选所述视频图像帧中所述目标主播的肢体部位;通过所述姿态检测模型对所述视频图像帧中位于所述目标框体内的图像进行姿态检测,得到所述姿态检测结果。
上述实施方式中,通过检测包含目标对象的目标框体,并对位于该目标框体内的图像进行姿态检测,可以提高姿态检测结果的准确性。
一种可选的实施方式中,所述通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,包括:在所述视频图像帧中未检测到所述目标主播的至少部分指定肢体部位的情况下,对所述视频图像帧进行处理,得到目标图像;所述目标图像中包含用于对该至少部分指定肢体部位进行姿态检测的区域;通过所述姿态检测模型对所述目标图像进行姿态检测,得到所述目标主播的姿态检测结果。
上述实施方式中,在视频图像帧中未检测到目标对象的部分指定肢体部位的情况下,通过对视频图像帧进行边缘填补处理,得到目标图像,以根据该目标图像进行姿态检测的方式,可以实现在视频图像帧中不包含完整的指定肢体部位的情况下,依然可以对该视频图像帧进行姿态检测,得到准确的姿态检测结果。
一种可选的实施方式中,所述基于所述肢体部位样本的肢体特征点的监督类型及所述肢体特征点对应的初始训练标签进行调整得到所述训练标签,包括:确定所述肢体特征点的监督类型;基于所述监督类型修正所述初始训练标签,以根据修正之后的初始训练标签确定所述目标样本图像的训练标签。
一种可选的实施方式中,所述确定所述肢体特征点的监督类型,包括:确定所述肢体特征点的特征信息,其中,所述特征信息用于指示对应肢体特征点和所述目标样本图像之间的位置关系,和/或,用于指示对应肢体特征点是否为待检测部位的特征点;基于所述特征信息,确定所述肢体特征点的监督类型;并在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签,得到所述训练标签。
上述实施方式中,通过确定肢体特征点的监督类型,并基于该监督类型对肢体特征点的初始训练标签进行修正的方式,可以保留需要进行监督的肢体特征点的初始训练标签,同时还可以提高初始训练样本中初始训练标签的利用率。
第二方面,本公开实施例提供了一种神经网络模型的训练方法,包括:获取初始训练样本;所述初始训练样本中包含样本图像和该样本图像中所包含的目标对象的肢体部位的肢体特征点的初始训练标签;基于待训练网络模型的检测任务在所述肢体部位中确定待检测部位;并对所述样本图像进行处理,得到包含所述待检测部位的目标样本图像;确定所述肢体特征点的监督类型,并基于所述监督类型修正所述初始训练标签,以根据修正之后的初始训练标签确定所述目标样本图像的训练标签;通过所述目标样本图像和所述训练标签,对所述待训练网络模型进行有监督训练。
在本公开实施例中,在确定出待训练网络模型之后,通过对初始训练样本中的样本图像和初始训练标签进行修正处理,得到目标样本图像和训练标签的方式,可以实现初始训练样本的复用,以使初始训练样本可以同时用于多个不同的待训练网络模型的训练过程,进而通过节省数据标注成本的方式降低神经网络的训练成本。
一种可选的实施方式中,所述待检测部位为多个,所述通过所述目标样本图像和所述训练标签,对所述待训练网络模型进行训练,包括:通过所述待训练网络模型对所述目标样本图像进行图像处理,得到所述目标样本图像中每个待检测部位的位置信息;基于多个所述待检测部位中第一部位和第二部位的位置信息,以及所述训练标签,确定用于约束所述第一部位和第二部位之间的位置差异的目标损失函数的函数值,其中,所述第一部位和所述第二部位为具有关联关系的待检测部位;根据所述目标损失函数的函数值调整所述待训练网络模型的模型参数,以通过调整之后的模型参数对所述待训练网络模型进行训练。
上述实施方式中,通过第一部位和第二部位的位置信息之间的位置差异和训练标签构建目标损失函数的方式,可以减少第一部位和第二部位之间位置差异较大的现象,从而提高待训练网络模型的处理精度。
一种可选的实施方式中,所述确定所述肢体特征点的监督类型,并基于所述监督类型修正所述初始训练标签,包括:确定所述肢体特征点的特征信息,其中,所述特征信息用于指示对应肢体特征点和所述目标样本图像之间的位置关系,和/或,用于指示对应肢体特征点是否为待检测部位的特征点;基于所述特征信息,确定所述肢体特征点的监督类型;在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签。
上述实施方式中,通过确定肢体特征点的监督类型,并基于该监督类型对肢体特征点的初始训练标签进行修正的方式,可以保留需要进行监督的肢体特征点的初始训练标签,同时还可以提高初始训练样本中初始训练标签的利用率。
通过监督类型对肢体特征点的初始训练标签进行修正的方式,不仅可以监督学习位于目标样本图像内的肢体特征点,还可以监督学习位于目标样本图像之外且需要进行监督学习的肢体特征点。在基于修正之后的初始训练标签(即,训练标签)和目标样本图像对待训练网络模型进行训练时,就可以实现通过该网络模型对未包含在图像中的部分肢体部位的肢体特征点进行预测。在采用本公开技术方案中确定出的修正之后的初始训练标签和目标样本图像,对待训练网络模型进行训练时,可以对未包含在图像中的部分肢体部位的肢体特征点进行预测,从而实现待检测目标的姿态识别。因此,通过本公开技术方案进行训练之后的网络模型的鲁棒性更强,且识别过程更加稳定。
一种可选的实施方式中,所述在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签,包括:在根据所述监督类型确定出所述肢体特征点位于所述目标样本图像之外,且不是所述待检测部位的特征点情况下,将所述肢体特征点的初始训练标签修正为第一初始训练标签;所述第一初始训练标签用于表征该肢体特征点为位于所述目标样本图像之外的非监督类特征点。
一种可选的实施方式中,所述在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签,包括:在根据所述初始训练标签确定出所述肢体特征点位于所述目标样本图像之外,且属于待检测部位的特征点的情况下,将所述肢体特征点的初始训练标签修正为第二初始训练标签;所述第二初始训练标签用于表征该肢体特征点为位于所述目标样本图像之外的监督类特征点。
上述实施方式中,通过上述所描述的初始训练标签的修正方式,可以使初始训练样本可以同时用于多个不同的检测任务,进而通过节省数据标注成本的方式降低神经网络的训练成本。进一步地,在采用本公开技术方案中确定出的修正之后的初始训练标签和目标样本图像,对待训练网络模型进行训练时,可以对未包含在图像中的部分肢体部位的肢体特征点进行预测,从而实现待检测目标的姿态识别。因此,通过本公开技术方案进行训练之后的网络模型的鲁棒性更强,且识别过程更加稳定。
一种可选的实施方式中,所述样本图像为对原始图像执行目标操作之后确定的图像,其中,所述原始图像为在场景复杂度不满足预设复杂要求的场景下采集到的包含目标对象的至少一个肢体部位的图像,所述目标操作包括以下至少之一:背景替换操作、调整图像亮度的操作、调整图像曝光度的操作、添加反光图层的操作。
在本公开实施例中,通过对原始图像进行背景替换,可以实现通过较低的成本,在较短的时间内获得海量训练样本。通过对原始图像进行调整图像亮度、调整图像曝光度以及添加反光图层中的至少一种图像增强处理,可以提高样本图像的背景复杂度。通过上述至少一种目标操作所得到的样本图像能够更加贴近真实的应用场景,在基于该样本图像对待训练网络模型进行训练时,可以提高待训练网络模型的处理精度。
一种可选的实施方式中,所述对所述样本图像进行处理,得到包含所述待检测部位的目标样本图像,包括:对所述样本图像中的目标检测部位进行图像遮挡处理,得到所述目标样本图像;其中,所述目标检测部位为所述肢体部位中除所述待检测部位之外的其他部位;和/或,对所述样本图像中位于目标区域内的图像进行图像裁剪处理,裁剪得到包含所述待检测部位的目标样本图像;所述目标区域为所述样本图像中包含所述待检测部位的图像区域。
上述实施方式中,通过对样本图像进行图像遮挡处理和/或图像裁剪处理,可以在不重新采集样本图像的基础上,得到与检测任务相匹配的样本图像,从而实现样本图像的复用,以降低神经网络的训练时间和训练成本。
第三方面,本公开实施例提供了一种姿态检测装置,包括:第一获取单元,用于获取包含目标主播的肢体部位的视频图像帧;姿态检测单元,用于通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,其中,所述姿态检测模型为利用目标样本图像及对应的训练标签进行监督训练得到,所述目标样本图像包含待检测的肢体部位样本,所述训练标签基于所述肢体部位样本的肢体特征点的监督类型及所述肢体特征点对应的初始训练标签进行调整得到;生成单元,用于在得到所述目标主播的姿态检测结果之后,根据所述姿态检测结果生成所述目标主播所对应的虚拟主播的姿态触发信号;控制单元,用于根据所述姿态触发信号控制所述虚拟主播执行相应的触发动作。
第四方面,本公开实施例提供了一种神经网络模型的训练装置,包括:第一获取单元,用于获取初始训练样本;所述初始训练样本中包含样本图像和该样本图像中所包含的目标对象的肢体部位的肢体特征点的初始训练标签;修正处理单元,用于基于待训练网络模型的检测任务在所述肢体部位中确定待检测部位;并对所述样本图像进行处理,得到包含所述待检测部位的目标样本图像;确定单元,用于确定所述肢体特征点的监督类型,并基于所述监督类型修正所述初始训练标签,以根据修正之后的初始训练标签确定所述目标样本图像的训练标签;训练单元,用于通过所述目标样本图像和所述训练标签,对所述待训练网络模型进行有监督训练。
第五方面,本公开实施例还提供一种计算机设备,包括:处理器、存储器和总线,所述存储器存储有所述处理器可执行的机器可读指令,当计算机设备运行时,所述处理器与所述存储器之间通过总线通信,所述机器可读指令被所述处理器执行时执行上述第一方面,或第一方面中任一种可能的实施方式中的步骤。
第六方面,本公开实施例还提供一种计算机可读存储介质,该计算机可读存储介质上存储有计算机程序,该计算机程序被处理器运行时执行上述第一方面,或第一方面中任一种可能的实施方式中的步骤。
为使本公开的上述目的、特征和优点能更明显易懂,下文特举较佳实施例,并配合所附附图,作详细说明如下。
附图说明
为了更清楚地说明本公开实施例的技术方案,下面将对实施例中所需要使用的附图作简单地介绍,此处的附图被并入说明书中并构成本说明书中的一部分,这些附图示出了符合本公开的实施例,并与说明书一起用于说明本公开的技术方案。应当理解,以下附图仅示出了本公开的某些实施例,因此不应被看作是对范围的限定,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他相关的附图。
图1示出了本公开实施例所提供的一种神经网络模型的训练方法的流程图;
图2示出了本公开实施例所提供的一种神经网络模型的训练方法中,确定所述肢体特征点的监督类型,并基于所述监督类型修正所述初始训练标签的具体流程图;
图3示出了本公开实施例所提供的一种姿态检测方法的流程图;
图4示出了本公开实施例所提供的一种姿态检测装置的示意图;
图5示出了本公开实施例所提供的一种神经网络模型的训练装置的示意图;
图6示出了本公开实施例所提供的一种计算机设备的示意图;
图7示出了本公开实施例所提供的另一种计算机设备的示意图。
具体实施方式
为使本公开实施例的目的、技术方案和优点更加清楚,下面将结合本公开实施例中附图,对本公开实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本公开一部分实施例,而不是全部的实施例。通常在此处附图中描述和示出的本公开实施例的组件可以以各种不同的配置来布置和设计。因此,以下对在附图中提供的本公开的实施例的详细描述并非旨在限制要求保护的本公开的范围,而是仅仅表示本公开的选定实施例。基于本公开的实施例,本领域技术人员在没有做出创造性劳动的前提下所获得的所有其他实施例,都属于本公开保护的范围。
应注意到:相似的标号和字母在下面的附图中表示类似项,因此,一旦某一项在一个附图中被定义,则在随后的附图中不需要对其进行进一步定义和解释。
本文中术语“和/或”,仅仅是描述一种关联关系,表示可以存在三种关系,例如,A和/或B,可以表示:单独存在A,同时存在A和B,单独存在B这三种情况。另外,本文中术语“至少一种”表示多种中的任意一种或多种中的至少两种的任意组合,例如,包括A、B、C中的至少一种,可以表示包括从A、B和C构成的集合中选择的任意一个或多个元素。
经研究发现,一般情况下,需要针对特定场景采集特定的图像,在采集到特定的图像之后,还需要对特定的图像进行标注,从而根据标注之后的特定图像来训练对应的神经网络。然而,针对特定场景确定特定训练样本的方式,浪费了大量的人力进行图像标注,且采用上述方式确定出的训练样本可复用性低。因此,采用上述所描述的神经网络的训练方法增大了网络训练的成本。
基于上述研究,本公开提供了一种神经网络模型的训练方法。在本公开实施例中,在获取到待训练网络模型后,可以通过该待训练网络模型的检测任务确定目标对象的待检测部位。之后,可以基于该待检测部位对样本图像进行处理,从而得到满足该待训练网络模型的训练要求的目标样本图像,即包含待检测部位的目标样本图像。通过确定肢体特征点的监督类型,并根据该监督类型修正初始训练标签,可以得到与该待训练网络模型的训练过程相匹配的训练标签,从而根据该相匹配的训练标签和目标样本图像对该待训练网络模型进行训练。
通过上述描述可知,本公开技术方案在确定出待训练网络模型之后,通过对初始训练样本中的样本图像和初始训练标签进行修正处理,得到目标样本图像和训练标签的方式,可以实现初始训练样本的复用,以使初始训练样本可以同时用于多个不同的待训练网络模型的训练过程,进而通过节省数据标注成本的方式降低神经网络的训练成本。在训练得到姿态检测模型之后,姿态检测模型对视频图像帧中的目标主播进行姿态检测,进而根据姿态检测结果控制虚拟主播执行相应触发动作的方式,可以实现自动的通过目标主播的姿态检测结果在视频直播界面上触发展示虚拟主播的相应部位执行相应触发操作。同时,通过确定待检测的肢体部位的姿态检测结果对虚拟主播进行触发,还可以实现自动的通过目标主播的姿态检测结果在视频直播界面上触发展示虚拟主播的相应部位执行相应触发操作,进而实现触发虚拟主播执行与主播的肢体产生动作关联的动作,从而提高实际互动效果。
为便于对本实施例进行理解,首先对本公开实施例所公开的一种神经网络模型的训练方法进行详细介绍,本公开实施例所提供的神经网络模型的训练方法的执行主体一般为具有一定计算能力的计算机设备。在一些可能的实现方式中,该神经网络模型的训练方法可以通过处理器调用存储器中存储的计算机可读指令的方式来实现。
参见图1所示,为本公开实施例提供的一种神经网络模型的训练方法的流程图,所述方法包括步骤S101~S107,其中:
S101:获取初始训练样本;所述初始训练样本中包含样本图像和该样本图像中所包含的目标对象的肢体部位的肢体特征点的初始训练标签。
目标对象可以为真实人体、虚拟人体、以及其他能够进行肢体检测的对象。目标对象的肢体部位可以为该目标对象的全身肢体部位,或者,部分肢体部位(例如,上半身肢体部位)。
在本公开实施例中,可以设置样本图像中所包含的目标对象的肢体部位能够满足大部分肢体检测任务。假设,目标对象为真实人体,那么肢体部位可以为该目标对象的完整上半身肢体部位(例如,头部,双臂,上肢躯干,手部),还可以为该目标对象的完整全身肢体部位(例如,头部,双臂,上肢躯干,手部,腿部和脚部)。
这里,肢体特征点的初始训练标签可以理解为:肢体特征点在样本图像中的位置信息和/或肢体类别信息(例如,该肢体关键点属于手部关键点),以及该肢体特征点在每个训练任务中的监督学习状态(例如,需要进行监督学习的特征点,或者,不需要进行监督学习的特征点)。
在本公开实施例中,初始训练样本的数量可以为多组。这里,可以基于目标对象的对象类型,和/或,目标对象的肢体部位确定多组初始训练样本。每个初始训练样本包含对应的样本标签,该样本标签可以用于指示对应目标对象的对象类型和/或目标对象的肢体部位。
例如,目标对象可以为人,除此之外,还可以为除了人之外其他能够进行肢体识别的对象,例如,仿真虚拟机器人。
具体实施时,可以基于“人”设置至少一组初始训练样本,例如,基于“人”的全身肢体部位设置一组初始训练样本,还可以基于“人”的上半身肢体部位设置一组初始训练样本。
在本公开实施方式中,通过设置多组初始训练样本,可以丰富训练样本的样本内容,从而使得该初始训练样本能够满足更加丰富的训练场景。
S103:基于待训练网络模型的检测任务在所述肢体部位中确定待检测部位;并对所述样本图像进行处理,得到包含所述待检测部位的目标样本图像。
这里,待训练网络模型的检测任务可以用于指示该待训练网络模型所需要进行检测的肢体部位(即,上述待检测部位)。之后,就可以根据该待检测部位对初始训练样本中的样本图像进行处理。例如,可以对该样本图像进行截取处理,得到包含待检测部位的目标样本图像。
在初始训练样本的数量为多组的情况下,在对样本图像进行处理之前,可以执行以下步骤:
基于检测任务确定目标对象的对象类型,并基于每组初始训练样本的样本标签,在多组初始训练样本中确定与目标对象的对象类型相匹配的初始训练样本。在确定出相匹配的初始训练样本的数量为多组的情况下,还可以基于检测任务确定目标对象的待检测部位,进而,在相匹配的多组初始训练样本中确定包含该待检测部位的初始训练样本,并对确定出的包含待检测部位的初始训练样本中的样本图像进行处理。
通过上述处理方式,可以从多组初始训练样本中确定出与待训练网络模型相匹配的初始训练样本,在根据该相匹配的初始训练样本进行网络模型的训练时,可以提高目标样本图像和训练标签的确定效率,并提高目标样本图像和训练标签准确率,从而能够得到训练精度较高的网络模型。
步骤S103中,对样本图像进行处理可以理解为:对样本图像进行遮挡处理,和/或,对样本图像进行裁剪处理。其中,遮挡处理可以理解为遮挡样本图像中除待检测部位之外的其他部位;裁剪处理可以理解为对样本图像中的待检测部位进行裁剪。
S105:确定所述肢体特征点的监督类型,并基于所述监督类型修正所述初始训练标签,以根据修正之后的初始训练标签确定所述目标样本图像的训练标签。
这里,可以基于待训练网络模型的检测任务和/或待检测部位确定每个肢体特征点的监督类型。监督类型用于指示在对待训练网络模型进行训练的过程中,是否需要对肢体特征点进行监督学习。
若该肢体特征点不属于待检测部位,那么该肢体特征点的监督类型可以确定为非监督类特征点。若该肢体特征点属于待检测部位,那么该肢体特征点的监督类型可以确定为监督类特征点。
在确定出监督类型之后,就可以确定出在待训练网络模型的训练过程中,需要进行监督学习的肢体特征点。在基于监督类型修正初始训练标签之后,就可以得到用于对待训练网络模型进行训练的目标样本图像的训练标签。
S107:通过所述目标训练样本和训练标签对所述待训练网络模型进行有监督训练。
在一个可选的实施方式中,在通过目标训练样本和训练标签,对待训练网络模型进行有监督训练之后,还可以基于目标训练样本和训练标签构建一组新的初始训练样本,此时,还可以为该新的初始训练样本添加样本标签,例如,该样本标签可以用于指示该新的初始训练样本所对应目标对象的对象类型,和/或,该新的初始训练样本中所包含的目标对象的肢体部位的部位信息。
通过创建新的初始训练样本,可以对初始训练样本进行扩充,从而使得该初始训练样本能够满足更加丰富的训练需求,同时,能够提高初始训练样本和对应训练任务的匹配度,从而进一步加快神经网络的训练速度。
在本公开实施例中,通过对初始训练样本中的样本图像和初始训练标签进行修正处理,得到目标样本图像和训练标签的方式,可以实现初始训练样本的复用,以使初始训练样本可以同时用于多个不同的检测任务,进而通过节省数据标注成本的方式降低神经网络的训练成本。
针对上述步骤S101,在一个可选的实施方式中,可以通过以下所描述的方式获取初始训练样本,具体包括如下过程:
首先,创建样本库,在该样本库中包含多个样本数据集,且每个样本数据集包含对应的样本标签,该样本标签用于指示每个样本数据集所对应的检测任务。
这里,样本库中的每个样本数据集即为上述过程中所描述的每组初始训练样本。这里,样本标签通过指示每个样本数据集所对应的检测任务,来确定每个样本数据集所对应目标对象的对象类型,和/或,该样本数据集中所包含的目标对象的肢体部位的部位信息。
之后,可以根据待训练网络模型的检测任务,从样本库中查找相对应的样本数据集作为初始训练样本。
例如,在该样本库中可以包含用于进行肢体检测的样本数据集A。为了提高样本数据集的复用程度,可以设置样本数据集A中的样本图像包含目标对象的全部肢体部位(上述检测部位),以及设置样本数据集A中包含该全部肢体部位的肢体关键点(也即,上述特征点)。
在从样本库中查找到初始训练样本之后,就可以根据待训练网络模型对应的检测任务,对选择出的初始训练样本中的样本图像进行处理,得到目标样本图像。之后,可以确定针对当前时刻待训练网络模型,选择出的初始训练样本中每个肢体特征点的监督类型,并基于该监督类型对肢体特征点的初始训练标签进行修正,从而得到目标训练样本的训练标签。
在一个可选的实施方式中,上述初始训练样本中的样本图像为对原始图像执行目标操作之后确定的图像,其中,原始图像为在场景复杂度不满足预设复杂要求的场景下采集到的包含目标对象的至少一个肢体部位的图像,所述目标操作包括以下至少之一:背景替换操作、调整图像亮度的操作、调整图像曝光度的操作、添加反光图层的操作。
在本公开实施例中,可以获取待训练网络模型的应用场景,进而,基于该应用场景确定与之相匹配的目标操作的操作类型;之后,基于该操作类型对原始图像执行相应的目标操作,得到样本图像。
通过基于应用场景确定与之相匹配的目标操作的操作类型的方式,可以使得初始训练样本中的样本图像能够更加接近待训练网络模型的真实应用场景,从而可以提高待训练网络模型的处理精度。
具体实施时,为了提高样本图像中所包含每个肢体部位的完整度,以及提高样本图像的清晰度,可以设置在简单背景(即,场景复杂度不满足预设复杂要求的场景)下采集原始图像,例如,可以在绿幕背景下采集包含目标对象的至少一个检测部位的原始图像。例如,可以采集至少一张包含目标人体的全部肢体部位的原始图像。
这里,简单场景可以理解为绿幕背景,或者,不包含或者包含数量减少的其他物体的任意一种背景下的场景。简单来说,上述简单场景可以理解为不会对目标对象的任意肢体部位进行遮挡,且能够对目标对象的各个肢体部位进行准确识别的场景。
在采集到原始图像之后,可以确定原始图像中的背景图像,并确定多个预设背景图像,从而通过多个预设背景图像对每张原始图像中的背景图像进行替换,得到初始训练样本中的样本图像。
在对原始图像中的背景图像进行替换时,可以确定多个预设背景图像,其中,每个预设背景图像可以为包含多种元素的复杂背景图像。然后,按照多个预设背景图像对原始图像中的背景图像进行替换,得到初始训练样本中的样本图像。
假设,原始图像的数量为1万,多个预设背景图像的数量为10个,那么通过10个预设背景图像对每个原始图像进行背景替换,可以得到10万个样本图像。如果多个预设背景图像的数量为100个,那么可以得到100万个样本图像。因此,通过上述处理方式还可以通过较低的成本,在较短的时间内获得海量训练样本,以提升模型鲁棒性。
这里,多个预设背景图像包含多种类型,其中,预设背景图像的类型与待训练网络模型的应用场景相关联。
为了提高待训练网络模型的处理精度,可以根据该网络模型的应用场景,确定与该应用场景相匹配的多个预设背景图像。
考虑到不同应用场景下,图像中所包含的背景元素可能是不相同的,因此,为了提高待训练网络模型的训练精度,可以根据待训练网络模型的应用场景,选择与该应用场景相匹配的预设背景图像进行原始图像的背景替换。
假设,待训练网络模型的应用场景为直播场景,考虑到直播场景的特殊性,可以在预设背景图像中选择与该直播场景相匹配的背景图像进行背景替换。这里,与直播场景相匹配的背景图像可以理解为模拟现实中的直播场景所构建的包含多种直播元素的复杂背景图像。
举例来说,待训练网络模型的应用场景为直播场景,该待训练网络模型的检测任务为对主播(目标对象)的半身肢体动作进行捕捉。
在直播场景中,复杂的背景会影响对主播半身肢体动作的捕捉,因为主播的肢体部位被画面边缘截断时,背景图像中一些跟主播的衣服颜色、肤色相近,或者肢体形状相似的物体会影响对主播肢体动作的捕捉。因此,在真实直播场景中,直播背景往往是复杂多变的,这时捕捉画面将会极度不稳定,从而将严重影响网络模型的处理精度。此时,可以获取包含多种直播元素的复杂背景图像,并基于该复杂背景图像替原始图像的背景图像,从而得到样本图像。
在本公开实施例中,通过对原始图像进行背景替换,可以实现通过较低的成本,在较短的时间内获得海量训练样本,以提升模型鲁棒性。上述实施方式中,通过与待训练网络模型的应用场景相匹配的背景图像与原始图像进行背景替换,可以使得初始训练样本中的样本图像能够更加接近待训练网络模型的真实应用场景,从而可以提高待训练网络模型的处理精度。
在本公开实施例中,除了可以按照上述所描述的方式对原始图像执行背景替换操作之外,还可以对原始图像执行以下至少一种目标操作(也即图像增强操作):调整图像亮度的操作、调整图像曝光度的操作、添加反光图层的操作。
具体实施时,可以根据待训练网络模型的应用场景,确定与之相匹配的图像增强操作。通过该处理方式,可以使得初始训练样本中的样本图像能够更加接近待训练网络模型的真实应用场景,从而可以提高待训练网络模型的处理精度。
举例来说,在直播场景下,除了复杂的背景之外,光照也是影响肢体检测性能的一个重要影响因素。例如,光线强弱,玻璃反光等也会影响到网络模型的检测性能。因此,在真实直播场景中,背景往往是复杂多变的,如背景中有与主播肤色相近的毛绒玩具、反光的玻璃墙面、昏暗或曝光的拍摄环境,这时捕捉到的画面将会极度不稳定。
基于此,在本公开实施例中,可以首先采集至少一张包含目标对象的至少一个肢体部位的原始图像。例如,可以采集至少一张包含目标人体的全部肢体部位的原始图像。
之后,对采集到的原始图像进行以下至少之一图像增强操作:调整图像亮度的操作、调整图像曝光度的操作、添加反光图层的操作;处理之后就可以得到样本图像。
上述实施方式中,通过对原始图像进行调整图像亮度、调整图像曝光度以及添加反光图层中的至少一种图像增强处理,可以提高样本图像的背景复杂度;通过基于待训练网络模型的应用场景确定与之相匹配的图像增强操作,可以使得样本图像能够更加贴近真实的应用场景,以提高待训练网络模型的处理精度。
在一个可选的实施方式中,针对上述步骤S103,对所述样本图像进行处理,得到包含所述待检测部位的目标样本图像,具体包括如下过程:
步骤S1031,对所述样本图像中的目标检测部位进行图像遮挡处理,得到所述目标样本图像;其中,所述目标检测部位为所述肢体部位中除所述待检测部位之外的其他部位。
和/或
步骤S1032,对所述样本图像中位于目标区域内的图像进行图像裁剪处理,裁剪得到包含所述待检测部位的目标样本图像;所述目标区域为所述样本图像中包含所述待检测部位的图像区域。
在本公开实施例中,首先确定待训练网络模型的检测任务,其中,该检测任务用于指示该待训练网络模型所需要进行检测的待检测部位的部位信息。
这里,部位信息可以为每个待检测部位的部位名称、部位编码、部位数量等信息,以及每个待检测部位的完整度需求。其中,完整度需求可以包含以下至少之一:要求完整、要求不完整、在要求不完整的情况下肢体不完整的程度。
例如,检测任务为对目标人体的上半身肢体进行检测,此时,待检测部位包含以下部位:上身躯干、手臂、头部和手部。
在确定出待检测部位的部位信息之后,可以根据该部位信息对样本图像进行图像遮挡处理和/或图像裁剪处理。
具体实施时,可以通过指定颜色的图像(例如,黑色图像)对目标检测部位进行图像遮挡处理,得到目标样本图像;和/或,对样本图像中位于目标区域内的图像进行图像裁剪处理,裁剪得到包含所述待检测部位的目标样本图像。
应理解的是,针对初始训练样本来说,要求该初始训练样本能够满足相同类型的检测任务的多种任务需求。其中,相同类型的检测任务可以理解为:检测任务均为肢体检测;不同检测任务的多种任务需求可以理解为:对不同的类型的肢体部位进行肢体检测。
假设,检测任务为肢体检测任务,那么该初始训练样本中应当包含目标人体的全部肢体部位,从而使得该初始训练样本能够满足相同类型的检测任务的多种任务需求。
上述实施方式中,通过对样本图像进行图像遮挡处理和/或图像裁剪处理,可以在不重新采集样本图像的基础上,得到与检测任务相匹配的样本图像,从而实现样本图像的复用,以降低神经网络的训练时间和训练成本。
在一个可选的实施方式中,如图2所示,针对上述步骤S105,确定所述肢体特征点的监督类型,并基于所述监督类型修正所述初始训练标签,具体包括如下过程:
步骤S1051,确定所述肢体特征点的特征信息,其中,所述特征信息用于指示对应肢体特征点和所述目标样本图像之间的位置关系,和/或,用于指示对应肢体特征点是否为待检测部位的特征点;
步骤S1052,基于所述特征信息,确定所述肢体特征点的监督类型;
步骤S1053,在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签。
在本公开实施例中,在按照上述内容所描述的方式对样本图像进行图像遮挡处理和/或图像裁剪处理,得到目标样本图像之后,其实是从样本图像中删除了该检测任务所不需要进行检测的肢体部位,也即,不需要进行监督学习的肢体部位。此时,还需要对该检测任务所不需要进行监督学习的肢体部位的肢体特征点的初始训练标签进行修正。
在对初始训练标签进行修正时,可以确定肢体特征点的特征信息,进而根据该特征信息确定该肢体特征点是否为需要进行监督学习的肢体特征点。
在本公开实施例中,按照肢体特征点的特征信息,可以将肢体特征点的监督类型确定为以下几种类型:
类型一:肢体特征点为监督类特征点,且该肢体特征点位于目标样本图像内。
类型二:肢体特征点为监督类特征点,且该肢体特征点位于目标样本图像外。
针对该类型二,可以理解为该肢体特征点并未包含在目标样本图像内,但是,需要进行监督学习的肢体特征点。
例如,待检测部位为人体的上半身(头部、上身躯干、两个手臂和两个手部)。由于在对样本图像进行图像遮挡处理和/或图像裁剪处理时,手部的部分部位可能被遮挡和/或裁剪。但是,被遮挡和/或被裁剪的手部部位依然为需要进行检测的肢体部位。此时,被遮挡和/或被裁剪的手部部位所对应的肢体特征点的类型即为上述类型二。
类型三:肢体特征点为非监督类特征点,且该肢体特征点位于目标样本图像外。
在基于上述特征信息确定出每个肢体特征点的监督类型之后,就可以基于监督类型确定肢体特征点的初始训练标签是否满足修正条件。
具体实施时,如果判断出肢体特征点的监督类型为上述类型二或者类型三,则确定肢体特征点的初始训练标签满足修正条件。此时,就可以监督类型修正该肢体特征点的初始训练标签。如果判断出肢体特征点的监督类型为类型一,则确定肢体特征点的初始训练标签不满足修正条件。
在本公开实施例中,在确定出肢体特征点的初始训练标签满足修正条件的情况下,对初始训练标签进行修正的目的是为了对初始训练样本中的初始训练标签进行处理,从而得到能够对待训练网络模型进行训练的训练标签。
上述实施方式中,通过确定肢体特征点的监督类型,并基于该监督类型对肢体特征点的初始训练标签进行修正的方式,可以保留需要进行监督的肢体特征点的初始训练标签,同时还可以提高初始训练样本中初始训练标签的利用率。
在现有的神经网络的训练过程中,在进行数据标注时,只能标注画面中可见的肢体特征点的位置信息。然而,在本公开技术方案中,通过监督类型对肢体特征点的初始训练标签进行修正的方式,不仅可以监督学习位于目标样本图像内的肢体特征点,还可以监督学习位于目标样本图像之外且需要进行监督学习的肢体特征点。在基于修正之后的初始训练标签(即,训练标签)和目标样本图像对待训练网络模型进行训练时,就可以实现通过该网络模型对未包含在图像中的部分肢体部位的肢体特征点进行预测。
在一些姿态识别的场景中,由于待识别目标的部分肢体可能未包含在待识别图像中,此时,通过现有训练方式训练得到的网络模型对该待识别图像中待识别目标进行姿态识别时,可能会出现由于无法识别到完整的肢体特征点,从而导致姿态识别失败的问题。然后,在采用本公开技术方案中确定出的修正之后的初始训练标签和目标样本图像,对待训练网络模型进行训练时,可以对未包含在图像中的部分肢体部位的肢体特征点进行预测,从而实现待检测目标的姿态识别。因此,通过本公开技术方案进行训练之后的网络模型的鲁棒性更强,且识别过程更加稳定。
基于肢体特征点的类型信息,可以确定未包含在处理之后的样本图像中的肢体特征点是否为需要进行检测的特征点。在根据该类型信息对肢体特征点的初始训练标签进行修正时,可以提高初始训练标签的准确性,从而,进而提高了待训练网络模型的训练精度。
下面,将具体介绍对肢体特征点的初始训练标签进行修正的修成过程。
在一个可选的实施方式中,上述步骤S1053,在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签,包括如下过程:
在确定出肢体特征点的监督类型为上述类型二的情况下,将所述肢体特征点的初始训练标签修正为第二初始训练标签。
具体实施时,在根据所述初始训练标签确定出所述肢体特征点位于所述目标样本图像之外,且属于待检测部位的特征点的情况下,将所述肢体特征点的初始训练标签修正为第二初始训练标签;所述第二初始训练标签用于表征该肢体特征点为位于所述目标样本图像之外的监督类特征点。
在本公开实施例中,上述待训练网络模型的一个应用场景可以为对不包含完整检测部位的图像进行检测,从而得到完整检测部位的特征点。
例如,一张图像中包含上半身肢体部位,且在该图像中包含部分手部部位,此时,该待训练网络模型可以对该上半身肢体部位进行肢体检测。具体地,可以对包含在该图像中的肢体部位进行肢体检测,还可以对未包含在该图像中且需要进行检测的肢体部位进行肢体检测,从而得到肢体特征点,其中,该肢体特征点包含未包含在该图像中的手部的特征点。
此时,在对该待训练网络模型进行训练时,需要构建包含不完整检测部位的样本图像,例如,包含不完整上半身肢体部位的样本图像。然而,针对该样本图像,需要设置未包含在样本图像中需要进行检测的肢体部位的肢体特征点的初始训练标签。
在此情况下,可以确定肢体特征点是否为待训练网络模型所需要进行处理的待检测部位的特征点。如果是,则需要将该肢体特征点的初始训练标签修改为第二初始训练标签,表征该肢体特征点为未包含目标样本图像中的监督类特征点。
举例来说,在获取到初始训练样本之后,可以对初始训练样本中的样本图像进行图像遮挡处理或者图像裁剪处理,得到不包含完整上半身肢体的样本图像。此时,被遮挡或者被裁剪的肢体部位未包含在该处理之后的样本图像中。然而,该被遮挡或者被裁剪的肢体部位是待训练网络模型所需要进行检测的肢体部位。因此,需要对初始训练样本中该样本图像的初始训练标签进行修正,具体修正过程为:将初始训练标签中被遮挡或者被裁剪的肢体部位所对应的肢体特征点(肢体特征点)的初始训练标签修改为第二初始训练标签,以通过该第二初始训练标签表征肢体特征点为未包含目标样本图像中的监督类特征点。
上述实施方式中,通过上述所描述的初始训练标签的修正方式,可以使初始训练样本可以同时用于多个不同的检测任务,进而通过节省数据标注成本的方式降低神经网络的训练成本。进一步地,在采用本公开技术方案中确定出的修正之后的初始训练标签和目标样本图像,对待训练网络模型进行训练时,可以对未包含在图像中的部分肢体部位的肢体特征点进行预测,从而实现待检测目标的姿态识别。因此,通过本公开技术方案进行训练之后的网络模型的鲁棒性更强,且识别过程更加稳定。
在另一个可选的实施方式中,上述步骤S1053,在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签,包括如下过程:
在确定出肢体特征点的监督类型为上述类型三的情况下,将所述肢体特征点的初始训练标签修正为第一初始训练标签。
具体实施时,在根据所述监督类型确定出所述肢体特征点位于所述目标样本图像之外,且不是所述待检测部位的特征点情况下,将所述肢体特征点的初始训练标签修正为第一初始训练标签;所述第一初始训练标签用于表征该肢体特征点为位于所述目标样本图像之外的非监督类特征点。
举例来说,待训练网络模型用于对上半身人体肢体部位进行肢体检测。初始训练样本的样本图像中包含完整人体肢体(即,上半身人体肢体部位和下半身人体肢体部位)。
此时,需要对样本图像进行裁剪,得到包含完整或者不完整上半身人体肢体部位的样本图像。其中,上半身人体肢体部位的肢体特征点即为上述指定特征点。
此时,需要将样本图像中不属于待检测部位的肢体特征点的初始训练标签进行修正,例如,修正下半身人体肢体检测部位的肢体特征点的初始训练标签。
这里,可以将肢体特征点的初始训练标签设置为第一初始训练标签,用于表征该肢体特征点为位于目标样本图像之外的非监督类特征点。
在一个可选的实施方式中,可以将该肢体特征点的位置信息修改为(0,0),其中,(0,0)表示该肢体特征点为目标样本图像左上角的点。除此之外,还可以删除该肢体特征点的位置信息,并为该肢体特征点添加相应的标识信息,以通过该标识信息指示该肢体特征点位于目标样本图像之外。
上述实施方式中,通过上述所描述的初始训练标签的修正方式,可以使初始训练样本可以同时用于多个不同的检测任务,进而通过节省数据标注成本的方式降低神经网络的训练成本。
针对上述步骤S107,在待检测部位为多个的情况下,通过所述目标样本图像和所述训练标签,对所述待训练网络模型进行训练,具体包括如下过程:
步骤S1051,通过所述待训练网络模型对所述目标样本图像进行图像处理,得到所述目标样本图像中每个待检测部位的位置信息;
步骤S1052,基于多个所述待检测部位中第一部位和第二部位的位置信息,以及所述训练标签,确定用于约束所述第一部位和第二部位之间的位置差异的目标损失函数的函数值,其中,所述第一部位和所述第二部位为具有关联关系的待检测部位;
步骤S1053,根据所述目标损失函数的函数值调整所述待训练网络模型的模型参数,以通过调整之后的模型参数对所述待训练网络模型进行训练。
在本公开实施例中,在按照上述所描述的方式得到目标训练样本和训练标签之后,就可以通过待训练网络模型对目标训练样本进行图像处理,得到各个待检测部位的位置信息。
为了提高待训练网络模型的处理精度,还可以基于待检测部位中的第一部位和第二部位的位置信息,以及训练标签确定目标损失函数。其中,该目标损失函数用于约束第一部位和第二部位之间的位置差异。
具体地,可以计算第一部位的位置信息和第二部位的位置信息之间的差值,并基于该差值和训练标签构建目标损失函数。然后,计算目标损失函数的函数值满足对应约束条件时,待训练网络模型的模型参数的最优解。进而,根据该最优解对待训练网络模型进行训练。
这里,第一部位和第二部位可以为具有联动关系的检测部位。例如,第一部位在第二部位的带动下进行移动,或者,第二部位在第一部位的带动下进行移动。假设,第一部位为手部,那么第二部位可以为手腕部位。第一部位可以手腕部位,第二部位可以为小臂部位。此处不再一一列举。
在本公开实施例中,针对不相同的第一部位和第二部位可以设置不同的约束条件。例如,可以根据第一部位和第二部位的部位类型设置对应的约束条件。
上述实施方式中,通过第一部位和第二部位的位置信息之间的位置差异构建目标损失函数的方式,可以减少第一部位和第二部位之间位置差异较大的现象,从而提高待训练网络模型的处理精度。
参见图3所示,为本公开实施例提供的一种姿态检测方法的流程图,所述方法包括步骤S301~S307,其中:
S301:获取包含目标主播的肢体部位的视频图像帧;
S303:通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,其中,所述姿态检测模型为利用目标样本图像及对应的训练标签进行监督训练得到,所述目标样本图像包含待检测的肢体部位样本,所述训练标签基于所述肢体部位样本的肢体特征点的监督类型及所述肢体特征点对应的初始训练标签进行调整得到;
S305:在得到所述目标主播的姿态检测结果之后,根据所述姿态检测结果生成所述目标主播所对应的虚拟主播的姿态触发信号;
S307:根据所述姿态触发信号控制所述虚拟主播执行相应的触发动作。
在本公开实施例中,在获取到视频图像帧之后,通过姿态检测模型对目标主播进行姿态检测,进而根据姿态检测结果控制虚拟主播执行相应触发动作的方式,可以实现自动的通过目标主播的姿态检测结果在视频直播界面上触发展示虚拟主播的相应部位执行相应触发操作,进而实现触发虚拟主播执行与主播的肢体产生动作关联的动作,从而提高实际互动效果。同时,通过确定待检测的肢体部位的姿态检测结果对虚拟主播进行触发,还可以实现虚拟主播中对应触发部位的精准触发,从而满足用户的丰富触发需求。同时,在本公开技术方案中,通过基于肢体部位样本的肢体特征点的监督类型及肢体特征点对应的初始训练标签进行调整得到训练标签,可以通过对初始训练标签的复用来训练得到姿态检测模型,通过该方式可以节省数据标注成本,并加快数据标注的速度。
在直播场景中的情况下,复杂的背景会影响对主播半身肢体动作的捕捉,因为主播(即,上述目标对象)的肢体部位被画面边缘截断时,背景图像中一些跟主播的衣服颜色、肤色相近,或者肢体形状相似的物体会影响对主播肢体动作的捕捉。因此,在真实直播场景中,直播背景往往是复杂多变的,这时捕捉画面将会极度不稳定,从而将严重影响网络模型的处理精度。
在本公开实施例中,通过上述所描述的神经网络模型的训练方法,构建目标训练样本和训练标签的方式,可以得到更加接近真实的直播场景的样本图像。在根据该目标训练样本和训练标签训练得到姿态检测模型之后,可以得到处理精度较高的姿态检测模型。在根据该姿态检测模型检测目标主播的肢体部位时,可以得到更加准确的姿态检测结果。
在本公开实施例中,在得到目标主播的姿态检测结果之后,可以根据该姿态检测结果生成触发虚拟主播执行相应触发动作的姿态触发信号,以触发虚拟主播执行相应的触发动作。
由于在根据该姿态检测模型检测目标主播的肢体部位时,可以得到更加准确的检测结果,因此,在根据该姿态检测结果触发虚拟主播执行相应的触发动作时,可以实现准确的控制虚拟主播执行相应的触发动作。
在一个可选的实施方式中,通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,具体包括如下过程:
(1)、在所述视频图像帧中检测目标框体,其中,所述目标框体用于框选所述视频图像帧中所述目标对象的肢体部位;
(2)、通过所述姿态检测模型对所述视频图像帧中位于所述目标框体内的图像进行姿态检测,得到所述姿态检测结果。
在本公开实施例中,在获取到视频图像帧之后,可以在视频图像帧中检测目标框体。其中,该目标框体为用于框选目标对象的肢体部位的框体。
这里,若视频图像帧中包含多个对象,则需要在多个对象中确定目标对象。在一个可选的实施方式中,可以在检测到视频图像帧中包含多个对象的情况下,将多个对象中位于画面最前方的对象确定为目标对象。在另一个可选的实施方式中,可以确定每个对象的框体,然后,将确定出的多个框体中,面积最大的框体确定为目标框体,并将该目标框体所框选的对象确定为目标对象。
在确定出目标框体之后,就可以通过采用上述所描述的神经网络模型的训练方法训练之后得到的姿态检测模型对位于该目标框体内的图像进行姿态检测,得到姿态检测结果。
上述实施方式中,通过检测包含目标对象的目标框体,并对位于该目标框体内的图像进行姿态检测,可以提高姿态检测结果的准确性。
在一个可选的实施方式中,上述步骤:通过姿态检测模型对所述视频图像帧中的所述目标对象进行姿态检测,得到姿态检测结果,具体包括如下过程:
(1)、在所述视频图像帧中未检测到所述目标对象的部分指定肢体部位的情况下,对所述视频图像帧进行处理,得到目标图像;所述目标图像中包含用于对该部分指定肢体部位进行姿态检测的区域。
对视频图像帧进行处理,得到目标图像的过程可以描述如下:
在视频图像帧中截取包含待检测部位的子图像,基于部分指定肢体部位的肢体类型信息和/或肢体尺寸信息,对该子图像进行边缘填补,得到包含用于对该部分指定肢体部位进行肢体检测的填补区域的目标图像。
(2)、通过所述姿态检测模型对所述目标图像进行姿态检测,得到所述目标主播的姿态检测结果。
在一个可选的实施方式中,该姿态检测模型能够对包含不完整的指定肢体部位进行姿态检测。
因此,在检测到视频图像帧中未检测到完整的指定肢体部位的情况下,可以确定该视频图像帧中不包含部分指定肢体部位(例如,缺少部分手部)。此时,可以对视频图像帧进行边缘填补处理,得到包含该部分指定部位的检测区域(即,上述填补区域)的目标图像。
在一个可选的实施方式中,可以对该视频图像帧进行边缘填补处理,以在该视频图像帧的边缘增加黑色区域,从而得到目标图像。
在对视频图像帧进行边缘填补处理,得到目标图像之后,就可以通过该姿态检测模型对目标图像进行姿态检测,得到目标对象的姿态检测结果。
上述实施方式中,在视频图像帧中未检测到目标对象的部分指定肢体部位的情况下,通过对视频图像帧进行边缘填补处理,得到目标图像,以根据该目标图像进行姿态检测的方式,可以实现在视频图像帧中不包含完整的指定肢体部位的情况下,依然可以对该视频图像帧进行姿态检测,得到准确的姿态检测结果。
在一个可选的实施方式中,上述步骤:基于所述肢体部位样本的肢体特征点的监督类型及所述肢体特征点对应的初始训练标签进行调整得到所述训练标签,包括:
(1)、确定所述肢体特征点的监督类型;
(2)、基于所述监督类型修正所述初始训练标签,以根据修正之后的初始训练标签确定所述目标样本图像的训练标签。
这里,可以基于待训练的姿态检测模型确定每个肢体特征点的监督类型。监督类型用于指示在对待训练的姿态检测模型进行训练的过程中,是否需要对肢体特征点进行监督学习。
若该肢体特征点不属于待检测部位,那么该肢体特征点的监督类型可以确定为非监督类特征点。若该肢体特征点属于待检测部位,那么该肢体特征点的监督类型可以确定为监督类特征点。
在确定出监督类型之后,就可以确定出在待训练的姿态检测模型的训练过程中,需要进行监督学习的肢体特征点。在基于监督类型修正初始训练标签之后,就可以得到用于对待训练的姿态检测模型进行训练的目标样本图像的训练标签。
在一个可选的实施方式中,上述步骤:确定所述肢体特征点的监督类型,具体包括如下步骤:
首先,确定所述肢体特征点的特征信息,其中,所述特征信息用于指示对应肢体特征点和所述目标样本图像之间的位置关系,和/或,用于指示对应肢体特征点是否为待检测部位的特征点;
其次,基于所述特征信息,确定所述肢体特征点的监督类型;并在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签,得到所述训练标签。
在本公开实施例中,在对初始训练标签进行修正时,可以确定肢体特征点的特征信息,进而根据该特征信息确定该肢体特征点是否为需要进行监督学习的肢体特征点。
在本公开实施例中,按照肢体特征点的特征信息,可以将肢体特征点的监督类型确定为以下几种类型:
类型一:肢体特征点为监督类特征点,且该肢体特征点位于目标样本图像内。
类型二:肢体特征点为监督类特征点,且该肢体特征点位于目标样本图像外。
针对该类型二,可以理解为该肢体特征点并未包含在目标样本图像内,但是,需要进行监督学习的肢体特征点。
类型三:肢体特征点为非监督类特征点,且该肢体特征点位于目标样本图像外。
在基于上述特征信息确定出每个肢体特征点的监督类型之后,就可以基于监督类型确定肢体特征点的初始训练标签是否满足修正条件。
具体实施时,如果判断出肢体特征点的监督类型为上述类型二或者类型三,则确定肢体特征点的初始训练标签满足修正条件。此时,就可以监督类型修正该肢体特征点的初始训练标签。如果判断出肢体特征点的监督类型为类型一,则确定肢体特征点的初始训练标签不满足修正条件。
在本公开实施例中,在确定出肢体特征点的初始训练标签满足修正条件的情况下,对初始训练标签进行修正的目的是为了对初始训练样本中的初始训练标签进行处理,从而得到能够对待训练的姿态检测模型进行训练的训练标签。
上述实施方式中,通过确定肢体特征点的监督类型,并基于该监督类型对肢体特征点的初始训练标签进行修正的方式,可以保留需要进行监督的肢体特征点的初始训练标签,同时还可以提高初始训练样本中初始训练标签的利用率。
本公开涉及增强现实领域,通过获取现实环境中的目标对象的图像信息,进而借助各类视觉相关算法实现对目标对象的相关特征、状态及属性进行检测或识别处理,从而得到与具体应用匹配的虚拟与现实相结合的AR效果。示例性的,目标对象可涉及与人体相关的脸部、肢体、手势、动作等。具体应用不仅可以涉及跟真实场景或物品相关的导览、导航、讲解、重建、虚拟效果叠加展示等交互场景,还可以涉及与人相关的特效处理,比如妆容美化、肢体美化、特效展示、虚拟模型展示等交互场景。可通过卷积神经网络,实现对目标对象的相关特征、状态及属性进行检测或识别处理。
本领域技术人员可以理解,在具体实施方式的上述方法中,各步骤的撰写顺序并不意味着严格的执行顺序而对实施过程构成任何限定,各步骤的具体执行顺序应当以其功能和可能的内在逻辑确定。
基于同一发明构思,本公开实施例中还提供了与姿态检测方法对应的姿态检测装置,由于本公开实施例中的装置解决问题的原理与本公开实施例上述神经网络模型的训练方法相似,因此装置的实施可以参见方法的实施,重复之处不再赘述。
参照图4所示,为本公开实施例提供的一种姿态检测装置的示意图,所述装置包括:第一获取单元41、姿态检测单元42、生成单元43、控制单元44;其中:
第一获取单元41,用于获取包含目标主播的肢体部位的视频图像帧;
姿态检测单元42,用于通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,其中,所述姿态检测模型为利用目标样本图像及对应的训练标签进行监督训练得到,所述目标样本图像包含待检测的肢体部位样本,所述训练标签基于所述肢体部位样本的肢体特征点的监督类型及所述肢体特征点对应的初始训练标签进行调整得到;
生成单元43,用于在得到所述目标主播的姿态检测结果之后,根据所述姿态检测结果生成所述目标主播所对应的虚拟主播的姿态触发信号;
控制单元44,用于根据所述姿态触发信号控制所述虚拟主播执行相应的触发动作。
一种可能的实施方式中,姿态检测单元42,还用于:在所述视频图像帧中检测目标框体,其中,所述目标框体用于框选所述视频图像帧中所述目标主播的肢体部位;通过所述姿态检测模型对所述视频图像帧中位于所述目标框体内的图像进行姿态检测,得到所述姿态检测结果。
一种可能的实施方式中,姿态检测单元42,还用于:在所述视频图像帧中未检测到所述目标主播的至少部分指定肢体部位的情况下,对所述视频图像帧进行处理,得到目标图像;所述目标图像中包含用于对该至少部分指定肢体部位进行姿态检测的区域;通过所述姿态检测模型对所述目标图像进行姿态检测,得到所述目标主播的姿态检测结果。
一种可能的实施方式中,该装置还用于:确定所述肢体特征点的监督类型;基于所述监督类型修正所述初始训练标签,以根据修正之后的初始训练标签确定所述目标样本图像的训练标签。
一种可能的实施方式中,该装置还用于:确定所述肢体特征点的特征信息,其中,所述特征信息用于指示对应肢体特征点和所述目标样本图像之间的位置关系,和/或,用于指示对应肢体特征点是否为待检测部位的特征点;基于所述特征信息,确定所述肢体特征点的监督类型;并在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签,得到所述训练标签。
基于同一发明构思,本公开实施例中还提供了与神经网络模型的训练方法对应的神经网络模型的训练装置,由于本公开实施例中的装置解决问题的原理与本公开实施例上述神经网络模型的训练方法相似,因此装置的实施可以参见方法的实施,重复之处不再赘述。
参照图5所示,为本公开实施例提供的一种神经网络模型的训练装置的示意图,所述装置包括:第二获取单元51、修正处理单元52、确定单元53、训练单元54;其中,
第一获取单元51,用于获取初始训练样本;所述初始训练样本中包含样本图像和该样本图像中所包含的目标对象的肢体部位的肢体特征点的初始训练标签;
修正处理单元52,用于基于待训练网络模型的检测任务在所述肢体部位中确定待检测部位;并对所述样本图像进行处理,得到包含所述待检测部位的目标样本图像;
确定单元53,用于确定所述肢体特征点的监督类型,并基于所述监督类型修正所述初始训练标签,以根据修正之后的初始训练标签确定所述目标样本图像的训练标签;
训练单元54,用于通过所述目标样本图像和所述训练标签,对所述待训练网络模型进行有监督训练。
通过上述描述可知,本公开技术方案在确定出待训练网络模型之后,通过对初始训练样本中的样本图像和初始训练标签进行修正处理,得到目标样本图像和训练标签的方式,可以实现初始训练样本的复用,以使初始训练样本可以同时用于多个不同的待训练网络模型的训练过程,进而通过节省数据标注成本的方式降低神经网络的训练成本。
一种可能的实施方式中,在所述待检测部位为多个的情况下,训练单元,还用于:通过所述待训练网络模型对所述目标样本图像进行图像处理,得到所述目标样本图像中每个待检测部位的位置信息;基于多个所述待检测部位中第一部位和第二部位的位置信息,以及所述训练标签,确定用于约束所述第一部位和第二部位之间的位置差异的目标损失函数的函数值,其中,所述第一部位和所述第二部位为具有关联关系的待检测部位;根据所述目标损失函数的函数值调整所述待训练网络模型的模型参数,以通过调整之后的模型参数对所述待训练网络模型进行训练。
一种可能的实施方式中,确定单元,还用于:确定所述肢体特征点的特征信息,其中,所述特征信息用于指示对应肢体特征点和所述目标样本图像之间的位置关系,和/或,用于指示对应肢体特征点是否为待检测部位的特征点;基于所述特征信息,确定所述肢体特征点的监督类型;在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签。
一种可能的实施方式中,确定单元,还用于:在根据所述监督类型确定出所述肢体特征点位于所述目标样本图像之外,且不是所述待检测部位的特征点情况下,将所述肢体特征点的初始训练标签修正为第一初始训练标签;所述第一初始训练标签用于表征该肢体特征点为位于所述目标样本图像之外的非监督类特征点。
一种可能的实施方式中,确定单元,还用于:在根据所述初始训练标签确定出所述肢体特征点位于所述目标样本图像之外,且属于待检测部位的特征点的情况下,将所述肢体特征点的初始训练标签修正为第二初始训练标签;所述第二初始训练标签用于表征该肢体特征点为位于所述目标样本图像之外的监督类特征点。
一种可能的实施方式中,所述样本图像为对原始图像执行目标操作之后确定的图像,其中,所述原始图像为在场景复杂度不满足预设复杂要求的场景下采集到的包含目标对象的至少一个肢体部位的图像,所述目标操作包括以下至少之一:背景替换操作、调整图像亮度的操作、调整图像曝光度的操作、添加反光图层的操作。
一种可能的实施方式中,修正处理单元,还用于:对所述样本图像中的目标检测部位进行图像遮挡处理,得到所述目标样本图像;其中,所述目标检测部位为所述肢体部位中除所述待检测部位之外的其他部位;和/或,对所述样本图像中位于目标区域内的图像进行图像裁剪处理,裁剪得到包含所述待检测部位的目标样本图像;所述目标区域为所述样本图像中包含所述待检测部位的图像区域。
关于装置中的各模块的处理流程、以及各模块之间的交互流程的描述可以参照上述方法实施例中的相关说明,这里不再详述。
对应于图1中的姿态检测方法,本公开实施例还提供了一种计算机设备600,如图6所示,为本公开实施例提供的计算机设备600结构示意图,包括:
处理器61、存储器62、和总线63;存储器62用于存储执行指令,包括内存621和外部存储器622;这里的内存621也称内存储器,用于暂时存放处理器61中的运算数据,以及与硬盘等外部存储器622交换的数据,处理器61通过内存621与外部存储器622进行数据交换,当所述计算机设备600运行时,所述处理器61与所述存储器62之间通过总线63通信,使得所述处理器61执行以下指令:
获取包含目标主播的肢体部位的视频图像帧;
通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,其中,所述姿态检测模型为利用目标样本图像及对应的训练标签进行监督训练得到,所述目标样本图像包含待检测的肢体部位样本,所述训练标签基于所述肢体部位样本的肢体特征点的监督类型及所述肢体特征点对应的初始训练标签进行调整得到;
在得到所述目标主播的姿态检测结果之后,根据所述姿态检测结果生成所述目标主播所对应的虚拟主播的姿态触发信号;
根据所述姿态触发信号控制所述虚拟主播执行相应的触发动作。
对应于图1中的神经网络模型的训练方法,本公开实施例还提供了一种计算机设备700,如图7所示,为本公开实施例提供的计算机设备700结构示意图,包括:
处理器71、存储器72、和总线73;存储器72用于存储执行指令,包括内存721和外部存储器722;这里的内存721也称内存储器,用于暂时存放处理器71中的运算数据,以及与硬盘等外部存储器722交换的数据,处理器71通过内存721与外部存储器722进行数据交换,当所述计算机设备700运行时,所述处理器71与所述存储器72之间通过总线73通信,使得所述处理器71执行以下指令:
获取初始训练样本;所述初始训练样本中包含样本图像和该样本图像中所包含的目标对象的肢体部位的肢体特征点的初始训练标签;
基于待训练网络模型的检测任务在所述肢体部位中确定待检测部位;并对所述样本图像进行处理,得到包含所述待检测部位的目标样本图像;
确定所述肢体特征点的监督类型,并基于所述监督类型修正所述初始训练标签,以根据修正之后的初始训练标签确定所述目标样本图像的训练标签;
通过所述目标样本图像和所述训练标签,对所述待训练网络模型进行有监督训练。
本公开实施例还提供一种计算机可读存储介质,该计算机可读存储介质上存储有计算机程序,该计算机程序被处理器运行时执行上述方法实施例中所述的神经网络模型的训练、姿态检测方法的步骤。其中,该存储介质可以是易失性或非易失的计算机可读取存储介质。
本公开实施例还提供一种计算机程序产品,该计算机程序产品承载有程序代码,所述程序代码包括的指令可用于执行上述方法实施例中所述的神经网络模型的训练、姿态检测方法的步骤,具体可参见上述方法实施例,在此不再赘述。
其中,上述计算机程序产品可以具体通过硬件、软件或其结合的方式实现。在一个可选实施例中,所述计算机程序产品具体体现为计算机存储介质,在另一个可选实施例中,计算机程序产品具体体现为软件产品,例如软件开发包(Software Development Kit,SDK)等等。
所属领域的技术人员可以清楚地了解到,为描述的方便和简洁,上述描述的***和装置的具体工作过程,可以参考前述方法实施例中的对应过程,在此不再赘述。在本公开所提供的几个实施例中,应该理解到,所揭露的***、装置和方法,可以通过其它的方式实现。以上所描述的装置实施例仅仅是示意性的,例如,所述单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,又例如,多个单元或组件可以结合或者可以集成到另一个***,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些通信接口,装置或单元的间接耦合或通信连接,可以是电性,机械或其它的形式。
所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
另外,在本公开各个实施例中的各功能单元可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中。
所述功能如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个处理器可执行的非易失的计算机可读取存储介质中。基于这样的理解,本公开的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质中,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行本公开各个实施例所述方法的全部或部分步骤。而前述的存储介质包括:U盘、移动硬盘、只读存储器(Read-OnlyMemory,ROM)、随机存取存储器(Random Access Memory,RAM)、磁碟或者光盘等各种可以存储程序代码的介质。
最后应说明的是:以上所述实施例,仅为本公开的具体实施方式,用以说明本公开的技术方案,而非对其限制,本公开的保护范围并不局限于此,尽管参照前述实施例对本公开进行了详细的说明,本领域的普通技术人员应当理解:任何熟悉本技术领域的技术人员在本公开揭露的技术范围内,其依然可以对前述实施例所记载的技术方案进行修改或可轻易想到变化,或者对其中部分技术特征进行等同替换;而这些修改、变化或者替换,并不使相应技术方案的本质脱离本公开实施例技术方案的精神和范围,都应涵盖在本公开的保护范围之内。因此,本公开的保护范围应所述以权利要求的保护范围为准。
Claims (16)
1.一种姿态检测方法,其特征在于,包括:
获取包含目标主播的肢体部位的视频图像帧;
通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,其中,所述姿态检测模型为利用目标样本图像及对应的训练标签进行监督训练得到,所述目标样本图像包含待检测的肢体部位样本,所述训练标签基于所述肢体部位样本的肢体特征点的监督类型及所述肢体特征点对应的初始训练标签进行调整得到;所述监督类型基于所述肢体特征点的特征信息确定,所述特征信息用于指示对应肢体特征点和所述目标样本图像之间的位置关系,和/或,用于指示对应肢体特征点是否为待检测部位的特征点;
在得到所述目标主播的姿态检测结果之后,根据所述姿态检测结果生成所述目标主播所对应的虚拟主播的姿态触发信号;
根据所述姿态触发信号控制所述虚拟主播执行相应的触发动作。
2.根据权利要求1所述的方法,其特征在于,所述通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,包括:
在所述视频图像帧中检测目标框体,其中,所述目标框体用于框选所述视频图像帧中所述目标主播的肢体部位;
通过所述姿态检测模型对所述视频图像帧中位于所述目标框体内的图像进行姿态检测,得到所述姿态检测结果。
3.根据权利要求1或2所述的方法,其特征在于,所述通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,包括:
在所述视频图像帧中未检测到所述目标主播的至少部分指定肢体部位的情况下,对所述视频图像帧进行处理,得到目标图像;所述目标图像中包含用于对该至少部分指定肢体部位进行姿态检测的区域;
通过所述姿态检测模型对所述目标图像进行姿态检测,得到所述目标主播的姿态检测结果。
4.根据权利要求1所述的方法,其特征在于,所述基于所述肢体部位样本的肢体特征点的监督类型及所述肢体特征点对应的初始训练标签进行调整得到所述训练标签,包括:
确定所述肢体特征点的监督类型;
基于所述监督类型修正所述初始训练标签,以根据修正之后的初始训练标签确定所述目标样本图像的训练标签。
5.根据权利要求4所述的方法,其特征在于,所述确定所述肢体特征点的监督类型,包括:
确定所述肢体特征点的特征信息;
基于所述特征信息,确定所述肢体特征点的监督类型;并在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签,得到所述训练标签。
6.一种神经网络模型的训练方法,其特征在于,包括:
获取初始训练样本;所述初始训练样本中包含样本图像和该样本图像中所包含的目标对象的肢体部位的肢体特征点的初始训练标签;
基于待训练网络模型的检测任务在所述肢体部位中确定待检测部位;并对所述样本图像进行处理,得到包含所述待检测部位的目标样本图像;
基于所述肢体特征点的特征信息确定所述肢体特征点的监督类型,并基于所述监督类型修正所述初始训练标签,以根据修正之后的初始训练标签确定所述目标样本图像的训练标签;所述特征信息用于指示对应肢体特征点和所述目标样本图像之间的位置关系,和/或,用于指示对应肢体特征点是否为待检测部位的特征点;
通过所述目标样本图像和所述训练标签,对所述待训练网络模型进行有监督训练,得到姿态检测模型。
7.根据权利要求6所述的方法,其特征在于,所述待检测部位为多个,所述通过所述目标样本图像和所述训练标签,对所述待训练网络模型进行训练,包括:
通过所述待训练网络模型对所述目标样本图像进行图像处理,得到所述目标样本图像中每个待检测部位的位置信息;
基于多个所述待检测部位中第一部位和第二部位的位置信息,以及所述训练标签,确定用于约束所述第一部位和第二部位之间的位置差异的目标损失函数的函数值,其中,所述第一部位和所述第二部位为具有关联关系的待检测部位;
根据所述目标损失函数的函数值调整所述待训练网络模型的模型参数,以通过调整之后的模型参数对所述待训练网络模型进行训练。
8.根据权利要求6或7所述的方法,其特征在于,所述基于所述监督类型修正所述初始训练标签,包括:
在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签。
9.根据权利要求8所述的方法,其特征在于,所述在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签,包括:
在根据所述监督类型确定出所述肢体特征点位于所述目标样本图像之外,且不是所述待检测部位的特征点情况下,将所述肢体特征点的初始训练标签修正为第一初始训练标签;所述第一初始训练标签用于表征该肢体特征点为位于所述目标样本图像之外的非监督类特征点。
10.根据权利要求8所述的方法,其特征在于,所述在基于所述监督类型确定所述肢体特征点的初始训练标签满足修正条件的情况下,基于所述监督类型修正所述初始训练标签,包括:
在根据所述初始训练标签确定出所述肢体特征点位于所述目标样本图像之外,且属于待检测部位的特征点的情况下,将所述肢体特征点的初始训练标签修正为第二初始训练标签;所述第二初始训练标签用于表征该肢体特征点为位于所述目标样本图像之外的监督类特征点。
11.根据权利要求6所述的方法,其特征在于,所述样本图像为对原始图像执行目标操作之后确定的图像,其中,所述原始图像为在场景复杂度不满足预设复杂要求的场景下采集到的包含目标对象的至少一个肢体部位的图像,所述目标操作包括以下至少之一:背景替换操作、调整图像亮度的操作、调整图像曝光度的操作、添加反光图层的操作。
12.根据权利要求6所述的方法,其特征在于,所述对所述样本图像进行处理,得到包含所述待检测部位的目标样本图像,包括:
对所述样本图像中的目标检测部位进行图像遮挡处理,得到所述目标样本图像;其中,所述目标检测部位为所述肢体部位中除所述待检测部位之外的其他部位;
和/或
对所述样本图像中位于目标区域内的图像进行图像裁剪处理,裁剪得到包含所述待检测部位的目标样本图像;所述目标区域为所述样本图像中包含所述待检测部位的图像区域。
13.一种姿态检测装置,其特征在于,包括:
第一获取单元,用于获取包含目标主播的肢体部位的视频图像帧;
姿态检测单元,用于通过姿态检测模型对所述视频图像帧中的所述目标主播进行姿态检测,得到姿态检测结果,其中,所述姿态检测模型为利用目标样本图像及对应的训练标签进行监督训练得到,所述目标样本图像包含待检测的肢体部位样本,所述训练标签基于所述肢体部位样本的肢体特征点的监督类型及所述肢体特征点对应的初始训练标签进行调整得到;所述监督类型基于所述肢体特征点的特征信息确定,所述特征信息用于指示对应肢体特征点和所述目标样本图像之间的位置关系,和/或,用于指示对应肢体特征点是否为待检测部位的特征点;
生成单元,用于在得到所述目标主播的姿态检测结果之后,根据所述姿态检测结果生成所述目标主播所对应的虚拟主播的姿态触发信号;
控制单元,用于根据所述姿态触发信号控制所述虚拟主播执行相应的触发动作。
14.一种神经网络模型的训练装置,其特征在于,包括:
第二获取单元,用于获取初始训练样本;所述初始训练样本中包含样本图像和该样本图像中所包含的目标对象的肢体部位的肢体特征点的初始训练标签;
修正处理单元,用于基于待训练网络模型的检测任务在所述肢体部位中确定待检测部位;并对所述样本图像进行处理,得到包含所述待检测部位的目标样本图像;
确定单元,用于基于所述肢体特征点的特征信息确定所述肢体特征点的监督类型,并基于所述监督类型修正所述初始训练标签,以根据修正之后的初始训练标签确定所述目标样本图像的训练标签;所述特征信息用于指示对应肢体特征点和所述目标样本图像之间的位置关系,和/或,用于指示对应肢体特征点是否为待检测部位的特征点;
训练单元,用于通过所述目标样本图像和所述训练标签,对所述待训练网络模型进行有监督训练。
15.一种计算机设备,其特征在于,包括:处理器、存储器和总线,所述存储器存储有所述处理器可执行的机器可读指令,当计算机设备运行时,所述处理器与所述存储器之间通过总线通信,所述机器可读指令被所述处理器执行时执行如权利要求1至5中任一所述的姿态检测方法的步骤,或者,执行如权利要求6至12中任一所述的神经网络模型的训练方法的步骤。
16.一种计算机可读存储介质,其特征在于,该计算机可读存储介质上存储有计算机程序,该计算机程序被处理器运行时执行如权利要求1至5中任一所述的姿态检测方法的步骤,或者,执行如权利要求6至12中任一所述的神经网络模型的训练方法的步骤。
Priority Applications (2)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110993216.2A CN113435431B (zh) | 2021-08-27 | 2021-08-27 | 姿态检测方法、神经网络模型的训练方法、装置及设备 |
PCT/CN2022/075197 WO2023024442A1 (zh) | 2021-08-27 | 2022-01-30 | 检测方法、训练方法、装置、设备、存储介质和程序产品 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110993216.2A CN113435431B (zh) | 2021-08-27 | 2021-08-27 | 姿态检测方法、神经网络模型的训练方法、装置及设备 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN113435431A CN113435431A (zh) | 2021-09-24 |
CN113435431B true CN113435431B (zh) | 2021-12-07 |
Family
ID=77798218
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110993216.2A Active CN113435431B (zh) | 2021-08-27 | 2021-08-27 | 姿态检测方法、神经网络模型的训练方法、装置及设备 |
Country Status (2)
Country | Link |
---|---|
CN (1) | CN113435431B (zh) |
WO (1) | WO2023024442A1 (zh) |
Families Citing this family (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113435431B (zh) * | 2021-08-27 | 2021-12-07 | 北京市商汤科技开发有限公司 | 姿态检测方法、神经网络模型的训练方法、装置及设备 |
CN115908665A (zh) * | 2021-09-30 | 2023-04-04 | 北京字节跳动网络技术有限公司 | 一种视频处理方法、装置、设备、介质及产品 |
CN115223002B (zh) * | 2022-05-09 | 2024-01-09 | 广州汽车集团股份有限公司 | 模型训练方法、开门动作检测方法、装置以及计算机设备 |
CN115997385A (zh) * | 2022-10-12 | 2023-04-21 | 广州酷狗计算机科技有限公司 | 基于增强现实的界面显示方法、装置、设备、介质和产品 |
CN116243623B (zh) * | 2023-05-10 | 2023-08-04 | 深圳墨影科技有限公司 | 应用于数字化机器人产业链的机器人场景仿真方法 |
CN116873690B (zh) * | 2023-09-06 | 2023-11-17 | 江苏省特种设备安全监督检验研究院 | 电梯安全监控数据处理*** |
Family Cites Families (10)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US10346723B2 (en) * | 2016-11-01 | 2019-07-09 | Snap Inc. | Neural network for object detection in images |
CN108399406B (zh) * | 2018-01-15 | 2022-02-01 | 中山大学 | 基于深度学习的弱监督显著性物体检测的方法及*** |
CN108712661B (zh) * | 2018-05-28 | 2022-02-25 | 广州虎牙信息科技有限公司 | 一种直播视频处理方法、装置、设备及存储介质 |
CN109120985B (zh) * | 2018-10-11 | 2021-07-23 | 广州虎牙信息科技有限公司 | 直播中的形象展示方法、装置和存储介质 |
CN109936774A (zh) * | 2019-03-29 | 2019-06-25 | 广州虎牙信息科技有限公司 | 虚拟形象控制方法、装置及电子设备 |
CN111641844B (zh) * | 2019-03-29 | 2022-08-19 | 广州虎牙信息科技有限公司 | 直播互动方法、装置、直播***及电子设备 |
CN109905724A (zh) * | 2019-04-19 | 2019-06-18 | 广州虎牙信息科技有限公司 | 直播视频处理方法、装置、电子设备及可读存储介质 |
CN110139115B (zh) * | 2019-04-30 | 2020-06-09 | 广州虎牙信息科技有限公司 | 基于关键点的虚拟形象姿态控制方法、装置及电子设备 |
CN111353555A (zh) * | 2020-05-25 | 2020-06-30 | 腾讯科技(深圳)有限公司 | 一种标注检测方法、装置及计算机可读存储介质 |
CN113435431B (zh) * | 2021-08-27 | 2021-12-07 | 北京市商汤科技开发有限公司 | 姿态检测方法、神经网络模型的训练方法、装置及设备 |
-
2021
- 2021-08-27 CN CN202110993216.2A patent/CN113435431B/zh active Active
-
2022
- 2022-01-30 WO PCT/CN2022/075197 patent/WO2023024442A1/zh unknown
Also Published As
Publication number | Publication date |
---|---|
WO2023024442A1 (zh) | 2023-03-02 |
CN113435431A (zh) | 2021-09-24 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN113435431B (zh) | 姿态检测方法、神经网络模型的训练方法、装置及设备 | |
CN109345556B (zh) | 用于混合现实的神经网络前景分离 | |
US20200273180A1 (en) | Deformable object tracking | |
CN113422977B (zh) | 直播方法、装置、计算机设备以及存储介质 | |
US20220383649A1 (en) | System and method for facilitating graphic-recognition training of a recognition model | |
CN110310288A (zh) | 用于混合现实环境中的对象分割的方法和*** | |
CN110473232A (zh) | 图像识别方法、装置、存储介质及电子设备 | |
CN112819944A (zh) | 三维人体模型重建方法、装置、电子设备及存储介质 | |
CN111325846B (zh) | 表情基确定方法、虚拟形象驱动方法、装置及介质 | |
CN112884881A (zh) | 三维人脸模型重建方法、装置、电子设备及存储介质 | |
CN113449696B (zh) | 一种姿态估计方法、装置、计算机设备以及存储介质 | |
CN108200334A (zh) | 图像拍摄方法、装置、存储介质及电子设备 | |
CN114363689B (zh) | 直播控制方法、装置、存储介质及电子设备 | |
CN113487709A (zh) | 一种特效展示方法、装置、计算机设备以及存储介质 | |
CN112668492A (zh) | 一种自监督学习与骨骼信息的行为识别方法 | |
CN111479087A (zh) | 3d监控场景控制方法、装置、计算机设备及存储介质 | |
CN109858402B (zh) | 一种图像检测方法、装置、终端以及存储介质 | |
CN115393532B (zh) | 脸部绑定方法、装置、设备及存储介质 | |
CN116863043A (zh) | 面部动态捕捉驱动方法、装置、电子设备及可读存储介质 | |
US11720233B2 (en) | Method for associating production elements with a production approach | |
CN113257055A (zh) | 一种智能舞蹈步伐学习装置及方法 | |
Qian et al. | Multi-Scale tiny region gesture recognition towards 3D object manipulation in industrial design | |
CN111639611A (zh) | 一种历史文物的展示控制方法及装置 | |
WO2023029289A1 (zh) | 模型评估方法、装置、存储介质及电子设备 | |
CN115937430B (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 | ||
REG | Reference to a national code |
Ref country code: HK Ref legal event code: DE Ref document number: 40056153 Country of ref document: HK |