车辆长度计算方法、装置、系统、计算机设备 及存储介质
本发明提供一种车辆长度计算方法,包括:
构建图片数据集;
对数据集中的图片进行标注,构建出标注数据集;
建立YOLO模型,根据标注数据集的格式更改模型参数,利用模型对训练集及其标注数据集进行训练获得初始权重文件;
构建车长信息表;
调用初始权重文件,利用模型对测试集及其标注数据集进行测试,根据结果的准确率调整参数,保存测试权重文件;
利用模型对待测照片进行识别,将识别结果与车长信息表进行匹配,获取车辆的长度信息。
本发明还提供了车辆长度计算装置、系统、计算机设备及存储介质。
本发明实现了对车辆长度的计算,应用于智能交通领域,可推动智能驾驶的发展,也可以用于车辆的归类,为交通、物流分析提供了基础数据。
1 .一种车辆长度计算方法,其特征在于,所述方法包括:
S201:构建一图片数据集,所述图片数据集分为训练集与测试集;
S202:调用一图片标注工具对所述图片数据集中的图片进行标注,得到对应的标注结果,将该标注结果与该图片分别存放到不同的文件夹中,从而构建出标注数据集;
S203:建立YOLO模型,根据所述标注数据集的格式更改该YOLO模型的参数,随后利用所述YOLO模型对所述训练集及其标注数据集进行训练,获得适用于当前应用环境的初始权重文件;
S204:依据网络上的车辆信息以及当前应用环境的车辆类型构建车长信息表;
S205:调用所述步骤S203中的初始权重文件,利用所述YOLO模型对所述测试集及其标注数据集进行车辆类型的识别测试,得到测试结果,根据测试结果的准确率调整网络超参数,最终得到最优解,保存对应的测试权重文件;
S206:调用所述步骤S205中的测试权重文件,利用所述YOLO模型对待测照片进行车辆类型的识别,并将识别的结果与所述步骤S204中的车长信息表进行匹配,获取车辆的长度信息。
2 .根据权利要求1所述的一种车辆长度计算方法,其特征在于,所述步骤S201包括:
S2011:获取当前应用环境的现场图片作为数据集主体;
S2012:从网络上获取与所述现场图片相近的图片作为所述数据集的补充;
S2013:将所述图片数据集的所有图片按照预设比例划分为训练集与测试集。
3 .根据权利要求1所述的一种车辆长度计算方法,其特征在于,所述步骤S203包括:
S2031:获取所述标注数据集的格式;
S2032:将所述标注数据集的格式传送给一GPU服务器,所述GPU服务器更改YOLO模型的参数使之与所述标注数据集的格式相适应,并对所述训练集及其标注数据集进行训练;
S2033:获取适用于当前应用环境的权重文件。
4 .一种车辆长度计算装置,其特征在于,所述装置包括:
数据集构建模块,设置为构建一图片数据集,并将其所述图片数据集分为训练集与测试集;
标注模块,设置为调用一图片标注工具对所述图片数据集中的图片进行标注,得到对应的标注结果,将该标注结果与该图片分别存放到不同的文件夹中,从而构建出标注数据集;
训练模块,用于获取所述标注数据集的格式,更改YOLO模型参数使之与所述标注数据集格式相适应,利用所述YOLO模型对所述标注数据集进行训练,训练结束之后,获得适用于当前应用环境的权重文件;
车长信息构建模块,设置为依据网络上的车辆信息以及当前应用环境的车辆类型构建车长信息表;
预处理模块,设置为调用所述步骤S203中的初始权重文件,利用所述YOLO模型对所述测试集及其标注数据集进行车辆类型的识别测试,得到测试结果,根据测试结果的准确率调整网络超参数,最终得到最优解,保存对应的测试权重文件;和匹配模块,设置为调用所述测试权重文件,利用所述YOLO模型对待测照片进行车辆类型的识别,并将识别的结果与所述车长信息表进行匹配,获取车辆的长度信息。
5 .根据权利要求4所述的一种车辆长度计算装置,其特征在于,所述数据集构建模块包括:
数据集主体单元,设置为获取当前应用环境的现场图片作为数据集主体;
数据集补充单元,设置为从网络上获取与所述现场图片相近的图片作为所述数据集的补充;和数据集划分单元,设置为将所述图片数据集的所有图片按照预设比例划分为训练集与测试集。
6 .根据权利要求4所述的一种车辆长度计算装置,其特征在于,所述训练模块包括:
标注格式获取单元,设置为获取所述标注数据集的格式;
训练单元,设置为将所述标注数据集的格式传送给一GPU服务器,所述GPU服务器更改YOLO模型的参数使之与所述标注数据集的格式相适应,并对所述训练集及其标注数据集进行训练;和权重文件获取单元,设置为获取适用于当前应用环境的权重文件。
7 .一种车辆长度计算系统,其特征在于,所述系统包括:
根据权利要求4-6之一所述的车辆长度计算装置;以及与所述车辆长度计算装置相连的图像获取装置,用于获取应用场景的现场图片。
8 .一种计算机设备,其特征在于,包括存储器和处理器,所述存储器中存储有计算机程序,所述计算机程序被所述处理器执行时,使得所述处理器实现根据权利要求1-3之一所述的车辆长度计算方法。
9 .一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被处理器执行时实现根据权利要求1-3之一所述的车辆长度计算方法。