欢迎访问《茶叶科学》,今天是

茶叶科学 ›› 2022, Vol. 42 ›› Issue (3): 387-396.doi: 10.13305/j.cnki.jts.2022.03.002

• 研究报告 • 上一篇    下一篇

结合改进胶囊网络与知识蒸馏的茶青分类方法研究

陈星燃1, 黄海松1,*, 韩正功1, 范青松1, 朱云伟1, 胡鹏飞2,3   

  1. 1.贵州大学现代制造技术教育部重点实验室,贵州 贵阳 550025;
    2.贵州装备制造职业学院,贵州 贵阳 551400;
    3.清镇红枫山韵茶场有限公司,贵州 贵阳 551400
  • 收稿日期:2021-11-30 修回日期:2022-02-16 出版日期:2022-06-15 发布日期:2022-06-17
  • 通讯作者: *hshuang@gzu.edu.cn
  • 作者简介:陈星燃,男,硕士研究生,主要从事深度学习与农业工程的交叉研究。
  • 基金资助:
    贵州省科技计划项目(黔科合支撑[2021]一般445、172、397,黔科合支撑[2022]一般165,黔科合基础[2020]1Y232)、贵州大学引进人才科研基金(贵大人基合字(2019)07号)、贵州省普通高等学校青年科技人才成长项目(黔教合KY 字[2021]096)

Research on the Classification Method of Tea Buds Combining Improved Capsule Network and Knowledge Distillation

CHEN Xingran1, HUANG Haisong1,*, HAN Zhenggong1, FAN Qingsong1, ZHU Yunwei1, HU Pengfei2,3   

  1. 1. Key laboratory of Advanced Manufacturing Technology of the Ministry of Education, Guizhou University, Guiyang 550025, China;
    2. Guizhou Vocational College of Equipment Manufacturing, Guiyang 551400, China;
    3. Qingzhen Hongfeng Mountain Yun Tea Factory Co., Ltd, Guiyang 551400, China
  • Received:2021-11-30 Revised:2022-02-16 Online:2022-06-15 Published:2022-06-17

摘要: 不同等级茶青的准确分类,对名优茶产业发展至关重要,采用传统感官审评方法进行分选会使结果存在一定的主观性。采集茶青图像建立数据集,结合幽灵注意力瓶颈层与胶囊网络提出一种新型网络模型:GA-CapsNet。通过基于线性衰减比例系数的成长知识蒸馏方法对该模型进行训练,在迁移教师模型参数矩阵的同时,使学生模型随着迭代自适应降低依赖程度。试验结果表明,对比其他同类算法,所提出的方法在小规模数据集上分类性能优异,精确率、召回率及F1-score分别为94.97%、95.51%、95.24%。本研究基于机器视觉与深度学习技术构建了一种GA-CapsNet模型,为解决茶青分类问题提供了一种新思路。

关键词: 胶囊网络, 知识蒸馏, 注意力模块, 茶青分类

Abstract: The accurate classification of different grades of tea buds is very important for the development of the famous tea industry. The use of traditional sensory evaluation methods for sorting makes the results subjective. In this research, a data set was established after tea leaf images were collected, and a new network model, GA-CapsNet, was proposed by combining the ghost attention bottleneck and capsule network. The model was trained by the method of growing knowledge distillation based on the linear decay scaling coefficient, while migrating the parameter matrix of teacher model, the student model was adaptively reduced with iteration. The experimental result shows that, compared with other similar algorithms, the proposed method had excellent classification performance on small-scale data sets. The accuracy, recall and F1-score were 94.97%, 95.51% and 95.24%, respectively. Here, a GA-CapsNet model based on machine vision and deep learning technology was established, which provided a new idea for solving the tea leaf classification problem.

Key words: capsule network, knowledge distillation, attention module, tea bud grading

中图分类号: