[发明专利]一种基于深度学习的航拍公路车辆计数方法及装置

摘要

本发明提供一种基于深度学习的航拍公路车辆计数方法及装置，包括获取多种成像条件下的航拍公路车辆样本图像，并对所述航拍公路车辆样本图像进行预处理；构建深度神经网络模型，利用预处理后的航拍公路车辆样本图像训练构建的深度神经网络模型；使用所述深度神经网络模型对待检测航拍公路车辆图像进行检测，输出检测的公路与车辆对象识别概率、对象位置、对象区域分割；根据识别的公路与车辆结果，对位于公路区域中的车辆计数，输出计数结果。通过共用深度图像特征提取网络，有效利用了计算资源，节省了模型训练与公路、车辆检测运行时间，准确率、检出率都大大提高。本发明应用于交通数据采集、交通流监测和图像数据处理、图像分析应用领域。

主权项

1.一种基于深度学习的航拍公路车辆计数方法，其特征在于，具体包括以下步骤：S1、获取多种成像条件下的航拍公路车辆样本图像组成样本库，并对所述样本库中的航拍公路车辆样本图像进行预处理；S2、构建深度神经网络模型，利用预处理后的航拍公路车辆样本图像训练构建的深度神经网络模型；S3、使用所述深度神经网络模型对待检测航拍公路车辆图像进行检测，输出检测的公路与车辆对象识别概率、对象位置、对象区域分割；S4、根据识别的公路与车辆结果，对位于公路区域中的车辆计数，输出计数结果;步骤S2中，构建的深度神经网络模型其网络结构包括基础主干网络与头部任务网络；基础主干网络包括：全局图像特征提取子网络，由若干卷积层、池化层组成，用于提取图像由浅至深的深度特征图；候选目标位置提名子网络，由若干卷积层、输出层组成，通过在深度特征图上滑动预设大小与步长的窗口，对不同窗口下的特征图进行判别，输出目标的候选位置；目标尺度适配子网络，由多个池化层组成，将提名的不同尺度大小的窗口对应的特征图进一步提取为固定维度的特征图，以满足头部任务网络全连接层对输入数据需要固定维度大小的要求；头部任务网络包括：车辆识别子网络，由多个全连接层组成，对提名的候选目标窗口特征图是否为车辆目标进行识别，输出该窗口在原图中的位置，以及识别概率，即车辆在图像中的位置与车辆的识别概率；车辆分割子网络，由多个卷积层组成，对提名的候选目标窗口特征图进行像素级分割，输出每个像素为车辆的概率，形成车辆分割掩膜热力图，即车辆的分割掩膜；公路识别子网络，由多个全连接层组成，对提名的候选目标窗口特征图是否为公路目标进行识别，输出该窗口在原图中的位置，以及识别概率，即公路在图像中的位置与公路的识别概率；公路分割子网络，由多个卷积层组成，对提名的候选目标窗口特征图进行像素级分割，输出每个像素为公路的概率，形成公路分割掩膜热力图，即公路的分割掩膜。