[发明专利]一种样本图像的处理方法及相关装置

说明书

本申请公开了一种样本图像的处理方法及相关装置，该方法包括：基于样本图像组中各样本图像的尺寸信息，确定图像尺寸参考信息；分别确定各个样本图像对应的目标数量，目标数量为对应的样本图像中包含的目标的数量；利用确定的各个目标数量和图像尺寸参考信息，分别对各个样本图像进行裁剪，获得各个样本图像对应的局部图像；拼接裁剪得到的各个局部图像，获得扩展图像并输出。本申请所提供的方法可以实现降低样本图像中无用信息产生的量，同时提高目标框的出现频率。

技术领域

本申请涉及人工智能技术领域，特别是涉及一种样本图像的处理方法及相关装置。

背景技术

目标检测属于计算机视觉中应用最为广泛的任务之一，在交通管制、视频监控、自动驾驶等领域中目标检测发挥着极其重要的作用。目前目标检测任务，广泛采用Mosaic数据增广方式获得训练样本，用以提高目标检测模型的检测性能。但是，Mosaic数据增广方式所生成的图像的宽高数值相同，故为了使得所生成的图像与视频图像的尺寸一致，往往需要对所生成的图像进行padding操作方可输出作为训练样本，但是过多的padding操作会产生大量无用信息，降低目标框的出现频率，并增加模型的训练成本，降低模型训练的速度，故急需一种技术方案可以解决上述技术问题。

发明内容

本申请主要解决的技术问题是提供样本图像的处理方法及相关装置，可以降低样本图像中无用信息产生的量，同时提高目标框的出现频率

为解决上述技术问题，本申请采用的一个技术方案是：提供一种样本图像的处理方法，所述方法包括：

基于样本图像组中各样本图像的尺寸信息，确定图像尺寸参考信息；

分别确定所述各个样本图像对应的目标数量，所述目标数量为对应的所述样本图像中包含的目标的数量；

利用确定的各个目标数量和所述图像尺寸参考信息，分别对所述各个样本图像进行裁剪，获得所述各个样本图像对应的局部图像；

拼接裁剪得到的各个所述局部图像，获得扩展图像并输出。

为解决上述技术问题，本申请采用的另一个技术方案是：提供一种样本图像的处理方法，所述方法包括：

针对每张扩展图像中所包括的每个目标框，基于目标框与各个锚框的尺寸比值筛选与所述目标框对应的第一类锚框，其中，所述扩展图像为基于如上任意一项所述的方法获得；

针对每个所述目标框，计算每个所述第一类锚框与所述目标框的相似度比值，并将相似度比值大于或等于第一阈值的所述第一类锚框输出作为所述目标框的正样本。

为解决上述技术问题，本申请采用的另一个技术方案是：提供一种电子设备，所述电子设备包括处理器以及与所述处理器耦接的存储器；其中，

所述存储器用于存储计算机程序；

所述处理器用于运行所述计算机程序以执行如上任意一项所述的方法。

为解决上述技术问题，本申请采用的又一个技术方案是：提供一种计算机可读存储介质，所述计算机可读存储介质存储有能够被处理器运行的计算机程序，所述计算机程序用于实现如上任一项所述的方法。

本申请的有益效果是：区别于现有技术的情况，本申请所提供的技术方案，通过基于样本图像组中各样本图像的尺寸信息，确定图像尺寸参考信息，然后分别确定各个样本图像对应的目标数量，再利用确定的各个目标数量和图像尺寸参考信息，分别对各个样本图像进行裁剪，获得各个样本图像对应的局部图像；最后拼接裁剪得到的各个局部图像，获得扩展图像并输出。在本申请所提供的技术方案中，通过结合样本图像中所包括的目标数量自各个样本图像中截取得到局部图像，然后对应拼接截取所得的局部图像，实现降低样本图像中无用信息产生的量，同时提高目标框的出现频率，起到了良好的技术效果。

附图说明