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 条
  • [1] YOLOv5-R: lightweight real-time detection based on improved YOLOv5
    Ren, Jian
    Wang, Zhijie
    Zhang, Yifan
    Liao, Lei
    JOURNAL OF ELECTRONIC IMAGING, 2022, 31 (03)
  • [2] Research on Lightweight Algorithm Model for Precise Recognition and Detection of Outdoor Strawberries Based on Improved YOLOv5n
    Cao, Xiaoman
    Zhong, Peng
    Huang, Yihao
    Huang, Mingtao
    Huang, Zhengyan
    Zou, Tianlong
    Xing, He
    AGRICULTURE-BASEL, 2025, 15 (01):
  • [3] Lightweight Detection Method for Real-Time Monitoring Tomato Growth Based on Improved YOLOv5s
    Tian, Suyu
    Fang, Chao
    Zheng, Xiaogang
    Liu, Jue
    IEEE ACCESS, 2024, 12 : 29891 - 29899
  • [4] A Lightweight Remote Sensing Aircraft Object Detection Network Based on Improved YOLOv5n
    Wang, Jiale
    Bai, Zhe
    Zhang, Ximing
    Qiu, Yuehong
    REMOTE SENSING, 2024, 16 (05)
  • [5] High-Precision Garbage Detection Algorithm of Lightweight YOLOv5n
    Tu, Chengfeng
    Yi, Anlin
    Yao, Tao
    He, Wenwei
    Computer Engineering and Applications, 2023, 59 (10) : 187 - 195
  • [6] YOLOFM: an improved fire and smoke object detection algorithm based on YOLOv5n
    Geng, Xin
    Su, Yixuan
    Cao, Xianghong
    Li, Huaizhou
    Liu, Linggong
    SCIENTIFIC REPORTS, 2024, 14 (01)
  • [7] A Lightweight Real-Time Recognition Algorithm for Tomato Leaf Disease Based on Improved YOLOv8
    Liu, Wenbo
    Bai, Chenhao
    Tang, Wei
    Xia, Yu
    Kang, Jie
    AGRONOMY-BASEL, 2024, 14 (09):
  • [8] YOLOFM: an improved fire and smoke object detection algorithm based on YOLOv5n
    Xin Geng
    Yixuan Su
    Xianghong Cao
    Huaizhou Li
    Linggong Liu
    Scientific Reports, 14
  • [9] SCD-YOLO: a lightweight vehicle target detection method based on improved YOLOv5n
    Li, Heng
    Zhuang, Xufei
    Bao, Shi
    Chen, Junnan
    Yang, Chenxi
    JOURNAL OF ELECTRONIC IMAGING, 2024, 33 (02)
  • [10] A Real-Time Fish Target Detection Algorithm Based on Improved YOLOv5
    Li, Wanghua
    Zhang, Zhenkai
    Jin, Biao
    Yu, Wangyang
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2023, 11 (03)