[发明专利]锚框生成及其标注框适配方法、装置及计算机存储介质

说明书

一种锚框生成及其标注框适配方法、装置及计算机存储介质，所述锚框生成方法包括将包括有网格单元的目标图像转换为包括有目标位置的目标特征子图像，根据锚框精细度参数将各网格单元划分为多个网格子单元，并在各网格子单元中生成满足锚框单位数量参数的锚框，所述标注框适配方法包括将具有标注框的待检测图像转换为待检测特征子图像，并根据标注框的中心位置以及预设步长，从各网格子单元中确定与标注框相适配的一个网格子单元以作为待适配网格子单元，再根据待适配网格子单元中生成的各锚框，分别计算各锚框与标注框之间的适配度，据以确定与标注框适配的一个锚框。借此，本申请可以增加小目标的检出率，并提高了标注框的定位准确度。

技术领域

本申请实施例涉及图像检测技术，尤指一种锚框生成及其标注框适配方法、装置及计算机存储介质。

背景技术

在全结构化-行人检测的任务中，通常检测模型被部署在交通要道、大楼出入口等较高的摄像头位置，因而需要检测的行人目标往往较小，特别是高度在60像素以下的行人较多，导致这些行人目标的人脸与人头部分更加小，经统计，在行人检测测试集中，高度在60像素以下的人体占总人体的63.46％，高度在30像素以下的人脸、人头分别占总人脸、总人头的73.81％以及70.71％。

由此可见，在全结构化-行人检测任务中，如何提升小目标的召回率(即检测出的正确目标数占总目标数的比例)，是提升模型整体性能的最重要因素之一。

除此之外，行人检测模型的推理速度(例如达到实时推理效果)亦是最基本的要求之一。因此，如何在提高小目标的召回率的同时，亦可兼顾模型的推理速度以及占用显存，即为本申请待解决的技术课题。

发明内容

鉴于上述问题，本申请提供一种锚框生成及其标注框适配方法、装置及计算机存储介质，以克服上述问题或者至少部分地解决上述问题。

本申请第一方面提供一种锚框生成方法，其包括：基于第一预设特征转换规则针对包含有多个网格单元的目标图像执行转换，获得包含有多个目标位置的目标特征子图像，其中，所述目标特征子图像中的各所述目标位置各自对应于所述目标图像中的各所述网格单元；根据锚框精细度参数，将各所述目标位置对应的各所述网格单元划分为多个网格子单元；以及根据锚框单位数量参数，在各所述网格子单元中生成满足所述锚框单位数量参数的锚框。

本申请第二方面提供一种计算机存储介质，所述计算机存储介质中存储有用于执行上述第一方面所述的锚框生成方法的各所述步骤的指令。

本申请第三方面提供一种标注框适配方法，其包括基于第二预设特征转换规则针对具有标注框的待检测图像执行转换，获得包含有所述标注框的待检测特征子图像；根据所述待检测特征子图像的所述标注框的中心位置以及预设步长，从各所述网格子单元中确定与所述标注框相适配的一个所述网格子单元以作为待适配网格子单元；以及根据所述待适配网格子单元中生成的各所述锚框，分别计算各所述锚框与所述标注框之间的适配度，获得各所述锚框对应的各适配值，并根据各所述锚框对应的各所述适配值，确定与所述标注框适配的一个所述锚框。

本申请第四方面提供一种计算机存储介质，所述计算机存储介质中存储有用于执行上述第三方面所述的标注框适配方法的各所述步骤的指令。

本申请第五方面提供一种锚框生成装置，其包括第一转换模块，其基于第一预设特征转换规则针对包含有多个网格单元的目标图像执行转换，获得包含有多个目标位置的目标特征子图像，其中，所述目标特征子图像中的各所述目标位置各自对应于所述目标图像中的各所述网格单元；划分模块，其根据锚框精细度参数，将各所述目标位置对应的各所述网格单元划分为多个网格子单元；锚框生成模块，其根据锚框单位数量参数，在各所述网格子单元中生成满足所述锚框单位数量参数的锚框。