基于迁移学习的树种识别

编号 zgly0001672757

文献类型 期刊论文

文献题名 基于迁移学习的树种识别

作者 高旋  赵亚凤  熊强  陈喆 

作者单位 东北林业大学机电工程学院 

母体文献 森林工程 

年卷期 2019年05期

年份 2019 

分类号 S718.49  TP391.41  TP18 

关键词 深度学习  迁移学习  卷积神经网络  混淆矩阵 

文摘内容 使用树干和树叶图像实现树种自动识别,目前深度学习可以有效的解决该类问题,但它需要大量样本做训练才能达到较高的识别精度。当面对有限图像数量时,提出基于迁移学习的方法,把经过预训练的卷积神经网络模型进行迁移,即共享卷积层和池化层的权重参数,对新的网络模型超参数进行调整,并建立一个包含10种共计2 000张树干图像和8种共计1 725张树叶图像的数据库,把图片分为训练集和测试集,分别利用迁移学习、普通深度学习和SVM分类方法进行训练和测试,并将这3种方法作对比。最后,通过构建树干和树叶图像的混淆矩阵对迁移学习进行具体分析与说明。实验结果表明,通过迁移学习得到的树干和树叶最高识别精度分别达到92. 51%和98. 20%,比普通深度学习提高了51. 38%和51. 69%,比SVM分类方法提高了43. 94%和45. 08%。迁移学习比普通深度学习和传统SVM分类方法更适合用于小样本数据集的分类识别,并且显著优于普通深度学习和SVM分类方法。