线上期刊服务咨询,发表咨询:400-808-1701 订阅咨询:400-808-1721

基于迁移学习的卷积神经网络植物叶片图像识别方法

郑一力; 张露 农业机械学报 2018年第S1期

摘要:为了提高植物叶片图像的识别准确率,考虑到植物叶片数据库属于小样本数据库,提出了一种基于迁移学习的卷积神经网络植物叶片图像识别方法。首先对植物叶片图像进行预处理,通过对原图的随机水平、垂直翻转、随机缩放操作,扩充植物叶片图像数据集,对扩充后的叶片图像数据集样本进行去均值操作,并以4∶1的比例划分为训练集和测试集;然后将训练好的模型(Alex Net、Inception V3)在植物叶片图像数据集上进行迁移训练,保留预训练模型所有卷积层的参数,只替换最后一层全连接层,使其能够适应植物叶片图像的识别;最后将本文方法与支持向量机(SVM)方法、深度信念网络(DBN)方法、卷积神经网络(CNN)方法在ICL数据库进行对比实验。实验使用Tensorflow训练网络模型,实验结果由Tensor Board可视化得到的数据绘制而成。结果表明,利用Alex Net、Inception V3预训练模型得到的测试集准确率分别为95. 31%、95. 40%,有效提高了识别准确率。

关键词:叶片识别卷积神经网络迁移学习

单位:北京林业大学工学院

注:因版权方要求,不能公开全文,如需全文,请咨询杂志社

农业机械学报

北大期刊

¥1272.00

关注 25人评论|1人关注