[发明专利]用于目标检测的预训练图片合成方法和装置

说明书

本发明提供一种用于目标检测的预训练图片合成方法和装置，该用于目标检测的预训练图片合成方法包括：获取带目标位置标注的训练图片，在所述训练图片的目标位置生成至少一个候选框；按照预设算法滤除所述目标位置上重叠的所述候选框，以及聚合相互接近的所述候选框；根据预设单元尺寸对预设算法处理后的所述候选框进行尺寸规范化处理，获得规范化候选框；截取所述训练图片上所述规范化候选框的图片块，合成预训练图片。本发明的用于目标检测的预训练图片合成方法，通过利用带目标位置标注的训练图片选取正样本的候选框，进行候选框的滤除、聚合以及规范化后，可以合成规范的预训练图片，从而提高预训练模型的精确度以及预训练模型的预训练效率。

技术领域

本发明涉及图像识别技术领域，具体而言，涉及一种用于目标检测的预训练图片合成方法、装置、计算机设备和可读存储介质。

背景技术

现有的图像识别模型在建立时会通过提前的预训练过程来提高模型建立的效率与稳定性。但是现有的预训练方法需要前置的图像分类过程，该图像分类过程比较费时，导致预训练过程效率较低，从而导致整体的图像识别模型在建立时效率较低。

发明内容

鉴于上述问题，本发明提供了一种用于目标检测的预训练图片合成方法、装置、计算机设备和可读存储介质，以提高预训练模型的精确度以及预训练模型的预训练效率。

为了实现上述目的，本发明采用如下的技术方案：

一种用于目标检测的预训练图片合成方法，包括：

获取带目标位置标注的训练图片，在所述训练图片的目标位置生成至少一个候选框；

按照预设算法滤除所述目标位置上重叠的所述候选框，以及聚合相互接近的所述候选框；

根据预设单元尺寸对预设算法处理后的所述候选框进行尺寸规范化处理，获得规范化候选框；

截取所述训练图片上所述规范化候选框的图片块，合成预训练图片。

优选地，所述的用于目标检测的预训练图片合成方法中，所述按照预设算法滤除所述目标位置上重叠的所述候选框，以及聚合相互接近的所述候选框包括：

计算所述候选框与目标候选框重叠面积的比例值，滤除所述比例值大于预设比例值的所述候选框；

以剩余的所述候选框的中心点作为聚类点，采用层次聚类算法聚合相互接近的多个所述候选框。

优选地，所述的用于目标检测的预训练图片合成方法中，所述比例值的计算算式包括：

式中，A()表示求面积，T_bbox表示目标候选框，R_bbox表示与所述目标候选框重叠的候选框，IoF为重叠的比例值。

优选地，所述的用于目标检测的预训练图片合成方法中，所述根据预设单元尺寸对预设算法处理后的所述候选框进行尺寸规范化处理，获得规范化候选框包括：

确定所述候选框的尺寸小于所述预设单元尺寸时，扩大所述候选框至所述预设单元尺寸；

确定所述候选框的尺寸大于所述预设单元尺寸时，扩大所述候选框至所述预设单元尺寸的整数倍。

优选地，所述的用于目标检测的预训练图片合成方法中，所述规范化候选框的算式包括：

式中，h₂和w₂为规范化候选框的长度和宽度，h₁和w₁为规范化前候选框的长度和宽度，[]为取整运算，G为预设单元尺寸对应的像素值，H和W为所述训练图片的长度和宽度。

优选地，所述的用于目标检测的预训练图片合成方法中，还包括：