基于Compact-YOLO v4的茶叶嫩芽移动端识别方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

江苏省高等学校自然科学研究重大项目(20KJA510007)、国家自然科学基金面上项目(61873120)和江苏省自然科学基金面上项目(BK20201469)


Mobile Recognition Solution of Tea Buds Based on Compact-YOLO v4 Algorithm
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    茶叶嫩芽精准识别是实现嫩芽智能化采摘的前提与基础,采用视觉和深度学习的嫩芽识别方法逐渐成熟,但该方法过度依赖于高性能硬件,不利于采茶机器人移动端的部署,针对这一问题,本文提出一种基于Compact-YOLO v4算法的茶叶嫩芽移动端识别方法。首先对YOLO v4算法的Backbone网络和Neck网络进行改进,将Backbone网络替换为GhostNet,将Neck网络中传统卷积替换为Ghost卷积,改进后的模型内存占用量仅为原来的1/5。接着运用迁移学习的训练方法提升模型精度,试验表明,Compact-YOLO v4算法模型的精度、召回率、平均精度均值、F1值分别为51.07%、78.67%、72.93%和61.45%。最后将本文算法模型移植到PRO-RK3568-B移动端开发板,通过转换模型、量化处理、改进部署环境3种方式,降低模型推理计算对硬件性能的需求,最终在保证嫩芽识别准确率的前提下,实现了优化模型推理过程、减轻移动端边缘计算压力的目的,为茶叶嫩芽采摘机器人的实际应用提供了技术支撑。

    Abstract:

    The precise recognition of tea buds is the prerequisite and foundation of intelligent bud picking. The method of bud identification using vision and deep learning is gradually established, but it relies excessively on high-performance hardware, which is not conducive to the deployment of mobile tea picking robots. To solve this problem, a mobile recognition solution for tea buds based on the Compact-YOLO v4 algorithm was proposed. Firstly, the Backbone and Neck networks of the YOLO v4 algorithm were improved by replacing the Backbone network with GhostNet and the traditional convolution in the Neck network with Ghost convolution, the size of the improved model was only one-fifth of the original one. Secondly, the training method of transfer learning was applied to improve the model accuracy. The experiments showed that the Compact-YOLO v4 algorithm model had P, R, mAP and F1 score values of 51.07%, 78.67%, 72.93% and 61.45%, respectively. Finally, the algorithm model was transplanted to the PRO-RK3568-B mobile development board to reduce the hardware performance requirements of the model inference calculation by three ways: converting the model, quantization processing, and improving the deployment environment. The aim of optimizing the model inference process and reducing the pressure on edge computing on mobile was eventually achieved, while also ensuring the accuracy of tea bud recognition, providing a theoretical and practical basis for the practical application of tea bud picking robots.

    参考文献
    相似文献
    引证文献
引用本文

黄家才,唐安,陈光明,张铎,高芳征,陈田.基于Compact-YOLO v4的茶叶嫩芽移动端识别方法[J].农业机械学报,2023,54(3):282-290. HUANG Jiacai, TANG An, CHEN Guangming, ZHANG Duo, GAO Fangzheng, CHEN Tian. Mobile Recognition Solution of Tea Buds Based on Compact-YOLO v4 Algorithm[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(3):282-290.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-05-24
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-03-10
  • 出版日期: