基于Enhanced VGG16的油茶品种分类

被引:8
|
作者
孟志超 [1 ]
贺磊盈 [1 ,2 ]
杜小强 [1 ,2 ]
张国凤 [1 ,2 ]
姚小华 [3 ]
吴顺凯 [1 ]
郭豪鉴 [1 ]
机构
[1] 浙江理工大学机械与自动控制学院
[2] 中国林业科学研究院亚热带林业研究所
[3] 浙江省种植装备技术重点实验室
基金
国家重点研发计划;
关键词
深度学习; 油茶叶; 分类; Enhanced VGG16; hard-Swish; ReLU6;
D O I
暂无
中图分类号
S794.4 [油茶];
学科分类号
0829 ; 0907 ;
摘要
随着油茶产业不断壮大,市场上也出现了油茶幼苗品系混乱、以假乱真、以次充好的现象,因此急需开发一种专门的分类识别算法实现不同油茶品种的准确识别。农业领域常用VGG、ResNet网络模型进行分类工作,但存在权重空间过大和准确率不高等问题。该研究对VGG16网络模型进行层间删减以及结构调整,提出了Enhanced VGG16网络模型,在油茶叶数据集上完成模型训练与测试,并与现有经典卷积神经网络(AlexNet、VGG16、Resnet50、InceptionV3、Xception)进行对比。结果表明,Enhanced VGG16网络模型的训练集准确率和测试集准确率分别为98.98%和98.44%,权重空间为90.6 MB。与原始VGG16模型相比,训练集准确率和测试集准确率分别提高3.08和2.05个百分点,权重空间下降165.4 MB,模型性能显著提升。Enhanced VGG16网络模型与经典卷积神经网络相对比,模型综合性能更优。该研究为通过油茶叶进行品种分类识别提供了依据,同时可为其他农作物品种识别提供参考。
引用
收藏
页码:176 / 181
页数:6
相关论文
共 27 条