[发明专利]对焦方法和装置、电子设备、计算机可读存储介质

权利要求书

1.一种对焦方法，其特征在于，包括：

获取第一预览图像；

生成与所述第一预览图像对应的中心权重图；

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

当检测到所述目标主体区域中包含光斑时，对所述目标主体区域进行处理，得到所述目标主体区域中不含光斑的区域；

将所述不含光斑的区域作为所述第一预览图像的对焦区域，进行自动对焦；

或者，

获取第一预览图像；

获取与所述第一预览图像对应的深度图；

生成与所述第一预览图像对应的中心权重图；

将所述第一预览图像、所述深度图和所述中心权重图输入到主体检测模型中，得到所述第一预览图像中的目标主体区域；所述主体检测模型是预先根据同一场景的可见光图、中心权重图、深度图及对应的已标注的主体进行训练得到的模型；

当检测到所述目标主体区域中包含光斑时，对所述目标主体区域进行处理，得到所述目标主体区域中不含光斑的区域；

将所述不含光斑的区域作为所述第一预览图像的对焦区域，进行自动对焦。

2.根据权利要求1所述的方法，其特征在于，检测目标主体区域中包含光斑的方式，包括：

将所述第一预览图像输入至光斑检测模型进行处理，得到所述目标主体区域中包含光斑的区域，其中，所述光斑检测模型是根据已标注的包含光斑的图像训练得到的。

3.根据权利要求1所述的方法，其特征在于，检测目标主体区域内包含光斑的方式，包括：

获取所述目标主体区域中像素点的亮度值，确定亮度值大于亮度阈值的目标像素点；

当检测到在所述目标像素点的预设范围内存在预设数量的目标像素点时，确定所述目标主体区域内包含光斑。

4.根据权利要求1所述的方法，其特征在于，所述主体检测模型包括依次相连的输入层、中间层和输出层；

所述将所述第一预览图像和所述中心权重图输入到主体检测模型中，包括：

将所述第一预览图像作用于主体检测模型的输入层；

将所述中心权重图作用于所述主体检测模型的输出层。

5.根据权利要求1所述的方法，其特征在于，所述方法还包括：

获取所述不含光斑的区域对应的主体区域置信度图；

对所述主体区域置信度图进行自适应置信度阈值过滤处理，得到二值化掩膜图；

对所述二值化掩膜图进行形态学处理和引导滤波处理，得到不含光斑的区域对应的主体掩膜图；

所述将所述不含光斑的区域作为所述第一预览图像的对焦区域，进行自动对焦，包括：

将所述不含光斑的区域对应的主体掩膜图作为所述第一预览图像的对焦区域，进行自动对焦。

6.根据权利要求1所述的方法，其特征在于，所述主体检测模型包括依次相连的输入层、中间层和输出层；

所述将所述第一预览图像、所述深度图和所述中心权重图输入到主体检测模型中，得到所述第一预览图像中的目标主体区域，包括：

将所述第一预览图像作用于主体检测模型的输入层；

将所述深度图和所述中心权重图作用于所述主体检测模型的输出层。

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

获取第二预览图像，其中，所述第二预览图像是所述第一预览图像的下一帧图像；

将所述第一预览图像中的对焦区域作为所述第二预览图像中的对焦区域，进行自动对焦。