欢迎来到知嘟嘟! 联系电话:13095918853 卖家免费入驻,海量在线求购! 卖家免费入驻,海量在线求购!
知嘟嘟
我要发布
联系电话:13095918853
知嘟嘟经纪人
收藏
专利号: 2021104310083
申请人: 北京百度网讯科技有限公司
专利类型:发明专利
专利状态:授权未缴费
专利领域: 计算;推算;计数
更新日期:2024-06-11
缴费截止日期: 暂无
价格&联系人
年费信息
委托购买

摘要:

权利要求书:

1.一种关键点的检测方法,所述方法包括:

获取包含目标的待测图像和训练后的关键点检测模型,将所述待测图像输入所述关键点检测模型,得到包含所述目标的关键点的热力图;

对各个所述热力图进行合并,将合并结果作为多点热力图,包括:将所述关键点检测模型输出的各个所述热力图进行合并,得到包含该各个所述热力图中关键点的多点热力图,其中,所述多点热力图中关键点的数量等于所述关键点检测模型输出的热力图的数量;

将所述多点热力图和所述待测图像,输入训练后的关键点修正模型,得到包含所述目标的关键点的热力图,将该热力图作为修正后热力图;

确定所述修正后热力图指示的关键点坐标。

2.根据权利要求1所述的方法,其中,所述关键点检测模型和所述关键点修正模型两者的参数数量小于预设数值。

3.根据权利要求1或2所述的方法,其中,所述将所述多点热力图和所述待测图像,输入训练后的关键点修正模型,包括:对所述多点热力图和所述待测图像进行拼接,得到拼接结果;

将所述拼接结果输入训练后的关键点修正模型。

4.根据权利要求1或2所述的方法,其中,所述关键点检测模型输出的热力图指示所述目标的一个关键点。

5.一种关键点修正模型的训练方法,所述方法包括:

获取包含目标的待测图像和训练后的关键点检测模型,将所述待测图像输入所述关键点检测模型,得到包含所述目标的关键点的热力图;

对各个所述热力图进行合并,将合并结果作为多点热力图,包括:将所述关键点检测模型输出的各个所述热力图进行合并,得到包含该各个所述热力图中关键点的多点热力图,其中,所述多点热力图中关键点的数量等于所述关键点检测模型输出的热力图的数量;

将所述多点热力图和所述待测图像,输入待训练的关键点修正模型,得到包含所述目标的关键点的热力图,将该热力图作为修正后热力图;

确定所述修正后热力图指示的关键点坐标;

基于所述关键点坐标确定所述关键点修正模型的损失值,通过所述损失值训练所述关键点修正模型,得到训练后的关键点修正模型。

6.根据权利要求5所述的方法,其中,所述关键点检测模型和所述关键点修正模型两者的参数数量小于预设数值。

7.根据权利要求5或6所述的方法,其中,所述将所述多点热力图和所述待测图像,输入训练后的关键点修正模型,包括:对所述多点热力图和所述待测图像进行拼接,得到拼接结果;

将所述拼接结果输入训练后的关键点修正模型。

8.根据权利要求5或6所述的方法,其中,所述关键点检测模型输出的热力图指示所述目标的一个关键点;

所述对各个所述热力图进行合并,将合并结果作为多点热力图,包括:将所述关键点检测模型输出的各个所述热力图进行合并,得到包含该各个所述热力图中关键点的多点热力图,其中,所述多点热力图中关键点的数量等于所述关键点检测模型输出的热力图的数量。

9.一种关键点的检测装置,所述装置包括:

获取单元,被配置成获取包含目标的待测图像和训练后的关键点检测模型,将所述待测图像输入所述关键点检测模型,得到包含所述目标的关键点的热力图;

合并单元,被配置成对各个所述热力图进行合并,将合并结果作为多点热力图,包括:将所述关键点检测模型输出的各个所述热力图进行合并,得到包含该各个所述热力图中关键点的多点热力图,其中,所述多点热力图中关键点的数量等于所述关键点检测模型输出的热力图的数量;

修正单元,被配置成将所述多点热力图和所述待测图像,输入训练后的关键点修正模型,得到包含所述目标的关键点的热力图,将该热力图作为修正后热力图;

确定单元,被配置成确定所述修正后热力图指示的关键点坐标。

10.根据权利要求9所述的装置,其中,所述关键点检测模型和所述关键点修正模型两者的参数数量小于预设数值。

11.根据权利要求9或10所述的装置,其中,所述修正单元,进一步被配置成按照如下方式执行所述将所述多点热力图和所述待测图像,输入训练后的关键点修正模型:对所述多点热力图和所述待测图像进行拼接,得到拼接结果;

将所述拼接结果输入训练后的关键点修正模型。

12.根据权利要求9或10所述的装置,其中,所述关键点检测模型输出的热力图指示所述目标的一个关键点。

13.一种关键点修正模型的训练装置,所述装置包括:

检测单元,被配置成获取包含目标的待测图像和训练后的关键点检测模型,将所述待测图像输入所述关键点检测模型,得到包含所述目标的关键点的热力图;

热力图单元,被配置成对各个所述热力图进行合并,将合并结果作为多点热力图,包括:将所述关键点检测模型输出的各个所述热力图进行合并,得到包含该各个所述热力图中关键点的多点热力图,其中,所述多点热力图中关键点的数量等于所述关键点检测模型输出的热力图的数量;

修正后单元,被配置成将所述多点热力图和所述待测图像,输入待训练的关键点修正模型,得到包含所述目标的关键点的热力图,将该热力图作为修正后热力图;

坐标确定单元,被配置成确定所述修正后热力图指示的关键点坐标;

训练单元,被配置成基于所述关键点坐标确定所述关键点修正模型的损失值,通过所述损失值训练所述关键点修正模型,得到训练后的关键点修正模型。

14.根据权利要求13所述的装置,其中,所述关键点检测模型和所述关键点修正模型两者的参数数量小于预设数值。

15.根据权利要求13或14所述的装置,其中,所述修正后单元,进一步被配置成按照如下方式执行所述将所述多点热力图和所述待测图像,输入训练后的关键点修正模型,包括:对所述多点热力图和所述待测图像进行拼接,得到拼接结果;

将所述拼接结果输入训练后的关键点修正模型。

16.根据权利要求13或14所述的装置,其中,所述关键点检测模型输出的热力图指示所述目标的一个关键点;

所述热力图单元,进一步被配置成按照如下方式执行所述对各个所述热力图进行合并,将合并结果作为多点热力图:将所述关键点检测模型输出的各个所述热力图进行合并,得到包含该各个所述热力图中关键点的多点热力图,其中,所述多点热力图中关键点的数量等于所述关键点检测模型输出的热力图的数量。

17.一种电子设备,包括:

至少一个处理器;以及

与所述至少一个处理器通信连接的存储器;其中,

所述存储器存储有可被所述至少一个处理器执行的指令,所述指令被所述至少一个处理器执行,以使所述至少一个处理器能够执行权利要求1-8中任一项所述的方法。

18.一种存储有计算机指令的非瞬时计算机可读存储介质,其中,所述计算机指令用于使所述计算机执行根据权利要求1-8中任一项所述的方法。