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

说明书

本申请涉及一种对焦方法和装置、电子设备、计算机可读存储介质。所述方法包括：获取第一预览图像；将第一预览图像输入至主体检测模型中，得到第一预览图像中的目标主体区域，其中，主体检测模型是预先根据同一场景的可见光图、中心权重图及对应的已标注的主体进行训练得到的模型，或者主体检测模型是预先根据同一场景的可见光图、中心权重图、深度图及对应的已标注的主体进行训练得到的模型；当检测到目标主体区域中包含光斑时，对目标主体区域进行处理，得到目标主体区域中不含光斑的区域；将不含光斑的区域作为第一预览图像的对焦区域，进行自动对焦，能够在对焦时避免光斑的干扰，从而提高对焦的准确性。

技术领域

本申请涉及影像领域，特别是涉及一种对焦方法和装置、电子设备、计算机可读存储介质。

背景技术

在使用摄像机进行拍摄时，一般会结合ROI(Region Of Interest，感兴趣区域)进行对焦。由于ROI范围相对固定，当出现干扰时，例如出现光斑，会影响摄像头对焦的准确性。

发明内容

本申请实施例提供一种对焦方法、装置、电子设备、计算机可读存储介质，可以提高摄像头对焦的准确性。

一种对焦方法，包括：

获取第一预览图像；

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

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

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

一种对焦装置，包括：

获取模块，用于获取第一预览图像；

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

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

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

一种电子设备，包括存储器及处理器，所述存储器中储存有计算机程序，所述计算机程序被所述处理器执行时，使得所述处理器执行以下步骤：

获取第一预览图像；

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

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

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

一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时实现以下步骤：

获取第一预览图像；