[发明专利]用于离焦量确定的模型训练方法及装置

说明书

本公开涉及成像技术领域，具体涉及一种用于离焦量确定的模型训练方法及装置、计算机可读存储介质及电子设备，方法包括：获取初始训练数据，初始训练数据包括多组子训练数据，每组子训练数据中包括物距相同、焦距相同且亮度不同的多对参考左右视差图以及参考左右视差图对应的参考离焦量；利用亮度大于等于预设亮度的参考左右视差图对应的参考离焦量，更新同一组子训练数据中亮度低于预设亮度的参考左右视差图对应的参考离焦量，以确定目标训练数据；基于目标训练数据训练离焦量确定网络。本公开实施例的技术方案克服相关技术中在亮度较暗时离焦量计算精度差的问题。

技术领域

本公开涉及成像技术领域，具体而言，涉及一种用于离焦量确定的模型训练方法及装置、计算机可读存储介质及电子设备。

背景技术

理想的手机自动对焦应具有对焦速度快、稳定和准确的特点，这就要求对焦算法在当前位置能够快速准确计算出离焦量。

但是，现有技术的离焦量计算方式在亮度较暗计算精度较差，以使得手机对焦不准确。

需要说明的是，在上述背景技术部分公开的信息仅用于加强对本公开的背景的理解，因此可以包括不构成对本领域普通技术人员已知的现有技术的信息。

发明内容

本公开的目的在于提供一种用于离焦量确定的模型训练方法、用于离焦量确定的模型训练装置、计算机可读介质和电子设备，进而至少在一定程度上克服相关技术中在亮度较暗时离焦量计算精度差的问题。

根据本公开的第一方面，提供一种用于离焦量确定的模型训练方法，包括：获取初始训练数据，所述初始训练数据包括多组子训练数据，每组子训练数据中包括物距相同、焦距相同且亮度不同的多对参考左右视差图以及所述参考左右视差图对应的参考离焦量；利用亮度大于等于预设亮度的参考左右视差图对应的参考离焦量，更新同一组子训练数据中亮度低于所述预设亮度的参考左右视差图对应的参考离焦量，以确定目标训练数据；基于所述目标训练数据训练离焦量确定网络。

根据本公开的第二方面，提供一种离焦量确定方法，包括获取待测左右视差图，并利用上述训练后的所述离焦量确定网络确定所述待测左右视差图对应的目标离焦量。

根据本公开的第三方面，提供一种用于离焦量确定的模型训练装置，包括：获取模块，用于获取初始训练数据，所述初始训练数据包括多组子训练数据，每组子训练数据中包括物距相同、焦距相同且亮度不同的多对参考左右视差图以及所述参考左右视差图对应的参考离焦量；更新模块，用于利用目标亮度下的参考左右视差图对应的参考离焦量，更新同一组子训练数据中亮度低于所述预设亮度的参考左右视差图对应的参考离焦量，以确定目标训练数据，所述预设亮度小于等于所述目标亮度；训练模块，用于基于所述目标训练数据训练离焦量确定网络。

根据本公开的第四方面，提供一种计算机可读介质，其上存储有计算机程序，计算机程序被处理器执行时实现上述的方法。

根据本公开的第五方面，提供一种电子设备，其特征在于，包括：一个或多个处理器；以及存储器，用于存储一个或多个程序，当一个或多个程序被一个或多个处理器执行时，使得一个或多个处理器实现上述的方法。