A Lightweight Detection Method for Blueberry Fruit Maturity Based on an Improved YOLOv5 Algorithm
-
Published:2023-12-24
Issue:1
Volume:14
Page:36
-
ISSN:2077-0472
-
Container-title:Agriculture
-
language:en
-
Short-container-title:Agriculture
Author:
Xiao Feng1ORCID, Wang Haibin1ORCID, Xu Yueqin1, Shi Zhen1
Affiliation:
1. College of Mechanical and Electrical Engineering, Northeast Forestry University, Harbin 150040, China
Abstract
In order to achieve accurate, fast, and robust recognition of blueberry fruit maturity stages for edge devices such as orchard inspection robots, this research proposes a lightweight detection method based on an improved YOLOv5 algorithm. In the improved YOLOv5 algorithm, the ShuffleNet module is used to achieve lightweight deep-convolutional neural networks. The Convolutional Block Attention Module (CBAM) is also used to enhance the feature fusion capability of lightweight deep-convolutional neural networks. The effectiveness of this method is evaluated using the blueberry fruit dataset. The experimental results demonstrate that this method can effectively detect blueberry fruits and recognize their maturity stages in orchard environments. The average recall (R) of the detection is 92.0%. The mean average precision (mAP) of the detection at a threshold of 0.5 is 91.5%. The average speed of the detection is 67.1 frames per second (fps). Compared to other detection algorithms, such as YOLOv5, SSD, and Faster R-CNN, this method has a smaller model size, smaller network parameters, lower memory usage, lower computation usage, and faster detection speed while maintaining high detection performance. It is more suitable for migration and deployment on edge devices. This research can serve as a reference for the development of fruit detection systems for intelligent orchard devices.
Funder
Natural Science Foundation of Heilongjiang Province of China China Postdoctoral Science Foundation
Subject
Plant Science,Agronomy and Crop Science,Food Science
Reference32 articles.
1. Krishna, P., Pandey, G., Thomas, R., and Parks, S. (2023). Improving Blueberry Fruit Nutritional Quality through Physiological and Genetic Interventions: A Review of Current Research and Future Directions. Antioxidants, 12. 2. Xiao, F., Wang, H., Li, Y., Cao, Y., Lv, X., and Xu, G. (2023). Object Detection and Recognition Techniques Based on Digital Image Processing and Traditional Machine Learning for Fruit and Vegetable Harvesting Robots: An Overview and Review. Agronomy, 13. 3. Wang, H., Lv, X., Xiao, F., and Sun, L. (2022). Analysis and Testing of Rigid–Flexible Coupling Collision Harvesting Processes in Blueberry Plants. Agriculture, 12. 4. Obsie, E.Y., Qu, H., Zhang, Y.J., Annis, S., and Drummond, F. (2023). Yolov5s-CA: An Improved Yolov5 Based on the Attention Mechanism for Mummy Berry Disease Detection. Agriculture, 13. 5. Yang, W., Ma, X., Hu, W., and Tang, P. (2022). Lightweight Blueberry Fruit Recognition Based on Multi-Scale and Attention Fusion NCBAM. Agronomy, 12.
|
|