一种目标检测方法、装置、终端设备和存储介质

文档序号:25859524发布日期:2021-07-13 16:14阅读:来源:国知局

技术特征:

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

5.如权利要求4所述的方法,其特征在于,所述分别确定各个所述物体检测框的中心区域包括:

针对每个所述物体检测框,分别计算该物体检测框内的每个位置点的中心度,并将该物体检测框内中心度大于预设阈值的位置点所处的区域确定为该物体检测框的中心区域,其中,每个位置点的中心度根据该位置点距离该物体检测框的各个边界之间的长度确定。

6.如权利要求5所述的方法,其特征在于,在去除中心区域内不含从所述内部热点图中提取到的位置点的所述物体检测框之后,还包括:

若从所述内部热点图中提取到的任意一个位置点处于两个以上的物体检测框的中心区域内,则分别计算所述任意一个位置点在所述两个以上的物体检测框内的中心度,并将所述两个以上的物体检测框中所述任意一个位置点的中心度非最大的物体检测框去除。

7.如权利要求1至6中任一项所述的方法,其特征在于,对构建得到的物体检测框进行识别,得到所述待测图像的目标检测结果,包括:

针对每个所述物体检测框,将该物体检测框输入预设的多元分类器,得到该物体检测框对应的类别向量,并根据该类别向量确定该物体检测框所属的类别,其中,所述类别向量包含该物体检测框分别属于各个所述预设类别的概率值。

8.一种目标检测装置,其特征在于,包括:

特征提取模块,用于将待测图像输入已训练的特征提取网络,得到所述待测图像的特征图像;

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

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

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

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

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

9.一种终端设备,包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,其特征在于,所述处理器执行所述计算机程序时实现如权利要求1至7中任一项所述的目标检测方法。

10.一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至7中任一项所述的目标检测方法。


技术总结
本申请涉及图像处理技术领域,提出一种目标检测方法、装置、终端设备和存储介质。该方法首先提取待测图像的特征图像,然后分别对该特征图像执行角点池化操作和内部点池化操作,以得到左上角点热图、右下角点热图以及内部点热图;接着,从这三个热图中分别提取指定数量的对应概率值最大的位置点,并基于这些位置点构建得到各个物体检测框,其中每个物体检测框都是根据该左上角点热图中提取到的一个位置点、该右下角点热图中提取到的一个位置点以及该内部点热图中提取到的一个位置点构建的;最后,对构建得到的物体检测框进行识别,从而得到该待测图像的目标检测结果。采用本申请提出的目标检测方法,能够提高对图像目标的内部信息的感知能力。

技术研发人员:王宏任;陈世峰
受保护的技术使用者:中国科学院深圳先进技术研究院
技术研发日:2021.04.02
技术公布日:2021.07.13
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1