A Lightweight Cherry Tomato Maturity Real-Time Detection Algorithm Based on Improved YOLOV5n

被引:19
|
作者
Wang, Congyue [1 ,2 ]
Wang, Chaofeng [1 ,2 ]
Wang, Lele [2 ]
Wang, Jing [1 ,2 ]
Liao, Jiapeng [1 ,2 ]
Li, Yuanhong [1 ,2 ]
Lan, Yubin [1 ,2 ,3 ,4 ]
机构
[1] South China Agr Univ, Coll Elect Engn, Coll Artificial Intelligence, Guangzhou 510642, Peoples R China
[2] South China Agr Univ, Natl Ctr Int Collaborat Res Precis Agr Aviat Pesti, Guangzhou 510642, Peoples R China
[3] South China Agr Univ, Guangdong Lab Lingnan Modern Agr, Guangzhou 510642, Peoples R China
[4] Texas A&M Univ, Dept Biol & Agr Engn, College Stn, TX 77843 USA
来源
AGRONOMY-BASEL | 2023年 / 13卷 / 08期
关键词
precision agriculture; YOLOv5; cherry tomato; maturity detection; CA; WIoU; COLOR;
D O I
10.3390/agronomy13082106
中图分类号
S3 [农学(农艺学)];
学科分类号
0901 ;
摘要
To enhance the efficiency of mechanical automatic picking of cherry tomatoes in a precision agriculture environment, this study proposes an improved target detection algorithm based on YOLOv5n. The improvement steps are as follows: First, the K-means++ clustering algorithm is utilized to update the scale and aspect ratio of the anchor box, adapting it to the shape characteristics of cherry tomatoes. Secondly, the coordinate attention (CA) mechanism is introduced to expand the receptive field range and reduce interference from branches, dead leaves, and other backgrounds in the recognition of cherry tomato maturity. Next, the traditional loss function is replaced by the bounding box regression loss with dynamic focusing mechanism (WIoU) loss function. The outlier degree and dynamic nonmonotonic focusing mechanism are introduced to address the boundary box regression balance problem between high-quality and low-quality data. This research employs a self-built cherry tomato dataset to train the target detection algorithms before and after the improvements. Comparative experiments are conducted with YOLO series algorithms. The experimental results indicate that the improved model has achieved a 1.4% increase in both precision and recall compared to the previous model. It achieves an average accuracy mAP of 95.2%, an average detection time of 5.3 ms, and a weight file size of only 4.4 MB. These results demonstrate that the model fulfills the requirements for real-time detection and lightweight applications. It is highly suitable for deployment in embedded systems and mobile devices. The improved model presented in this paper enables real-time target recognition and maturity detection for cherry tomatoes. It provides rapid and accurate target recognition guidance for achieving mechanical automatic picking of cherry tomatoes.
引用
收藏
页数:24
相关论文
共 50 条
  • [21] Real-time object detection algorithm based on improved YOLOv3
    Zhang, Xiuling
    Dong, Xiaopeng
    Wei, Qijun
    Zhou, Kaixuan
    JOURNAL OF ELECTRONIC IMAGING, 2019, 28 (05)
  • [22] Improved two-stage deep learning algorithm and lightweight YOLOv5n for classifying cottonseed damage
    He, Weilong
    Wu, Fan
    Snyder, Lori Unruh
    Cheng, Jean
    Wilcox, Evelynn
    Xiang, Lirong
    COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2025, 232
  • [23] Improved YOLOv5 Algorithm for Real-Time Detection of Irregular Driving Behavior
    Zou, Peng
    Yang, Kaijun
    Liang, Chen
    Computer Engineering and Applications, 2023, 59 (13) : 186 - 193
  • [24] Lightweight tomato real-time detection method based on improved YOLO and mobile deployment
    Zeng, Taiheng
    Li, Siyi
    Song, Qiming
    Zhong, Fenglin
    Wei, Xuan
    COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2023, 205
  • [25] A Real-Time Detection Algorithm for Sweet Cherry Fruit Maturity Based on YOLOX in the Natural Environment
    Li, Zhiyong
    Jiang, Xueqin
    Shuai, Luyu
    Zhang, Boda
    Yang, Yiyu
    Mu, Jiong
    AGRONOMY-BASEL, 2022, 12 (10):
  • [26] Lightweight Fire Detection Algorithm Based on Improved YOLOv5
    Zhang, Dawei
    Chen, Yutang
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2024, 15 (06) : 809 - 816
  • [27] Lightweight UAV Detection Algorithm Based on Improved YOLOv5
    Peng Y.
    Tu X.
    Yang Q.
    Li R.
    Hunan Daxue Xuebao/Journal of Hunan University Natural Sciences, 2023, 50 (12): : 28 - 38
  • [28] Lightweight chip pad real-time alignment detection method and application based on improved YOLOv5s
    Zou, Yanli
    Zhang, Zongjian
    Zhou, Chiyang
    Tan, Yufei
    JOURNAL OF ELECTRONIC IMAGING, 2025, 34 (01)
  • [29] A Real-Time Detection Algorithm for Kiwifruit Defects Based on YOLOv5
    Yao, Jia
    Qi, Jiaming
    Zhang, Jie
    Shao, Hongmin
    Yang, Jia
    Li, Xin
    ELECTRONICS, 2021, 10 (14)
  • [30] An improved lightweight and real-time YOLOv5 network for detection of surface defects on indocalamus leaves
    Tang, Zhe
    Zhou, Lin
    Qi, Fang
    Chen, Huarong
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2023, 20 (01)