搜索

x

基于改进YOLOv8模型的火龙果成熟度检测方法

A Ripeness Identification Method of Pitaya (Dragon Fruit) Based on the Improved YOLOv8 Model

  • 摘要: 为了解决非结构环境中复杂背景及目标遮挡导致火龙果成熟度识别精度低的问题,本研究提出了一种基于改进YOLOv8n的火龙果成熟度识别模型。该模型采用MobileNetV3-Small作为骨干网络,并引入DIoU损失函数,构建了YOLOv8-MD模型。此模型通过结合田间图像中火龙果局部低维信息与全局特征,提高了对火龙果成熟度特征的获取能力,有效应对了非结构化环境中目标遮挡与小尺寸目标相关的问题。结果表明:在果园非结构化环境中,YOLOv8-MD模型的精准率达到94.2%,召回率为87.5%,平均精度达95.4%。相比基线YOLOv8n模型,这些指标分别提升了7.4%、7.0%和6.1%。此外,模型参数量降至2.77 M,总浮点运算量压缩至6.4 GFlops,相对YOLOv8n基线模型减少了20.9%。在Nvidia Orin Nano平台上的部署测试表明,该模型实现了83.13 FPS的推理速度,相对传统YOLO模型增幅达9.2%~41.1%。研究结果不仅显著提升了火龙果成熟度检测的精度和效率,还为火龙果或相似特征果实的田间成熟度识别模型构建提供了思路。

     

    Abstract: To address the low accuracy of pitaya ripeness recognition caused by complex backgrounds and target occlusions in unstructured environments, this study proposed an improved YOLOv8n-based model for pitaya ripeness detection. The YOLOv8-MD model was built based on YOLOv8n by integrating the MobileNetV3-Small backbone network and incorporating the DIoU loss function. This design fused local low-dimensional information and global features of pitayas in field images, thereby enhancing the model’s capability to extract ripeness-related features and mitigating challenges associated with target occlusion and small-sized targets in unstructured settings. Experiment results demonstrated that in unstructured orchard environments, the proposed model achieves a precision of 94.2%, a recall of 87.5% and a mean average precision (mAP) of 95.4%, representing improvements of 7.4%, 7.0%, and 6.1% respectively over the baseline YOLOv8n model. Concurrently, the model’s parameter count was reduced to 2.77 M, and the total floating-point operations (FLOPs) were compressed to 6.4 GFLOPS, a 20.9% reduction compared to the baseline. Deployment testing on the Nvidia Orin Nano platform yielded an inference speed of 83.13 FPS, which was 9.2% to 41.1% faster than the traditional YOLO models. This research provided a valuable framework for the development of field-based ripeness detection models for pitayas and other fruits with similar characteristics.

     

/

返回文章
返回