您好,欢迎访问北京市农林科学院 机构知识库!

SGW-YOLOv8n: An Improved YOLOv8n-Based Model for Apple Detection and Segmentation in Complex Orchard Environments

文献类型: 外文期刊

作者: Wu, Tao 1 ; Miao, Zhonghua 1 ; Huang, Wenlei 1 ; Han, Wenkai 2 ; Guo, Zhengwei 2 ; Li, Tao 2 ;

作者机构: 1.Shanghai Univ, Sch Mechatron Engn & Automat, Shanghai 200444, Peoples R China

2.Beijing Acad Agr & Forestry Sci, Intelligent Equipment Res Ctr, Beijing 100097, Peoples R China

关键词: fruit detection; fruit segmentation; deep learning; occluded targets; attention mechanisms

期刊名称:AGRICULTURE-BASEL ( 影响因子:3.6; 五年影响因子:3.8 )

ISSN:

年卷期: 2024 年 14 卷 11 期

页码:

收录情况: SCI

摘要: This study addresses the problem of detecting occluded apples in complex unstructured environments in orchards and proposes an apple detection and segmentation model based on improved YOLOv8n-SGW-YOLOv8n. The model improves apple detection and segmentation by combining the SPD-Conv convolution module, the GAM global attention mechanism, and the Wise-IoU loss function, which enhances the accuracy and robustness. The SPD-Conv module preserves fine-grained features in the image by converting spatial information into channel information, which is particularly suitable for small target detection. The GAM global attention mechanism enhances the recognition of occluded targets by strengthening the feature representation of channel and spatial dimensions. The Wise-IoU loss function further optimises the regression accuracy of the target frame. Finally, the pre-prepared dataset is used for model training and validation. The results show that the SGW-YOLOv8n model significantly improves relative to the original YOLOv8n in target detection and instance segmentation tasks, especially in occlusion scenes. The model improves the detection mAP to 75.9% and the segmentation mAP to 75.7% and maintains a processing speed of 44.37 FPS, which can meet the real-time requirements, providing effective technical support for the detection and segmentation of fruits in complex unstructured environments for fruit harvesting robots.

  • 相关文献
作者其他论文 更多>>