基于ResNet卷积神经网络的绿茶种类识别模型构建

作者：张怡; 赵珠蒙; 王校常; 冯海强; 林杰^*

来源：茶叶科学, 2021, 41(02): 261-271.

DOI：10.13305/j.cnki.jts.2021.02.008

• 卷积神经网络
• 深度学习
• 绿茶分类

绿茶是我国种类最多、产量最大的茶类,外形是其分类的重要依据。图像分类是计算机视觉的核心技术之一,但其在茶叶领域的应用较少,茶类识别仍依赖感官审评方法。采集8种常见绿茶(丽水香茶、信阳毛尖、六安瓜片、太平猴魁、安吉白茶、碧螺春、竹叶青和龙井)共1 713张图片,基于ResNet卷积神经网络,从识别模型的预测能力、收敛速度、模型大小和识别均衡性等角度探索了不同网络深度和不同优化算法的建模效果,最终选择ResNet-18结构、SGD优化算法,建立了区分8种绿茶的深度学习模型,其对复杂背景茶叶图片的识别准确率达到了90.99%,单张图片识别时间仅为0.098 s,模型大小为43.7 MB。本研究为构建茶叶视觉识别模型并应用于移动端提供了基础,为茶叶种类识别提供了一种快捷而高效的新方法。

• 单位
  浙江大学; 浙江农林大学