YOLOv5-ASFF: A Multistage Strawberry Detection Algorithm Based on Improved YOLOv5

被引:18
作者
Li, Yaodi [1 ]
Xue, Jianxin [1 ]
Zhang, Mingyue [1 ]
Yin, Junyi [1 ]
Liu, Yang [1 ]
Qiao, Xindan [1 ]
Zheng, Decong [1 ]
Li, Zezhen [2 ]
机构
[1] Shanxi Agr Univ, Coll Agr Engn, Jinzhong 030801, Peoples R China
[2] Shanxi Agr Univ, Coll Food Sci & Engn, Jinzhong 030801, Peoples R China
来源
AGRONOMY-BASEL | 2023年 / 13卷 / 07期
基金
中国国家自然科学基金;
关键词
strawberry; YOLOv5; intelligent monitoring; automatic picking;
D O I
10.3390/agronomy13071901
中图分类号
S3 [农学(农艺学)];
学科分类号
0901 ;
摘要
The smart farm is currently a hot topic in the agricultural industry. Due to the complex field environment, the intelligent monitoring model applicable to this environment requires high hardware performance, and there are difficulties in realizing real-time detection of ripe strawberries on a small automatic picking robot, etc. This research proposes a real-time multistage strawberry detection algorithm YOLOv5-ASFF based on improved YOLOv5. Through the introduction of the ASFF (adaptive spatial feature fusion) module into YOLOv5, the network can adaptively learn the fused spatial weights of strawberry feature maps at each scale as a way to fully obtain the image feature information of strawberries. To verify the superiority and availability of YOLOv5-ASFF, a strawberry dataset containing a variety of complex scenarios, including leaf shading, overlapping fruit, and dense fruit, was constructed in this experiment. The method achieved 91.86% and 88.03% for mAP and F1, respectively, and 98.77% for AP of mature-stage strawberries, showing strong robustness and generalization ability, better than SSD, YOLOv3, YOLOv4, and YOLOv5s. The YOLOv5-ASFF algorithm can overcome the influence of complex field environments and improve the detection of strawberries under dense distribution and shading conditions, and the method can provide technical support for monitoring yield estimation and harvest planning in intelligent strawberry field management.
引用
收藏
页数:15
相关论文
共 36 条
[11]   YOLOMuskmelon: Quest for Fruit Detection Speed and Accuracy Using Deep Learning [J].
Lawal, Olarewaju M. .
IEEE ACCESS, 2021, 9 :15221-15227
[12]   Development of tomato detection model for robotic platform using deep learning [J].
Lawal, Olarewaju Mubashiru .
MULTIMEDIA TOOLS AND APPLICATIONS, 2021, 80 (17) :26751-26772
[13]   Lightweight target detection for the field flat jujube based on improved YOLOv5 [J].
Li, Shilin ;
Zhang, Shujuan ;
Xue, Jianxin ;
Sun, Haixia .
COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2022, 202
[14]   Feature Pyramid Networks for Object Detection [J].
Lin, Tsung-Yi ;
Dollar, Piotr ;
Girshick, Ross ;
He, Kaiming ;
Hariharan, Bharath ;
Belongie, Serge .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :936-944
[15]   Path Aggregation Network for Instance Segmentation [J].
Liu, Shu ;
Qi, Lu ;
Qin, Haifang ;
Shi, Jianping ;
Jia, Jiaya .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :8759-8768
[16]  
Liu ST, 2019, Arxiv, DOI arXiv:1911.09516
[17]   SSD: Single Shot MultiBox Detector [J].
Liu, Wei ;
Anguelov, Dragomir ;
Erhan, Dumitru ;
Szegedy, Christian ;
Reed, Scott ;
Fu, Cheng-Yang ;
Berg, Alexander C. .
COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 :21-37
[18]  
Loshchilov I., 2017, arXiv
[19]   Vision systems for harvesting robots: Produce detection and localization [J].
Luis-Enrique Montoya-Cavero ;
de Leon Torres, Rocio Diaz ;
Gomez-Espinosa, Alfonso ;
Escobedo Cabello, Jesus Arturo .
COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2022, 192
[20]  
[吕志远 Lyu Zhiyuan], 2023, [农业工程学报, Transactions of the Chinese Society of Agricultural Engineering], V39, P124