数字农科院2.0

A lightweight tea bud detection model based on Yolov5

文献类型： 外文期刊

作者： Gui, Zhiyong;Chen, Jianneng;Li, Yang;Chen, Zhiwei;Wu, Chuanyu;Dong, Chunwang

作者机构：

关键词： Tea bud detection;Deep learning;Lightweight;Attention mechanism;Weighted feature fusion

期刊名称： COMPUTERS AND ELECTRONICS IN AGRICULTURE

ISSN： 0168-1699

年卷期： 2023 年 205 卷

页码：

收录情况： SCIE（2023版） ; ; EI（2023版）

摘要： Tea bud detection technology is highly significant in realizing the automation and intelligence of tea bud picking. However, there are still some challenges with tea bud detection technology. For example, the problems of low detection accuracy, heavy computing, and large detection model size make the technology inconducive to the deployment of mobile terminals. Thus, a lightweight tea bud detection model based on the Yolov5 model was proposed in this study. Improvements were made in the following aspects: the Ghost_conv module was introduced to replace the original convolution, considerably reducing the computing and model size; the bottleneck attention module (BAM) was added to the backbone network to suppress invalid information and improve the model detection accuracy; the weighted feature fusion was used in the neck network to efficiently fuse the lowlevel and high-level features, helping the network to extract effective information for recognition and improve detection accuracy; and CIoU was used as the bounding box loss function to accelerate regression prediction and improve the positioning accuracy of the bounding box of the model. A test was conducted on the collected dataset to verify whether the modified model improved the detection performance of tea buds. The results showed that compared with that of the original Yolov5 model, the mean average precision of the modified model increased by 9.66%, and the floating-point operations and params reduced by 52.402 G and 22.71 M, respectively. An ablation study proved that the proposed method improves the detection performance of the Yolov5 model for tea buds. Compared with other detection algorithm models, the superiority of the proposed algorithm in tea bud detection can be seen. The proposed improved Yolov5 can effectively detect tea buds, which provides theoretical research and technical support for intelligent picking of tea buds in actual scenes.

分类号：

  • 相关文献

[1]Small target tea bud detection based on improved YOLOv5 in complex background. Mengjie Wang,Yang Li,Hewei Meng,Zhiwei Chen,Zhiyong Gui,Yaping Li,Chunwang Dong. 2024

[2]Lightweight Fruit-Detection Algorithm for Edge Computing Applications. Wenli Zhang,Yuxin Liu,Kaizhen Chen,Huibin Li,Yulin Duan,Wenbin Wu,Yun Shi,Wei Guo. 2021

[3]A Non-Destructive Method for Identification of Tea Plant Cultivars Based on Deep Learning. Yi Ding,Haitao Huang,Hongchun Cui,Xinchao Wang,Yun Zhao. 2023

[4]Lightweight cotton diseases real-time detection model for resource-constrained devices in natural environments. Pan Pan,Mingyue Shao,Peitong He,Lin Hu,Sijian Zhao,Longyu Huang,Guomin Zhou,Jianhua Zhang. 2024

[5]Estimation of the orientation of potatoes and detection bud eye position using potato orientation detection you only look once with fast and accurate features for the movement strategy of intelligent cutting robots. Jie Huang,Xiangyou Wang,Chengqian Jin,Fernando Auat Cheein,Xinyu Yang. 2025

[6]Automatic counting of rapeseed inflorescences using deep learning method and UAV RGB imagery. Li J.,Li Y.,Qiao J.,Li L.,Wang X.,Yao J.,Liao G.. 2023

[7]Semi-supervised learning and attention mechanism for weed detection in wheat. Teng Liu,Xiaojun Jin,Luyao Zhang,Jie Wang,Yong Chen,Chengsong Hu,Jialin Yu. 2023

[8]TBTA-D2Net: a novel hyperspectral image classification method based on triple-branch ternary-attention mechanism and Dense2Net. Ting Tang,Shengwei Zhang,Shaopeng Liu,Weihong Yan,Xin Pan. 2023

[9]Multi-Scale Mixed Attention Tea Shoot Instance Segmentation Model. Chen, Dongmei,Cao, Peipei,Yan, Lijie,Chen, Huidong,Lin, Jia,Li, Xin,Yuan, Lin,Wu, Kaihua. 2024

[10]An Unstructured Orchard Grape Detection Method Utilizing YOLOv5s. Wang, Wenhao,Shi, Yun,Liu, Wanfu,Che, Zijin. 2024

[11]MBSFC: hyperspectral image classification based on multi-branch and spectral feature conversion. Tang, Ting,Liu, Shaopeng,Fu, Xueliang,Yan, Weihong,Luo, Xiaoling,Pan, Xin. 2024

[12]Improved PM2.5 prediction with spatio-temporal feature extraction and chemical components: The RCG-attention model. Ao Li,Yafei Wang,Qianqian Qi,Yunfeng Li,Haixia Jia,Xin Zhou,Haixin Guo,Shuyang Xie,Junfeng Liu,Yujing Mu. 2024

[13]Lightweight Structure and Attention Fusion for In-Field Crop Pest and Disease Detection. Zijing Luo,Yunsen Liang,Naimin Kong,Lirui Liang,Wenjun Peng,Yujie Yao,Chi Qin,Xiaohan Lu,Mingman Xu,Yining Zhang,Chenyang Lin,Chengyao Jiang,Mengyao Li,Yangxia Zheng,Yameng Jiang,Wei Lu. 2025

[14]Tea Bud Detection and 3D Pose Estimation in the Field with a Depth Camera Based on Improved YOLOv5 and the Optimal Pose-Vertices Search Method. Chen, Zhiwei,Chen, Jianneng,Li, Yang,Gui, Zhiyong,Yu, Taojie. 2023

[15]TBD-Y: Automatic tea bud detection with synergistic object-spatial attention and global-local attention guided feature fusion. Zhongyuan Liu,Li Zhuo,Chunwang Dong,Jiafeng Li,Yang Li. 2025

[16]B-FLACS: blockchain-based flexible lightweight access control scheme for data sharing in cloud. Tao, Qi,Cui, Xiaohui. 2022

[17]Lightweight SM-YOLOv5 Tomato Fruit Detection Algorithm for Plant Factory. Xinfa Wang,Zhenwei Wu,Meng Jia,Tao Xu,Canlin Pan,Xuebin Qi,Mingfu Zhao. 2023

[18]VGNet: A Lightweight Intelligent Learning Method for Corn Diseases Recognition. Xiangpeng Fan,Zhibin Guan. 2023

[19]Weed Recognition at Soybean Seedling Stage Based on YOLOV8nGP + NExG Algorithm. Tao Sun,Longfei Cui,Lixuan Zong,Songchao Zhang,Yuxuan Jiao,Xinyu Xue,Yongkui Jin. 2024

[20]Design and Research on a Reed Field Obstacle Detection and Safety Warning System Based on Improved YOLOv8n. Zhang, Yuanyuan,Mu, Zhongqiu,Tian, Kunpeng,Zhang, Bing,Huang, Jicheng. 2025

作者其他论文 更多>>