基于改进YOLO v8模型的轻量化梨果实检测OA
Lightweight pear fruit detection based on an improved YOLO v8 model
为了解决梨果实采摘机器人目标检测模型在复杂果园环境下检测效果不佳,且难以部署于资源受限嵌入式平台的问题,本研究提出一种基于改进YOLO v8模型的轻量化梨果实检测模型:设计一种PR_Bottleneck模块,通过创新结构设计,有效削减模型参数量和计算量,显著降低模型运行负担;构建SPPF_CBAM模块,大幅增强模型对目标区域的聚焦能力,显著提升其在复杂场景下的目标识别精度;引入ADown下采样模块,降低模型复杂度和参数量,增强下采样过程的信息保留能力;引入WIoU损失函数,使模型在训练过程中的回归损失函数能够集中在平均质量的锚框,从而提高检测器的整体性能指标,增强模型的泛化能力.试验结果表明,与YOLO v8原始模型相比,改进模型精确率、召回率、平均精度均值(mAP50)分别提高了0.7个百分点、3.7个百分点、1.7个百分点,其参数量和计算量分别降低38.72%和30.86%,模型体积缩小38.33%,检测速率为1 s 85.4帧,较原始模型提升1 s 12.8帧.因此,改进YOLO v8轻量化模型不仅提高了检测精度和检测速率,而且明显降低了计算量和参数量,能够部署在嵌入式平台上快速有效进行梨果实的实时目标检测.
To address the challenges of poor detection performance of target detection models for pear-picking robots in complex orchard environments and the difficulty of deployment on resource-constrained embedded platforms,a lightweight pear fruit detection model based on an improved YOLO v8 is proposed in this study.First,a novel PR_Bottleneck module was intro-duced,which employed an innovative structural design to effectively reduce the model's parameter count and computational load,significantly easing the operational burden.Second,the SPPF_CBAM module was proposed to greatly enhance the model's focus on target regions,thereby significantly improving detection accuracy in complex scenarios.The model also incorporated an ADown downsampling module,which not only reduced model complexity and parameter count but also strengthened information retention during the downsampling process.Finally,the WIoU loss function was introduced,which enabled the model to concentrate on medium-quality anchor boxes during training,thus improving overall detection performance and enhancing the model's generali-zation capability.Experimental results showed that compared with the original YOLO v8 model,the precision,recall,and mean average precision(mAP50)of the improved model increased by 0.7,3.7,and 1.7 percentage points,respectively.The parameter count and computational load were reduced by 38.72%and 30.86%,respectively.The model size was reduced by 38.33%,and the detection speed reached 85.4 frames per second,an improvement of 12.8 frames per second over the original model.Therefore,the improved YOLO v8 lightweight model not only improves detection accuracy and speed,but also sig-nificantly decreases the computational load and parameter count,enabling rapid and effective real-time pear fruit detec-tion on embedded platforms.
罗云涛;张志安
南京理工大学机械工程学院,江苏南京 210094南京理工大学机械工程学院,江苏南京 210094
信息技术与安全科学
梨果实YOLO v8模型检测精度检测速率计算量参数量
pearfruitYOLO v8 modeldetection accuracydetection speedcomputational loadparameter count
《江苏农业学报》 2026 (3)
582-590,9
江苏省现代农机装备与技术示范推广项目(NJ2023-13)
评论