EMA-YOLO: A Novel Target-Detection Algorithm for Immature Yellow Peach Based on YOLOv8

Dandan Xu,Hao Xiong,Yue Liao,Hongruo Wang,Zhizhang Yuan,Hua Yin

DOI: https://doi.org/10.3390/s24123783

IF: 3.9

2024-06-12

Sensors

Abstract:Accurate determination of the number and location of immature small yellow peaches is crucial for bagging, thinning, and estimating yield in modern orchards. However, traditional methods have faced challenges in accurately distinguishing immature yellow peaches due to their resemblance to leaves and susceptibility to variations in shooting angles and distance. To address these issues, we proposed an improved target-detection model (EMA-YOLO) based on YOLOv8. Firstly, the sample space was enhanced algorithmically to improve the diversity of samples. Secondly, an EMA attention-mechanism module was introduced to encode global information; this module could further aggregate pixel-level features through dimensional interaction and strengthen small-target-detection capability by incorporating a 160 × 160 detection head. Finally, EIoU was utilized as a loss function to reduce the incidence of missed detections and false detections of the target small yellow peaches under the condition of high density of yellow peaches. Experimental results show that compared with the original YOLOv8n model, the EMA-YOLO model improves mAP by 4.2%, Furthermore, compared with SDD, Objectbox, YOLOv5n, and YOLOv7n, this model's mAP was improved by 30.1%, 14.2%,15.6%, and 7.2%, respectively. In addition, the EMA-YOLO model achieved good results under different conditions of illumination and shooting distance and significantly reduced the number of missed detections. Therefore, this method can provide technical support for smart management of yellow-peach orchards.

engineering, electrical & electronic,instruments & instrumentation,chemistry, analytical

What problem does this paper attempt to address?

The paper primarily focuses on the precise determination of the quantity and location of immature yellow peaches in modern orchard management. In practical applications, this information is crucial for operations such as bagging, thinning, and yield estimation. However, traditional identification methods struggle to accurately distinguish immature yellow peaches because they are very similar to leaves and are easily affected by changes in shooting angles and distances.

To overcome these challenges, researchers have proposed an improved object detection model based on YOLOv8 (EMA-YOLO). Specifically, the main contributions of this model include:

1. **Enhancing the sample space**: Enhancing the sample space through algorithmic means to improve sample diversity.
2. **Introducing the EMA attention mechanism module**: This module can encode global information and further aggregate pixel-level features through dimensional interaction, thereby enhancing the detection capability for small targets.
3. **Combining a 160×160 size object detection head**: Adding a smaller size detection head to enhance sensitivity to small targets.
4. **Using EIoU as the loss function**: EIoU can effectively reduce the miss rate and false detection rate of small target peaches in dense environments.

Experimental results show that compared to the original YOLOv8n model, the EMA-YOLO model achieved better results under different lighting conditions and shooting distances, significantly reducing missed detections and improving the mean Average Precision (mAP) by 4.2%. Additionally, compared to several other models (such as SDD, ObjectBox, YOLOv5n, and YOLOv7n), the mAP of this model increased by 30.1%, 14.2%, 15.6%, and 7.2%, respectively.

In summary, this paper aims to improve the detection accuracy and efficiency of immature yellow peaches by proposing a new object detection algorithm, providing technical support for intelligent orchard management.