[发明专利]一种目标检测方法、装置、终端设备和存储介质

权利要求书

1.一种目标检测方法，其特征在于，包括：

将待测图像输入已训练的特征提取网络，得到所述待测图像的特征图像；

对所述特征图像执行角点池化操作，得到所述特征图像对应的左上角点热图和右下角点热图，所述左上角点热图记录所述特征图像中每个位置点属于各个预设类别的物体检测框的左上角关键点的概率值，所述右下角点热图记录所述特征图像中每个位置点属于各个预设类别的物体检测框的右下角关键点的概率值；

对所述特征图像执行内部点池化操作，得到所述特征图像对应的内部点热图，所述内部点热图记录所述特征图像中每个位置点属于各个预设类别的物体检测框的内部关键点的概率值，所述内部点池化操作为采用最大池化层提取所述特征图像的指定内部点在水平方向上的特征值最大的特征向量以及在垂直方向上的特征值最大的特征向量，然后将提取到的特征向量相加的操作；

从所述左上角点热图、所述右下角点热图和所述内部点热图中分别提取指定数量且对应的所述概率值最大的位置点；

根据提取到的各个所述位置点构建物体检测框，其中，每个物体检测框根据从所述左上角点热图中提取到的一个位置点、从所述右上角点热图中提取到的一个位置点以及从所述内部点热图中提取到的一个位置点构建；

对构建得到的物体检测框进行识别，得到所述待测图像的目标检测结果。

2.如权利要求1所述的方法，其特征在于，对所述特征图像执行内部点池化操作，得到所述特征图像对应的内部点热图，包括：

采用最大池化层对所述特征图像的指定内部点到所述特征图像的上边界之间的所有特征向量进行处理，得到其中特征值最大的第一特征向量；

采用最大池化层对所述特征图像的指定内部点到所述特征图像的下边界之间的所有特征向量进行处理，得到其中特征值最大的第二特征向量；

采用最大池化层对所述特征图像的指定内部点到所述特征图像的左边界之间的所有特征向量进行处理，得到其中特征值最大的第三特征向量；

采用最大池化层对所述特征图像的指定内部点到所述特征图像的右边界之间的所有特征向量进行处理，得到其中特征值最大的第四特征向量；

将所述第一特征向量、所述第二特征向量、所述第三特征向量和所述第四特征向量相加，得到目标特征向量；

根据所述目标特征向量确定所述内部点热图。

3.如权利要求1所述的方法，其特征在于，从所述左上角点热图、所述右下角点热图和所述内部点热图中分别提取指定数量且对应的所述概率值最大的位置点，包括：

针对所述各个预设类别中的每个类别，分别从所述左上角点热图中提取属于该类别的物体检测框的左上角关键点的概率值最大的前K个位置点，从所述右下角点热图中提取属于该类别的物体检测框的右下角关键点的概率值最大的前K个位置点，以及从所述内部点热图中提取属于该类别的物体检测框的内部关键点的概率值最大的前K个位置点，K为正整数；

所述根据提取到的各个所述位置点构建物体检测框，包括：

针对所述各个预设类别中的每个类别，按照遍历的方式分别从所述左上角点热图中提取出的与该类别对应的K个位置点中选取一个位置点、从所述右上角点热图中提取出的与该类别对应的K个位置点中选取一个位置点以及从所述内部点热图中提取出的与该类别对应的K个位置点中选取一个位置点，得到该类别对应的所有位置点组合，并根据所述所有位置点组合中的有效位置点组合构建得到与该类别对应的所有物体检测框，其中，在所述有效位置点组合包含的3个位置点中，从所述左上角点热图中提取出的位置点处于从所述右下角点热图中提取出的位置点的左上方，从所述内部点热图中提取出的位置点处于从所述右下角点热图中提取出的位置点的左上方且处于从所述左上角点热图中提取出的位置点的右下方。

4.如权利要求1所述的方法，其特征在于，在根据提取到的各个所述位置点构建物体检测框之后，还包括：

分别确定各个所述物体检测框的中心区域，所述中心区域为对应的物体检测框中包含中心位置点的预设大小的区域；

去除中心区域内不含从所述内部热点图中提取到的位置点的所述物体检测框。