[发明专利]一种基于切片筛选的目标检测推理加速方法及系统

权利要求书

1.一种基于切片筛选的目标检测推理加速方法，其特征在于，包括以下步骤：

S1、对鸟瞰图原图按设定的大小以相邻切片有一定重叠的方式进行划分，得到大小相同的切片；

S2、对划分得到的切片，使用由卷积神经网络来预测切片中每个区域包含目标的概率，输出关于切片的概率矩阵，取所有概率中最大值作为每个切片包含目标的概率；

S3、利用双阈值筛选方法对每个切片包含目标的概率进行筛选，得到明显和潜在包含目标的切片；

S4、对筛选出的明显和潜在包含有目标的切片进行目标检测，同时以原图作为辅助预测，输出相对于切片和原图的检测结果；

S5、将相对于切片的检测结果映射到原图上，再对所有检测结果进行后处理去除冗余项，得到融合后的最终检索结果。

2.根据权利要求1所述的基于切片筛选的目标检测推理加速方法，其特征在于，步骤S1中，切片大小设定方法为：基于预训练模型数据集中目标面积与原图面积的比率以及将数据集目标大小的统计结果，保持将数据集和预训练模型数据集目标大小与送入检测器的图像大小比率一致。

3.根据权利要求1所述的基于切片筛选的目标检测推理加速方法，其特征在于，步骤S1中相邻切片的重叠比例为0.25。

4.根据权利要求1所述的基于切片筛选的目标检测推理加速方法，其特征在于，步骤S1中，按照划分的顺序将切出的切片按照行列顺序记为表示该切片来自于第i行第j列，每一块的左上、右上、左下、右下坐标分别为(X_min,Y_min)、(X_max,Y_min)、(X_min,Y_max)、(X_max,Y_max)，其中：

X_min＝i*S_width-i*Overlap ratio

Y_min＝j*S_height-j*Overlap ratio

X_max＝X_min+S_width

Y_max＝Y_min+S_height

其中，Overlap ratio为重叠比例，采用0.25；S_width为切片宽度，S_height为切片高度，I_width为原图宽度，I_height为原图高度。

5.根据权利要求1所述的基于切片筛选的目标检测推理加速方法，其特征在于，步骤S2中切片为覆盖数据集85％以上目标大小的最小方格。

6.根据权利要求1所述的基于切片筛选的目标检测推理加速方法，其特征在于，步骤S3中，设置阈值为0.5，初步筛选出概率大于等于0.5的切片作为明显包含目标的切片S_obvious，然后在明显包含目标的切片周围设置概率阈值为0.3，进一步筛选出潜在包含目标的切片S_potential。