[发明专利]一种车辆位置检测装置、方法及存储介质

说明书

本发明公开了一种车辆位置检测装置，包括图像输入单元、特征提取单元和分类单元；特征提取单元提取车辆图像的图像特征，特征提取单元包括深度聚合网络，深度聚合网络的主干网络包括多个第一特征提取模块，第一特征提取模块包括分辨率调整模块，分辨率调整模块基于双线性插值算法和双三次插值算法调整图像的分辨率；分类单元根据图像特征得到与车辆位置对应的分类结果；将不同分辨率、不同层级的车辆图像进行空间信息和语义信息的融合，根据任务需求自适应调整输入图像的分辨率，能在网络的上采样、下采样训练过程中调整图像分辨率。

技术领域

本发明涉及智能检测领域，特别是一种车辆位置检测装置、方法及存储介质。

背景技术

车辆检测及追踪技术在智能交通系统和无人驾驶领域里发挥着重要的作用。车辆检测及追踪技术常利用神经网络进行识别，目前神经网络中的特征提取网络对输入图像要求统一尺寸输入，将尺寸统一化有缩放、裁剪等方法，但这可能会丢失一些车辆图像的局部信息，这些方法是应用在特征提取网络前的，无法在特征提取网络训练过程中适用，这容易导致图像缩放的尺寸并非最佳尺寸。而分辨率问题会影响网络的特征提取能力，进而降低分类结果准确性。

发明内容

本发明的目的在于至少解决现有技术中存在的技术问题之一，提供一种车辆位置检测装置、方法及存储介质。

本发明解决其问题所采用的技术方案是：

本发明的第一方面，一种车辆位置检测装置，包括：

图像输入单元，所述图像输入单元用于输入车辆图像；

特征提取单元，所述特征提取单元用于提取所述车辆图像的图像特征，所述特征提取单元包括深度聚合网络，所述深度聚合网络的主干网络包括依次连接的多个第一特征提取模块，多个所述第一特征提取模块的输出图像的分辨率相同，所述第一特征提取模块包括分辨率调整模块，所述分辨率调整模块基于双线性插值算法和双三次插值算法调整图像的分辨率；

分类单元，所述分类单元用于根据所述图像特征得到与车辆位置对应的分类结果。

根据本发明的第一方面，所述分辨率调整模块包括第一分辨率调整模块，所述第一特征提取模块包括依次连接的多个降维模块，所述降维模块包括第一降维模块，所述第一降维模块包括最大池化层、所述第一分辨率调整模块和第一卷积层，所述最大池化层和所述第一分辨率调整模块并行连接至所述第一卷积层，且所述最大池化层的输出和所述第一分辨率调整模块的输出尺寸相同。

根据本发明的第一方面，所述分辨率调整模块包括第二分辨率调整模块，所述降维模块还包括第二降维模块，所述第二降维模块包括下采样层和批标准化层，所述下采样层、所述第二分辨率调整模块和所述批标准化层依次连接。

根据本发明的第一方面，所述降维模块还包括第三降维模块和第四降维模块，所述第一降维模块的输出作为所述第二降维模块的输入，所述第二降维模块的输出作为所述第三降维模块的输入，所述第二降维模块的输出和所述第三降维模块的输出作为所述第四降维模块的输入。

根据本发明的第一方面，所述分辨率调整模块包括第一调整分支、第二调整分支和第三调整分支；所述第一调整分支包括基于双三次插值算法的第一调整模块，所述第二调整分支包括基于双线性插值算法的第二调整模块，所述第三调整分支包括基于双线性插值算法的第三调整模块；所述第一调整分支、所述第二调整分支和所述第三调整分支的输入均相同，所述第一调整分支、所述第二调整分支和所述第三调整分支的输出聚合。

根据本发明的第一方面，所述第二调整分支还包括第一卷积块、残差块和第二卷积块；所述第一卷积块、所述第二调整模块、所述残差块和所述第二卷积块依次连接。