학술논문
基于深度残差网络与迁移学习的地形识别方法 / Terrain Recognition Method Based on Deep Residual Network and Transfer Learning
Document Type
Academic Journal
Author
Source
科学技术与工程 / Science Technology and Engineering. 23(9):3779-3786
Subject
Language
Chinese
ISSN
1671-1815
Abstract
针对现有机器人基于深度网络的地形识别方法准确率低、网络训练时间长且需要大量训练数据的问题,提出一种基于深度残差网络与迁移学习的地形识别方法.首先,基于Resnet网络构建一种深度残差网络;其次,利用现有Imagenet大型数据集对构建的深度残差网络进行预训练,作为预训练网络,保留预训练网络除全连接层的训练权重,实现预训练网络大规模的参数迁移;最后,利用自建地形图像数据集对深度残差网络的全连接层进行训练,实现深度残差网络微调.实验结果表明,通过迁移学习的方法,利用深度残差网络对石子路、水泥路、砖地、沥青、草地、泥地6种自建地形图像进行分类,平均准确率达到了99.3％,同时网络训练时间也显著降低.