具有遮挡处理的轻量级茶芽检测模型


发布时间:

2024-08-15

来源:

作者:

准确检测茶芽对实现采收茶芽所需的自动化和智能化起着至关重要的作用。针对当前目标检测算法对遮挡场景下的茶芽检测精度低、模型过于庞大不适合移动部署的问题,本文提出了一种基于Yolox的改进模型Yolo-GSG用于茶芽检测。在特征融合网络中引入全局上下文机制,融合全局上下文信息,用于复杂背景下的茶叶芽识别。在检测头上增加分离增强注意模块SEAM,收集有利于茶芽分类的信息,抑制无用信息。在后处理阶段引入α-IoU方法,改善遮挡导致的漏检。在CSPLayer中引入ghost模块,大大减少了计算量和模型大小。最后,建立了茶芽数据集。消融实验结果表明,采用合适的技术可以进一步提高模型的性能。该模型优于前人研究中提出的模型(SSD、Faster RCNN、Yolov3、Yolov4、Yolov5、Yolov7)以及后续研究中引入的模型(Yolov8、RT-DETR、Yolov9),其mAP值为92.71%,达到了最佳性能。浮点运算和参数分别减少到23.437G和8.782 M。综上所述,本文提供了理论研究和技术支持,为现实场景下的智能采摘茶芽奠定了基础。

 

图1  相机定位和拍摄角度。

 

图2  茶芽图像在不同的条件下获得:a背景复杂性,b采摘时间,c光照。

 

图3  茶芽的标签图像。A是原始图像,b是标记图像。

 

图4  花蕾类型:a单芽采摘,b一芽一叶采摘,c一芽两叶采摘。

 

图5  不同标记芽类型之间的准确率和召回率差异。

 

图6 数据增强的结果。

 

图7  YOLO-GSG算法的架构。

 

图8  C3-GC的结构。

 

图9  全局上下文(GC)模块。

 

图10  分离和增强注意模块的结构。

 

图11  Ghost模块的结构。

 

图12  GhostBottleneckV2的结构。

 

图13  DFC的结构。

 

图14  GSPGhostV2的结构。

 

图15  不同模型的检测图像。

 

图16  不同模型的图像类别的激活图。

 

来 源

Gui, J., Wu, J., Wu, D. et al. A lightweight tea buds detection model with occlusion handling. Food Measure (2024).

 

编辑

王春颖

推荐新闻

石时之约|韩志国:透过表型数据,看见植物的喜怒哀乐!

本期石时之约,我们将对话慧诺瑞德(北京)科技有限公司总经理、国际植物表型学会(IPPN)执委会委员/工业分会副主席韩志国,一起从表型数据的科学角度,去读懂农作物的喜怒哀乐和前世今生。

慧科研、慧育种、慧种田——慧聚改变的力量

让我们“慧聚”在一起,为“慧科研、慧育种、慧种田”赋能。

高通量植物表型平台建设注意事项

育种,是在给定的环境条件下,选择各种表型指标(产量、品质、抗性)最优的基因型材料的过程(AI育种,从这里起步)。育种工作中大约70%的工作量来自表型观察测量和筛选,是最耗人力物力的过程。

作物生理表型测量基础原理

生理表型测量的核心在于“早、快”,要在肉眼可见之前就能测量并预判出变化趋势,才是这个技术的核心价值。叶绿素荧光成像,恰好满足了这个要求。