[发明专利]主体对焦方法、装置、电子设备和存储介质

权利要求书

1.一种主体对焦方法，其特征在于，所述主体对焦方法应用于电子设备，所述电子设备包括第一摄像头、第二摄像头和ISP处理器，所述方法包括：

所述第二摄像头获取飞行时间TOF图像；

所述ISP处理器对所述TOF图像进行主体识别，确定目标主体；当检测到TOF图像中包括多个主体时，取所述多个主体中权重最高的主体作为所述目标主体，或者，提醒用户选择其中一个主体作为目标主体；其中，与镜头之间的距离越小，主体的权重越高；与TOF图像对角线的交点的距离越小，主体的权重越高；人物的权重大于动物的权重，动物的权重大于植物的权重；根据用户指令确定各种类型的主体的权重；

所述ISP处理器获取所述目标主体在预览图像中的位置信息，所述第一摄像头的预览镜头根据所述位置信息对所述目标主体进行对焦。

2.根据权利要求1所述的方法，其特征在于，所述对所述TOF图像进行主体识别，确定目标主体，包括：

将所述TOF图像输入预设的主体检测网络，得到至少两个候选主体；

从所述至少两个候选主体中确定所述目标主体。

3.根据权利要求2所述的方法，其特征在于，所述从所述至少两个候选主体中确定所述目标主体，包括：

根据预设的权重规则，确定各所述候选主体的权重；

将权重最大的候选主体确定为所述目标主体。

4.根据权利要求2所述的方法，其特征在于，所述从所述至少两个候选主体中确定所述目标主体，包括：

获取用户选择指令；所述用户选择指令为用户对所述至少两个候选主体的主体标识进行选择触发的指令；

将所述用户选择指令对应的候选主体确定为所述目标主体。

5.根据权利要求2-4任一项所述的方法，其特征在于，所述将所述TOF图像输入预设的主体检测网络，得到至少两个候选主体，包括：

生成与所述TOF图像对应的中心权重图，其中，所述中心权重图所表示的权重值从中心到边缘逐渐减小；

将所述TOF图像和所述中心权重图输入到所述主体检测模型中，得到主体区域置信度图，其中，所述主体检测模型是预先根据同一场景的TOF图像、中心权重图及对应的已标注的主体掩膜图进行训练得到的模型；

根据所述主体区域置信度图确定所述TOF图像中的至少两个候选主体。

6.根据权利要求1-4任一项所述的方法，其特征在于，所述确定所述目标主体在预览图像中的位置信息，预览镜头根据所述位置信息对所述目标主体进行对焦，包括：

获取所述目标主体在所述TOF图像中的位置坐标；

根据TOF镜头的坐标系与所述预览镜头的坐标系之间的预设对应关系表，获得所述目标主体在所述预览图像的位置坐标；

所述预览镜头根据所述目标主体在所述预览图像中的位置坐标，对所述目标主体进行对焦。

7.根据权利要求1-4任一项所述的方法，其特征在于，所述确定所述目标主体在预览图像中的位置信息，预览镜头根据所述位置信息对所述目标主体进行对焦，包括：

获取所述目标主体的深度信息；

根据所述目标主体的深度信息确定所述目标主体在预览图像中的对焦位置信息；

所述预览镜头根据所述目标主体在预览图像中的对焦位置信息，对所述目标主体进行对焦。

8.根据权利要求1-4任一项所述的方法，其特征在于，所述方法还包括：

获取所述预览镜头的RGB图像；

所述对所述TOF图像进行主体识别，确定目标主体，包括：

对所述TOF图像和所述RGB图像进行主体识别，确定所述目标主体。

9.一种计算机设备，包括存储器和处理器，所述存储器存储有计算机程序，其特征在于，所述处理器执行所述计算机程序时实现权利要求1至8中任一项所述方法的步骤。