【技术实现步骤摘要】
目标检测方法和装置
[0001]本公开涉及人工智能领域,具体涉及计算机视觉、图像识别和深度学习技术,具体可用于智慧城市和智能交通场景下。
技术介绍
[0002]激光雷达在自动驾驶系统中起着关键的作用,利用激光雷达,自动驾驶系统可以准确地对车辆所处环境做实时3D建模,特别是在一些基于相机的视觉感知系统失效的场景下,如:大雾、夜晚、白车等;激光雷达可以极大的提升自动驾驶系统的安全性,同时准确感知某个3D目标(如:车辆、行人等)在激光雷达点云坐标系中的位置、大小及姿态,我们称之为雷达点云3D目标检测任务。
[0003]目前在使用的3D检测网络中,在检测头部分,输入的是单个分辨率的特征图,在单张特征图上进行检测操作,若同时包含大物体和小物体时,如:卡车、行人,特征图的分辨率中最小像素的大小要小于等于需要识别的最小物体类别大小。
技术实现思路
[0004]本公开提供了一种目标检测方法、装置、电子设备、存储介质以及计算机程序产品。
[0005]根据本公开的一方面,提供了一种目标检测方法,该方法包括:响应于获取到包括待检测对象的三维点云数据,获取三维点云数据对应的二维图像;基于二维骨干网络对二维图像进行特征提取,得到二维图像对应的图像特征金字塔,其中,图像特征金字塔包括多张不同分辨率的目标特征图;根据特征图分辨率和检测头之间的对应关系,分别将多张不同分辨率的目标特征图输入到对应的检测头中,基于检测头对目标特征图进行检测处理,生成待检测对象对应的三维检测结果。
[0006]根据本公开的另一方面 ...
【技术保护点】
【技术特征摘要】
1.一种目标检测方法,包括:响应于获取到包括待检测对象的三维点云数据,获取所述三维点云数据对应的二维图像;基于二维骨干网络对所述二维图像进行特征提取,得到所述二维图像对应的图像特征金字塔,其中,所述图像特征金字塔包括多张不同分辨率的目标特征图;根据特征图分辨率和检测头之间的对应关系,分别将所述多张不同分辨率的目标特征图输入到对应的检测头中,基于所述检测头对所述目标特征图进行检测处理,生成所述待检测对象对应的三维检测结果。2.根据权利要求1所述的方法,其中,所述三维点云数据对应多个待检测对象;所述每张目标特征图中包括当前分辨率下需要检测的至少一个待检测对象;以及所述根据特征图分辨率和检测头之间的对应关系,分别将所述多张不同分辨率的目标特征图输入到对应的检测头中,基于所述检测头对所述目标特征图进行检测处理,生成所述待检测对象对应的三维检测结果,包括:根据特征图分辨率和检测头之间的对应关系,确定每张目标特征图对应的检测头;分别将每张目标特征图输入到对应的检测头中,基于所述检测头对所述目标特征图进行检测处理,得到每张目标特征图中至少一个待检测对象对应的预测结果;对每张目标特征图中至少一个待检测对象对应的预测结果进行特征融合,得到所述多个待检测对象对应的三维检测结果。3.根据权利要求2所述的方法,其中,所述根据特征图分辨率和检测头之间的对应关系,确定每张目标特征图对应的检测头,包括:根据特征图分辨率和检测头之间的对应关系,确定每张目标特征图对应的检测头集合,其中,所述检测头集合包括用于检测相同分辨率的特征图、且检测不同对象的至少一个检测头;以及所述分别将每张目标特征图输入到对应的检测头中,基于所述检测头对所述目标特征图进行检测处理,得到每张目标特征图中至少一个待检测对象对应的预测结果,包括:针对每张目标特征图,根据所述目标特征图中的至少一个待检测对象,从所述检测头集合中确定出用于检测所述目标特征图的至少一个待用检测头;分别将所述目标特征图输入至所述至少一个待用检测头中,基于所述至少一个待用检测头对所述目标特征图进行检测处理,得到所述目标特征图中至少一个待检测对象对应的预测结果。4.根据权利要求1所述的方法,其中,所述响应于获取到包括待检测对象的三维点云数据,获取所述三维点云数据对应的二维图像,包括:响应于获取到包括待检测对象的三维点云数据,基于三维骨干网络对所述三维点云数据进行特征提取,得到所述三维点云数据对应的三维特征;根据所述三维特征,获取所述三维点云数据对应的二维图像。5.根据权利要求1所述的方法,其中,所述基于二维骨干网络对所述二维图像进行特征提取,得到所述二维图像对应的图像特征金字塔,包括:基于二维骨干网络对所述二维图像进行特征提取,得到所述二维图像对应的初始图像特征金字塔,其中,所述初始图像特征金字塔包括多张不同分辨率的初始特征图;
通过多尺度特征融合网络对所述多张不同分辨率的初始特征图进行特征融合,得到所述二维图像对应的图像特征金字塔。6.一种目标检测装置,包括:获取模块,被配置成响应于获取到包括待检测对象的三维点云数据,获取所述三维点云数据对应的二维图像;特征提取模块,被配置成基于二维骨干...
【专利技术属性】
技术研发人员:鞠波,叶晓青,谭啸,孙昊,
申请(专利权)人:北京百度网讯科技有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。