一种基于SSD算法的管道周边机械设备识别方法
摘要:
本发明公开了一种基于SSD算法的管道周边机械设备识别方法,其涉及管道周边机械识别技术领域,旨在解决目标检测网络SSD并没有实际运用到工程机械检测的问题,其技术方案要点是其步骤如下:将工地上获得的原始工程机械的图片进行清洗分类;通过网络获取不含有工程机械的背景图片,并生成对应的.xml文件,加入数据集;使用多张数据集内部图片拼凑成单张包含工程机械的图片,并生成对应的.xml文件加入数据集;搭建目标检测的骨干网络;搭建基于SSD算法的神经网络模型;使用反向传播算法对神经网络进行训练;使用保存的最优模型对测试集进行测试。达到了使目标的规模更具鲁棒性、使得网络可达到实时检测性能和快速判定待识别图像中工程机械类别的效果。
0/0