• 中国科学引文数据库(CSCD)来源期刊
  • 中文核心期刊中文科技核心期刊
  • Scopus RCCSE中国核心学术期刊
  • 美国EBSCO数据库 俄罗斯《文摘杂志》
  • 《日本科学技术振兴机构数据库(中国)》
二维码

隧道建设(中英文) ›› 2021, Vol. 41 ›› Issue (S1): 161-.DOI: 10.3973/j.issn.2096-4498.2021.S1.020

• 研究与探索 • 上一篇    下一篇

岩渣分类边缘终端轻量化算法研究

王珩   

  1. (上海工业自动化仪表研究院有限公司, 上海 200233
  • 出版日期:2021-07-30 发布日期:2021-08-27
  • 作者简介:王珩(1985—),湖北武汉人,2011年毕业于上海海事大学,通信与信息工程专业,硕士,工程师,主要从事深度学习、图像处理、工业智能终端研究。E-mail: 185265118@qq.com。
  • 基金资助:
    国家重点研发计划项目(2018YFB1702500

Research on Lightweight Algorithm for Edge Terminal of Rock Slag Classification

WANG Heng   

  1. Shanghai Institute of Process Automation & Instrumentation, Shanghai 200233, China
  • Online:2021-07-30 Published:2021-08-27

摘要: 为达到掘进过程中掘进机可实时预测前方地质状况的目的,以实际工程案例的岩渣图片数据为基础,分析传统神经网络模型的结构特征,自主设计轻量化网络模型,并对岩渣图片进行训练,使用岩渣图像数据调整Mobilenet轻量化网络模型参数。结果表明: 1)传统的神经网络算法参数量多,占用内存量大,无法在现场部署算力有限的岩渣识别边缘终端,而自主设计的轻量化网络可以满足现场岩渣图像分类准确性的要求; 2Mobilenet轻量化网络模型比传统网络模型减少了80%~90%的计算量,但岩渣图像分类准确性达到97%

关键词: TBM, 卷积神经网络, 岩渣分类, 出渣系统, 轻量化模型, AlexNet网络模型, VGG网络模型, Mobilenet 模型

Abstract: In order to predict the front geological conditions in real time during boring, the structural characteristics of traditional neural network model are analyzed based on the slag pictures of practical projects and an autonomous lightweight network model is designed. After training with slag pictures, the parameters of Mobilenet lightweight network model is adjusted with the data. The results show that: (1) With excess parameters and memory, the traditional neural network algorithm cannot deploy the edge terminal of slag recognition onsite, while the selfdesigned lightweight network can meet the accuracy requirements of onsite slag image classification. (2) The classification accuracy of Mobilenet lightweight network model reaches 97%, but its calculation amount is only 80%~90% of the amount of traditional neural network model.

Key words: TBM, convolutional neural network, rock slag classification, slagging system, lightweight model, AlexNet network model, VGG network model, Mobilenet model

中图分类号: