[发明专利]数据扩增方法、装置、计算机设备、存储介质及程序产品

说明书

本申请涉及一种数据扩增方法、装置、计算机设备、存储介质及程序产品。该方法包括：在确定需要对当前样本图像进行第一类别目标扩增处理时，获取第一类别目标的扩增目标图像，确定第一类别目标的扩增目标图像在当前样本图像中的合理放置区域，并在合理放置区域内确定目标放置区域，然后基于第一类别目标的扩增目标图像对应的目标放置区域，对第一类别目标进行扩增处理；该方法可以确定目标检测的所有样本图像中待扩增的样本图像，并对待扩增的样本图像进行特定类别目标扩增处理，以缩小目标检测的所有样本图像中各类别目标的总数量之间的差别，从而提高样本图像中目标的平衡性。

技术领域

本申请涉及通信技术领域，特别是涉及一种数据扩增方法、装置、计算机设备、存储介质及程序产品。

背景技术

随着目标检测技术的发展，目标检测的应用范围已经涵盖了自动驾驶、身份检测、安全、医疗等领域。其中，目标检测算法主要依赖于样本图像，为了提高目标检测算法的泛化能力，通常会采集不同时间段不同场景下的大量样本图像来对目标检测算法进行训练。

然而，相关技术中会存在样本图像不平衡的问题。

发明内容

基于此，有必要针对上述技术问题，提供一种数据扩增方法、装置、计算机设备、存储介质及程序产品。

第一方面，本申请提供了一种数据扩增方法，该方法包括：

若确定需要对当前样本图像进行第一类别目标扩增处理，则获取第一类别目标的扩增目标图像；

确定第一类别目标的扩增目标图像在当前样本图像中的合理放置区域，并在合理放置区域内确定目标放置区域；

基于第一类别目标的扩增目标图像对应的目标放置区域，对第一类别目标进行扩增处理。

在其中一个实施例中，确定第一类别目标的扩增目标图像在当前样本图像中的合理放置区域，包括：

获取第一类别目标的扩增目标图像的尺寸信息；

根据第一类别目标的扩增目标图像的尺寸信息，获取当前样本图像中参考类别目标的分布区域；参考类别目标为与扩增目标图像的尺寸信息误差小于预设误差值的第一类别目标；

根据当前样本图像中参考类别目标的分布区域，确定第一类别目标的扩增目标图像在当前样本图像中的合理放置区域。

在其中一个实施例中，在合理放置区域内确定目标放置区域，包括：

遍历第一类别目标的扩增目标图像在合理放置区域中的至少一个放置位置，依次判断第一类别目标的扩增目标图像在每一个放置位置是否满足预设的放置条件；

将第一个满足放置条件的放置位置，确定为第一类别目标的扩增目标图像在当前样本图像中的目标放置位置。

在其中一个实施例中，遍历第一类别目标的扩增目标图像在合理放置区域中的至少一个放置位置，依次判断第一类别目标的扩增目标图像在每一个放置位置是否满足预设放置条件，包括：

以合理放置区域中的预设位置为当前放置位置，判断第一类别目标的扩增目标图像在当前放置位置处是否满足预设放置条件；

若是，则确定当前放置位置为目标放置位置；

若否，按照预设的移动规则将第一类别目标的扩增目标图像从当前放置位置移动至下一个放置位置，继续判断移动后的扩增目标图像在移动后的当前放置位置处是否满足预设放置条件，直至确定出第一个满足预设放置条件的放置位置，得到目标放置位置。

在其中一个实施例中，判断第一类别目标的扩增目标图像在当前放置位置处是否满足预设放置条件，包括：

获取第一类别目标的扩增目标图像在当前放置位置处的缩放比例；