收藏切换
Study of tea buds recognition and detection based on improved YOLOv7 model
收藏切换
Tangwei WEI, Jincheng ZHANG, Jing WANG, Qingyan ZHOU*
Journal of Intelligent Agricultural Mechanization | 2024, 5(2) : 42 - 50
Less
收藏切换
Journal of Intelligent Agricultural Mechanization | 2024, 5(2): 42-50
Study of tea buds recognition and detection based on improved YOLOv7 model
Full
Tangwei WEI, Jincheng ZHANG, Jing WANG, Qingyan ZHOU*
Affiliations
  • College of Information and Artificial Intelligence, Anhui Agricultural University, Hefei 230036, China
Published: 2024-05-15 doi: 10.12398/j.issn.2096-7217.2024.02.005
Outline
收藏切换

To effectively identify tea buds in complex environments and improve the precision of intelligent harvesting while minimizing damage to tea trees, this study addresses the issues of low detection accuracy and poor robustness exhibited by traditional target detection algorithms in tea gardens, and proposes YOLOv7-tea model for tea bud identification and detection based on an improved YOLOv7, so as to achieve rapid recognition and detection of tea buds. First, tea bud images were collected and annotated, and data augmentation was performed to construct a tea bud dataset. Next, the CBAM attention mechanism module was introduced into three feature extraction layers of the YOLOv7 backbone network to enhance the model's feature extraction capability; the SPD-Conv module was used to replace the SConv module in the neck network's downsampling module to reduce the loss of small object features; and the EIoU loss function was employed to optimize box regression, thereby improving the accuracy of the predicted boxes. Finally, a comparative experiment was conducted between other target detection models and the YOLOv7-tea model using the tea bud image dataset as a sample, and the recognition effect of tea buds shot at different distances and angles was tested. The experimental results show that the YOLOv7-tea network model outperforms the YOLOv7 model in terms of precision (P), recall (R), and mean average precision (mAP) by 2.87, 6.91, and 8.69 percentage points, respectively. Additionally, the model has a faster detection speed and exhibits higher confidence scores in the recognition and detection of tea buds in complex backgrounds. The YOLOv7-tea model constructed in this study demonstrates better recognition performance for small-sized tea leaf buds, reducing instances of missed detection and false alarms. It exhibits good robustness and real-time performance, offering valuable insights for estimating tea yield and implementing intelligent harvesting.

tea buds  /  object detection  /  CBAM attention mechanism  /  automated picking  /  YOLOv7
魏堂伟, 张津诚, 王晶, 周庆燕. 基于改进YOLOv7的茶叶嫩芽识别模型研究. 智能化农业装备学报(中英文), 2024 , 5 (2) : 42 -50 . DOI: 10.12398/j.issn.2096-7217.2024.02.005
Tangwei WEI, Jincheng ZHANG, Jing WANG, Qingyan ZHOU. Study of tea buds recognition and detection based on improved YOLOv7 model[J]. Journal of Intelligent Agricultural Mechanization, 2024 , 5 (2) : 42 -50 . DOI: 10.12398/j.issn.2096-7217.2024.02.005
Year 2024 volume 5 Issue 2
39
10
Cite this Article
BibTeX
Article Info
doi: 10.12398/j.issn.2096-7217.2024.02.005
  • Receive Date:2024-02-01
  • Online Date:2026-11-05
  • Published:2024-05-15
Article Data
Affiliations
History
  • Received:2024-02-01
  • Revised:2024-03-20
Funding
Affiliations
    College of Information and Artificial Intelligence, Anhui Agricultural University, Hefei 230036, China
References
Share
https://castjournals.cast.org.cn/joweb/znhnyzbxb/EN/10.12398/j.issn.2096-7217.2024.02.005
Share to
QR

Scan QR to access full text

Cite this article
BibTeX
Citations
表12种不同金属材料的力学参数
科
Family
属数
Number of
genus
种数
Number of
species
占总种数比例
Percentage of
total species (%)
属
Genus
种数
Number of
species
占总种数比例
Percentage of total
species (%)
鹅膏菌科Amanitaceae 2 11 5.26 鹅膏菌属 Amanita 10 4.78
小菇科 Mycenaceae 2 12 5.74 丝盖伞属 Inocybe 5 2.39
多孔菌科 Polyporaceae 8 14 6.70 蜡蘑属 Laccaria 5 2.39
红菇科 Russulaceae 3 23 11.00 小皮伞属 Marasmius 6 2.87
小菇属 Mycena 11 5.26
光柄菇属 Pluteus 5 2.39
红菇属 Russula 17 8.13
栓菌属 Trametes 5 2.39
关闭全屏
  • BibTeX
  • EndNote
  • RefWorks
  • TxT