基于改进YOLOv8n的设施环境下成熟番茄检测方法

    Detection method for mature tomatoes in facility environments based on improved YOLOv8n

    • 摘要:
      目的 解决设施环境下成熟番茄检测所面临的光线变化、枝叶遮挡、果实重叠及不同距离识别等问题,
      方法 本研究设计出一种针对上述问题的成熟番茄果实检测模型YOLOv8n-SPMF。首先,使用SPDConv替换YOLOv8n主干网络中的Conv模块,提高小目标番茄的检测精度;其次,在主干网络中添加PSCEA注意力机制,提取番茄的局部细节和边缘信息,增强模型特征提取能力;然后,将主干网络中的SPPF模块替换为混合池化模块(MixSPPF),以增强不同层次特征之间的信息融合能力;最后,采用Focaler-MPDIoU损失函数,提升了番茄目标检测中复杂场景下的边界框回归表现。
      结果 试验结果表明,YOLOv8n-SPMF模型在测试集上的mAP50为96.24%, mAP50-95为81.36%,召回率为90.33%,模型参数量为4.13M,每张图像的检测速度为11.7ms。相比于YOLOv3-tiny、YOLOv5n、YOLOv6n、YOLOv7-tiny、Faster-RCNN、YOLOv8n、YOLOv9t、YOLOv10n、YOLOv11n、YOLOv12n,模型mAP50分别提升3.57、0.78、1.40、0.05、2.64、0.74、0.66、0.72、0.50、1.26个百分点,精确率分别提高了0.48、0.95、0.69、0.38、4.90、0.11、0.63、1.90、0.43、0.61个百分点。
      结论 本文提出的YOLOv8n-SPMF模型在设施环境下对成熟番茄果实检测具有较高的准确性,可为设施环境下番茄智能采摘提供有效的技术支持。

       

      Abstract:
      Objective To address the challenges of varying illumination, occlusion by foliage and branches, fruit overlapping, and multi-distance object recognition in mature tomato detection within facility environments.
      Method This study designed a mature tomato fruit detection model, YOLOv8n-SPMF, to solve the aforementioned problems. Firstly, the Conv module in the YOLOv8n backbone network was replaced with SPDConv to improve the detection accuracy of small tomatoes. Secondly, a PSCEA attention mechanism was added to the backbone network to extract local details and edge information of tomatoes, thereby enhancing the model's feature extraction capability. Then, the SPPF module in the backbone was replaced with a mixed pooling module (MixSPPF) to strengthen the information fusion among different level features. Finally, a Focaler-MPDIoU loss function was adopted to improve the bounding box regression performance in complex scenarios.
      Result Experimental results showed that the YOLOv8n-SPMF model achieved an mAP50 of 96.24% on the test set, with an mAP50-95 of 81.36%, a Recall of 90.33%, a model parameter count of 4.13 M, and a detection speed of 11.7 ms per image. Compared with YOLOv3-tiny, YOLOv5n, YOLOv6n, YOLOv7-tiny, Faster-RCNN, YOLOv8n, YOLOv9t, YOLOv10n, YOLOv11n and YOLOv12n, the model's mAP50 was improved by 3.57, 0.78, 1.40, 0.05, 2.64, 0.74, 0.66, 0.72, 0.50 and 1.26 percentage points, respectively, and the Precision was increased by 0.48, 0.95, 0.69, 0.38, 4.90, 0.11, 0.63, 1.90, 0.43 and 0.61 percentage points, respectively.
      Conclusion The YOLOv8n-SPMF model proposed in this paper exhibits high accuracy for mature tomato fruit detection in facility environments and can provide effective technical support for intelligent tomato harvesting.

       

    /

    返回文章
    返回