利索能及
我要发布
收藏
专利号: 2022104206440
申请人: 北京百度网讯科技有限公司
专利类型:发明专利
专利状态:授权未缴费
更新日期:2025-02-06
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种目标检测的方法,包括:

获取待识别图像,并从所述待识别图像中提取第一特征图;

根据所述第一特征图生成检测框;

基于所述检测框生成所述检测框对应的类别特征向量;

根据所述检测框对应的类别特征向量和相关性矩阵生成所述检测框对应的推荐类别。

2.根据权利要求1所述的方法,其中,所述根据所述检测框对应的类别特征向量和相关性矩阵生成所述检测框对应的推荐类别,包括:根据所述检测框对应的类别特征向量和所述相关性矩阵生成权重特征向量;

将所述权重特征向量与所述类别特征向量进行融合,以获取推荐类别特征向量;

根据所述推荐类别特征向量生成所述检测框的推荐类别。

3.根据权利要求2所述的方法,其中,所述相关性矩阵通过以下步骤生成:获取训练数据集中多个类别对应的第一训练图像的第一数量;

将所述多个类别组成类别对,并获取所述类别对对应的第二训练图像的第二数量,其中,所述第二训练图像同时包含所述类别对中的类别对应的检测框;

根据所述第一数量和所述第二数量生成所述相关性矩阵。

4.根据权利要求3所述的方法,其中,所述相关性矩阵中的元素与所述类别对分别对应,其中,所述元素的行对应所述类别对的第一类别,所述元素的列对应所述类别对的第二类别,所述根据所述第一数量和所述第二数量生成所述相关性矩阵包括:获取所述元素对应的所述类别对对应的所述第二数量;

获取所述第一类别对应的所述第一数量;

将所述第二数量与所述第一数量的比值作为所述元素的值。

5.根据权利要求2所述的方法,其中,所述类别特征向量中的行与检测框对应,所述类别特征向量中的列与类别对应,所述根据所述检测框对应的类别特征向量和所述相关性矩阵生成权重特征向量,包括:将所述类别特征向量与所述相关性矩阵相乘,以获取权重矩阵;

获取所述权重矩阵每列元素的平均值,以生成所述权重特征向量。

6.根据权利要求2所述的方法,其中,所述将所述权重特征向量与所述类别特征向量进行融合,以获取推荐类别特征向量,包括:将所述权重特征向量与所述类别特征向量相加,以获取所述推荐类别特征向量;或,将所述权重特征向量与所述类别特征向量中对应元素相乘,以获取所述推荐类别特征向量。

7.根据权利要求6所述的方法,其中,所述根据所述推荐类别特征向量生成所述检测框的推荐类别,包括:获取所述推荐类别特征向量中元素的最大值,并将所述最大值对应的类别作为所述检测框的推荐类别。

8.一种目标检测的装置,包括:

特征提取模块,用于获取待识别图像,并从所述待识别图像中提取第一特征图;

检测框生成模块,用于根据所述第一特征图生成检测框;

类别特征向量获取模块,用于基于所述检测框生成所述检测框对应的类别特征向量;

类别识别模块,用于根据所述检测框对应的类别特征向量和相关性矩阵生成所述检测框对应的推荐类别。

9.根据权利要求8所述的装置,其中,所述类别识别模块,包括:权重特征向量生成子模块,用于根据所述检测框对应的类别特征向量和所述相关性矩阵生成权重特征向量;

融合子模块,用于将所述权重特征向量与所述类别特征向量进行融合,以获取推荐类别特征向量;

类别获取子模块,用于根据所述推荐类别特征向量生成所述检测框的推荐类别。

10.根据权利要求9所述的装置,其中,所述相关性矩阵通过以下模块生成:第一统计子模块,用于获取训练数据集中多个类别对应的第一训练图像的第一数量;

第二统计子模块,用于将所述多个类别组成类别对,并获取所述类别对对应的第二训练图像的第二数量,其中,所述第二训练图像同时包含所述类别对中的类别对应的检测框;

矩阵生成子模块,用于根据所述第一数量和所述第二数量生成所述相关性矩阵。

11.根据权利要求10所述的装置,其中,所述相关性矩阵中的元素与所述类别对分别对应,其中,所述元素的行对应所述类别对的第一类别,所述元素的列对应所述类别对的第二类别,所述矩阵生成子模块包括:第一统计单元,用于获取所述元素对应的所述类别对对应的所述第二数量;

第二统计单元,用于获取所述第一类别对应的所述第一数量;

矩阵生成单元,用于将所述第二数量与所述第一数量的比值作为所述元素的值。

12.根据权利要求9所述的装置,其中,所述类别特征向量中的行与检测框对应,所述类别特征向量中的列与类别对应,所述权重特征向量生成子模块包括:权重矩阵获取单元,用于将所述类别特征向量与所述相关性矩阵相乘,以获取权重矩阵;

平均单元,用于获取所述权重矩阵每列元素的平均值,以生成所述权重特征向量。

13.根据权利要求9所述的装置,其中,所述融合子模块包括:第一融合单元,用于将所述权重特征向量与所述类别特征向量相加,以获取所述推荐类别特征向量;或,第二融合单元,用于将所述权重特征向量与所述类别特征向量中对应元素相乘,以获取所述推荐类别特征向量。

14.根据权利要求13所述的装置,其中,所述类别获取子模块包括:类别获取单元,用于获取所述推荐类别特征向量中元素的最大值,并将所述最大值对应的类别作为所述检测框的推荐类别。

15.一种电子设备,包括:

至少一个处理器;以及

与所述至少一个处理器通信连接的存储器;其中,所述存储器存储有可被所述至少一个处理器执行的指令,所述指令被所述至少一个处理器执行,以使所述至少一个处理器能够执行权利要求1‑7中任一项所述的方法。

16.一种存储有计算机指令的非瞬时计算机可读存储介质,其中,所述计算机指令用于使所述计算机执行根据权利要求1‑7中任一项所述的方法。

17.一种计算机程序产品,包括计算机程序,所述计算机程序在被处理器执行时实现根据权利要求1‑7中任一项所述的方法。