基于优化VGG16网络的TBM围岩识别方法及装置

本发明提供一种基于优化VGG16网络的TBM围岩识别方法及装置，涉及隧道施工技术领域。该方法包括：获取用于训练的岩渣图像数据集；构建用于当前任务的VGG_ROCK网络；构建输入通道，使得数据得以进入网络训练；从训练完成的VGG16网络中，迁移加载全部卷积池化模块的参数，冻结卷积池化模块；单独训练VGG_ROCK网络分类层；冻结VGG_ROCK网络的分类层；重新激活VGG_ROCK网络最深层的卷积池化模块并训练；将训练完成的VGG_ROCK网络模型保存，以用于围岩识别任务。对比现有深度网络方法，本方法兼具训练速度快且识别准确率好的优势，这更贴合TBM硬岩掘进的实际条件，能切实解决围岩识别问题。
上海工业自动化仪表研究院有限公司 | 上海交通大学
孙凯文 | 尹德斌 | 陶建峰 | 谷朝臣 | 刘成良