Xu Dandan, Xiong Hao, Liao Yue, Wang Hongruo, Yuan Zhizhang, Yin Hua
School of Software, Jiangxi Agricultural University, Nanchang 330045, China.
School of Software, Jiangxi Normal University, Nanchang 330045, China.
Sensors (Basel). 2024 Jun 11;24(12):3783. doi: 10.3390/s24123783.
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.
准确确定未成熟小黄桃的数量和位置对于现代果园的套袋、疏果和产量估计至关重要。然而,传统方法在准确区分未成熟黄桃方面面临挑战,因为它们与树叶相似,并且容易受到拍摄角度和距离变化的影响。为了解决这些问题,我们提出了一种基于YOLOv8的改进目标检测模型(EMA-YOLO)。首先,通过算法增强样本空间以提高样本的多样性。其次,引入了EMA注意力机制模块来编码全局信息;该模块可以通过维度交互进一步聚合像素级特征,并通过合并一个160×160的检测头来增强小目标检测能力。最后,使用EIoU作为损失函数,以减少在黄桃高密度情况下目标小黄桃漏检和误检的发生率。实验结果表明,与原始的YOLOv8n模型相比,EMA-YOLO模型的平均精度均值(mAP)提高了4.2%。此外,与SDD、Objectbox、YOLOv5n和YOLOv7n相比,该模型的mAP分别提高了30.1%、14.2%、15.6%和7.2%。此外,EMA-YOLO模型在不同光照和拍摄距离条件下均取得了良好的效果,并显著减少了漏检数量。因此,该方法可为黄桃果园的智能管理提供技术支持。