工程机械厂家自动识别系统研究.docVIP

工程机械厂家自动识别系统研究 摘要：工程机械在施工的过程中，需要加强机械设备的统一管理，因而建立和健全机械设备的统计管理制度至关重要。对此，本文设计研究了一种用于工程机械厂家的图像识别系统，该系统使用matlab构建alexnet迁移学习模型对工程机械厂家数据集进行训练，通过对工程机械目标的外观和标志为特征进行训练,以达到提高施工现场机械设备调度效率的目的。 关键词：深度学习；系统开发；工程机械 工程机械行业的产业升级对国家基础建设的发展至关重要。但是，现在存在的问题就是现在有的工程项目在施工时，当参与建设的施工设备过多时，就会产生设备调度和管理无序的情况。因此，设计一套用于工程机械厂家的图像识别系统，对于实现我国工程机械的施工统一管理具有重要的意义。对此，本文选定alexnet深度学习网络进行训练，通过不断调试深度学习模型的层次、丢失层丢失率、学习率、训练步数等参数，探讨不同深度的特征提取网络对识别效果的影响。 1图像识别网络模型的选用 合适的识别网络对于系统的构建极为重要，由于本文所使用的工程机械的图像数据集规模较小，所以要求网络具有良好的特征学习能力。对此针对实验的实际的情况和识别算法的可行性在经过对现有主流的图像分类方法分析后，发现Alexnet和vgg16网络在训练速度和精度方面都有着不俗的效果。本文主要以Alexnet作为实验的基础框架，并将其与层次较深的vgg16网络进行了简单对比。 2图像样本数据集建立 良好的图像样本数据库对于图像识别网络的分析极为重要，在进行数据库构建时，其中主要有数据集的构建、图像数据集的标签化处理、图像数据集归一化、图像数据集扩充、图像数据集的划分。（1）数据集构建。数据集共含有18家工程机械厂家，共计900张，基本涵盖了目前市面上的常见主流工程机械厂家。为了保证识别的精度，每个图像中仅有一或两辆工程机械，并且图像中的厂家特征没有重叠或损坏。（2）图像数据集的标签化处理。由于深度学习算法需要采用有标签的样本图片进行训练，而本文采集的工程机械图像是无标签的，因此，需要对样本图像进行打标签处理。使用matlab内置的app“imagelabeler”对厂家图像进行人工标定，将标定完的含有标签信息的mat文件保存在特定的文件夹，随后批量将mat文件导出为.jpg标签图片，保存训练样本图片用于厂家识别网络的训练。（3）图像数据集归一化。在深度学习模型中的输入样本来自imagelabeler裁剪出来的图像,因此，存在裁剪的图像大小不一致的问题，而因为AlexNet模型要求图像的输入尺寸为227×227×3，VGG16的要求为224×224×3，所以需要进行归一化处理，将数据集的图像尺寸统一。（4）图像数据集扩充。为了获得更好的训练效果，提高识别精度，应使数据集的规模尽可能地大，这样网络能学习到更多的特征，训练出的模型效果更好，因此，数据增强成为一种非常便捷的数据集扩充途径。本文采取的数据增强途径有旋转、镜像、添加噪声，以此通过较小的人工采集工作量获得大量的数据样本。（5）图像数据集的划分。按照网络的训练步骤，在训练开始前，需要将样本数据集划分为训练集和测试集两部分，由于样本数据集较小，为了保证训练效果将训练集的比例增大，测试集的比例减小，将训练集和测试集按照4:1的比例进行划分，规定每种厂家的样本图片中4/5为训练集，1/5为测试集。其中，训练集作为网络的训练提供数据支持,测试集用于在训练完成后对模型的识别精度进行检测和自我评估。 3模型训练以及数据对比 （1）模型的训练结果。对模型的训练选择使用迁移学习，采用ImageNet数据集上进行过预训练的AlexNet网络模型，将自建的样本数据集输入网络的最后三层，新模型预训练后其收敛速度得到了极大提升。采用迁移学习方法训练得到网络模型的识别精度更高，训练效果更好。通过对alexnet迁移学习后的网络进行主要参数的不断对比调试，得到了11组不同参数下的网络模型，第一组为训练网络的最佳结果，准确率为87.89%。具体如表1所示。（2）使用不同网络模型的对比实验，本节中分别使用alexnet和vgg16模型进行训练，为防止训练参数变化以及不同数据集对训练结果的影响，在对比实验中，两次的训练学习使用了相同的参数和数据集，训练结果如图1所示。图1由实验结果可知，使用相同网络和参数的情况下，使用alexnet模型训练的网络识别精度为87.89%,训练时间为23min10s,使用vgg16模型训练网络的识别精度为86.11%，训练时长为160min53s。alexnet模型的层数较少，训练速度较快，且识别效果较好，vgg16模型的层数较多，训练速度明显减慢，且识别精度未有明显