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

摘要:

权利要求书:

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

获取待处理的图像,并提取所述图像中的图像特征;

基于所述图像特征,预测所述图像中的各个部位属性组,其中,所述部位属性组中包括至少两个部位的属性信息,所述至少两个部位属于同一个目标;

根据各个所述部位属性组,生成所述图像的目标检测结果。

2.根据权利要求1所述的方法,其中,所述提取所述图像中的图像特征,包括:提取所述图像在多个维度上的维度特征;

对所述多个维度上的维度特征进行融合处理,得到所述图像特征。

3.根据权利要求1或2所述的方法,其中,所述提取所述图像中的图像特征,包括:采用目标检测模型中的主干网络层+融合网络层,提取所述图像特征,其中,所述主干网络层用于提取所述图像中多个维度上的维度特征,所述融合网络层用于对所述多个维度上的维度特征进行融合处理。

4.根据权利要求3所述的方法,其中,所述主干网络层为残差网络层;

所述融合网络层为特征图金字塔网络层。

5.根据权利要求1所述的方法,其中,所述基于所述图像特征,预测所述图像中的各个部位属性组,包括:

采用所述目标检测模型中的各个分支对所述图像特征进行预测处理,获取各个所述部位属性组,其中,每个分支用于预测一个部位属性组。

6.根据权利要求1所述的方法,其中,所述目标为车辆,所述部位包括以下属性中的至少一种:车灯、车窗、车轮。

7.一种目标检测模型的训练方法,包括:获取初始的目标检测模型,其中,所述目标检测模型包括:用于提取图像中图像特征的主干网络层+融合网络层、以及各个分支;其中,每个分支用于结合所述图像特征预测所述图像中的一个部位属性组;所述部位属性组中包括至少两个部位的属性信息,所述至少两个部位属于同一个目标;

获取训练数据,其中,所述训练数据包括:样本图像、所述样本图像中各个样本目标上各个部位的属性信息;

以所述样本图像为输入,结合所述目标检测模型输出的各个预测目标上各个部位的预测属性信息以及所述样本图像中各个样本目标上各个部位的属性信息构建损失函数,对所述初始的目标检测模型进行训练。

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

第一获取模块,用于获取待处理的图像,并提取所述图像中的图像特征;

预测模块,用于基于所述图像特征,预测所述图像中的各个部位属性组,其中,所述部位属性组中包括至少两个部位的属性信息,所述至少两个部位属于同一个目标;

生成模块,用于根据各个所述部位属性组,生成所述图像的目标检测结果。

9.根据权利要求8所述的装置,其中,所述第一获取模块,具体用于:提取所述图像在多个维度上的维度特征;

对所述多个维度上的维度特征进行融合处理,得到所述图像特征。

10.根据权利要求8或9所述的装置,其中,所述第一获取模块,还用于:采用目标检测模型中的主干网络层+融合网络层,提取所述图像特征,其中,所述主干网络层用于提取所述图像中多个维度上的维度特征,所述融合网络层用于对所述多个维度上的维度特征进行融合处理。

11.根据权利要求10所述的装置,其中,所述主干网络层为残差网络层;

所述融合网络层为特征图金字塔网络层。

12.根据权利要求8所述的装置,其中,所述预测模块,还用于:采用所述目标检测模型中的各个分支对所述图像特征进行预测处理,获取各个所述部位属性组,其中,每个分支用于预测一个部位属性组。

13.根据权利要求8所述的装置,其中,所述目标为车辆,所述部位包括以下属性中的至少一种:车灯、车窗、车轮。

14.一种目标检测模型的训练装置,包括:第二获取模块,用于获取初始的目标检测模型,其中,所述目标检测模型包括:用于提取图像中图像特征的主干网络层+融合网络层、以及各个分支;其中,每个分支用于结合所述图像特征预测所述图像中的一个部位属性组;所述部位属性组中包括至少两个部位的属性信息,所述至少两个部位属于同一个目标;

所述第二获取模块,还用于获取训练数据,其中,所述训练数据包括:样本图像、所述样本图像中各个样本目标上各个部位的属性信息;

训练模块,用于以所述样本图像为输入,结合所述目标检测模型输出的各个预测目标上各个部位的预测属性信息以及所述样本图像中各个样本目标上各个部位的属性信息构建损失函数,对所述初始的目标检测模型进行训练。

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

至少一个处理器;以及

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

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

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