图像目标深度的检测方法、装置、电子设备和存储介质制造方法及图纸

本公开提出了一种图像目标深度的检测方法、装置、设备及存储介质，涉及自动驾驶技术领域。该方法包括：获取待检测图像的多个尺寸的第一特征图；从所述多个尺寸的第一特征图中，抽取至少两个第一特征图；基于通道注意力机制，对每个抽取的第一特征图中各个通道的特征值进行加权，以生成所述每个抽取的第一特征图对应的第二特征图；将各个所述第二特征图进行特征融合，以生成第三特征图；对所述第三特征图进行深度检测，以确定所述待检测图像的目标深度。由此，能够使得融合后的第三特征图细节特征的表达能力更强，提高了深度检测的准确度，也使得对图像目标深度的检测更加准确、可靠。靠。靠。

【技术实现步骤摘要】
图像目标深度的检测方法、装置、电子设备和存储介质


[0001]本公开涉及自动驾驶
，尤其涉及一种图像目标深度的检测方法、装置、电子设备和存储介质。

技术介绍

[0002]在自动驾驶技术中，基于单目视觉的目标检测技术近年随着深度学习技术的发展逐步进入到各类目标检测的场景中，但依然存在着目标物深度估计不准确，从而导致目标感知位置不准的缺点。

技术实现思路

[0003]本公开旨在至少在一定程度上解决相关技术中的技术问题之一。
[0004]根据本公开第一方面，提出了一种图像目标深度的检测方法，包括：
[0005]获取待检测图像的多个尺寸的第一特征图；
[0006]从所述多个尺寸的第一特征图中，抽取至少两个第一特征图；
[0007]基于通道注意力机制，对每个抽取的第一特征图中各个通道的特征值进行加权，以生成所述每个抽取的第一特征图对应的第二特征图；
[0008]将各个所述第二特征图进行特征融合，以生成第三特征图；
[0009]对所述第三特征图进行深度检测，以确定所述待检测图像的目标深度本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种图像目标深度的检测方法，其特征在于，包括:获取待检测图像的多个尺寸的第一特征图；从所述多个尺寸的第一特征图中，抽取至少两个第一特征图；基于通道注意力机制，对每个抽取的第一特征图中各个通道的特征值进行加权，以生成所述每个抽取的第一特征图对应的第二特征图；将各个所述第二特征图进行特征融合，以生成第三特征图；对所述第三特征图进行深度检测，以确定所述待检测图像的目标深度。2.根据权利要求1所述的方法，其特征在于，所述获取待检测图像的多个尺寸的第一特征图，包括：将待检测图像输入至预先建立的特征金字塔神经网络，以获取所述特征金字塔神经网络的多个网络层输出的第一特征图。3.根据权利要求1所述的方法，其特征在于，所述基于通道注意力机制，对每个抽取的第一特征图中各个通道的特征值进行加权，以生成所述每个抽取的第一特征图对应的第二特征图，包括：获取每个抽取的第一特征图中每个特征通道对应的特征向量；根据每个所述特征向量，计算所述每个特征通道对应的权重系数；将每个所述权重系数与每个所述特征通道对应的特征值相乘，以生成所述每个抽取的第一特征图对应的第二特征图。4.根据权利要求1所述的方法，其特征在于，所述对所述第三特征图进行深度检测，以确定所述待检测图像的目标深度，包括：将所述第三特征图输入至训练完成的深度预测模型，以获取所述待检测图像的目标深度。5.根据权利要求4所述的方法，其特征在于，在所述将所述第三特征图输入至训练完成的深度预测模型之前，还包括：获取多个训练图像及每个所述训练图像中每个像素点对应的标注深度值；预测每个所述像素点落入每个预设的深度范围的概率；将每个所述像素点落入每个所述深度范围的概率与所述深度区间对应的深度值的乘积之和，确定为每个所述像素点对应的预测深度；根据各个所述像素点对应的预测深度和所述标注深度值之间的差异，对初始神经网络进行修正，以生成深度预测模型。6.一种图像目标深度的检测装置，其特征在于，包括:获取模块，用于获取待检测图像的多个尺寸的第一特征图；抽取模块，用于从所述多个尺寸的第一特征图中，抽取至少两个第一特征图；第一生成模块，用于基于通道注...

【专利技术属性】
技术研发人员：武鹏，
申请(专利权)人：小米汽车科技有限公司，
类型：发明
国别省市：