(19)中华 人民共和国 国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202111629715.X (22)申请日 2021.12.28 (71)申请人 深圳市优必选科技股份有限公司 地址 518000 广东省深圳市南 山区学苑大 道1001号南山智园C1栋16、 2 2楼 (72)发明人 林灿然　张雯圆　邵池　庞建新　 (74)专利代理 机构 深圳中一联合知识产权代理 有限公司 4 4414 代理人 刘永康 (51)Int.Cl. G06V 10/72(2022.01) G06V 10/774(2022.01) G06V 10/82(2022.01) G06V 40/10(2022.01) G06N 3/04(2006.01)G06N 3/08(2006.01) G06K 9/62(2022.01) (54)发明名称 一种人体姿态估计的数据增强方法、 装置及 终端设备 (57)摘要 本申请适用于数据增强技术领域， 提供了一 种人体姿态估计的数据增强方法、 装置及终端设 备。 本申请实施例中获取目标图像， 对上述目标 图像中的人体进行姿态估计， 得到上述人体的初 始热图； 从上述初始热图中确定出置信度符合预 设阈值范围的目标关键点； 确定上述目标关键点 对应的上述人体的局部部件， 从预设的部件库中 获取上述局部部件对应的预设数量的部件图； 根 据上述预设数量的部件图对上述初始热图进行 数据增强， 得到增强后的人体热图， 从而提高提 取人体关键点的精度。 权利要求书2页 说明书9页 附图3页 CN 114429554 A 2022.05.03 CN 114429554 A 1.一种人体姿态估计的数据增强方法， 其特 征在于， 包括： 获取目标图像， 对所述目标图像中的人体进行姿态估计， 得到所述人体的初始热图； 从所述初始热图中确定出置信度符合预设阈值范围的目标关键点； 确定所述目标关键点对应的所述人体的局部部件， 从预设的部件库中获取所述局部部 件对应的预设数量的部件图； 根据所述预设数量的部件图对所述初始热图进行 数据增强， 得到增强后的人体热图。 2.如权利要求1所述的人体姿态估计的数据增强方法， 其特征在于， 在得到增强后的人 体热图之后， 还 包括： 获取所述人体的标签热图， 根据所述标签热图和所述初始热图确定第一损失值； 根据所述增强后的人体热图和所述初始热图确定第二损失值； 对所述第一损失值和所述第 二损失值进行加权处理， 并根据加权处理后的所述第 一损 失值和所述第二损失值确定总损失值； 根据所述总损失值更新所述人体的姿态估计模型的模型参数。 3.如权利要求1所述的人体姿态估计的数据增强方法， 其特征在于， 所述确定目标关键 点对应的所述人体的局部 部件， 包括： 确定与所述目标关键点连接的相邻关键点； 根据所述目标关键点和所述相邻关键点确定所述局部 部件。 4.如权利要求3所述的人体姿态估计的数据增强方法， 其特征在于， 所述根据 所述目标 关键点和所述相邻关键点确定所述局部 部件， 包括： 当存在至少两个相邻关键点 时， 从所述至少两个相邻关键点中选取置信度最小的相邻 关键点； 将所述目标关键点和所述置信度最小的相邻关键点之间的部件确定为所述局部 部件。 5.如权利要求1所述的人体姿态估计的数据增强方法， 其特征在于， 所述根据 所述预设 数量的部件图对所述初始热图进行 数据增强， 得到增强后的人体热图， 包括： 将所述预设数量的部件图随机粘贴在所述目标图像中的人体上， 得到预设数量的人体 样本图； 通过预设的姿态估计算法对所述预设数量的人体样本图进行处理， 得到增强后的人体 热图。 6.如权利要求1所述的人体姿态估计的数据增强方法， 其特征在于， 在根据所述预设数 量的部件图对所述初始热图进行 数据增强之前， 还 包括： 根据所述目标关键点计算所述局部 部件的部件长度； 根据所述部件长度对所述预设数量的部件图进行归一 化处理。 7.如权利要求1所述的人体姿态估计的数据增强方法， 其特征在于， 在从预设的部件库 中获取所述局部 部件对应的预设数量的部件图之前， 还 包括： 获取所述人体的全局图， 根据预设的分割 网络对所述全局图进行裁剪， 得到各个局部 部件图； 计算所述各个局部部件图的置信度， 将所述置信度 大于预设置信度阈值的局部部件图 存储于所述部件库中的对应位置 。 8.一种人体姿态估计的数据增强装置， 其特 征在于， 包括：权　利　要　求　书 1/2 页 2 CN 114429554 A 2图像获取模块， 用于获取目标图像， 对所述目标图像 中的人体进行姿态估计， 得到所述 人体的初始热图； 关键点确定模块， 用于从所述初始热图中确定出置信度符合预设阈值范围的目标关键 点； 部件图获取模块， 用于确定所述目标关键点对应的所述人体的局部部件， 从预设的部 件库中获取 所述局部 部件对应的预设数量的部件图； 数据增强模块， 用于根据所述预设数量的部件图对所述初始热图进行数据增强， 得到 增强后的人体热图。 9.一种终端设备， 包括存储器、 处理器以及存储在所述存储器中并可在所述处理器上 运行的计算机程序， 其特征在于， 所述处理器执行所述计算机程序时实现如权利要求1至7 任一项所述的一种人体姿态估计的数据增强方法的步骤。 10.一种计算机可读存储介质， 所述计算机可读存储介质存储有计算机程序， 其特征在 于， 所述计算机程序被处理器执行时实现如权利要求 1至7任一项 所述的一种人体姿态估计 的数据增强方法的步骤。权　利　要　求　书 2/2 页 3 CN 114429554 A 3