基于改进 Yolov5s 的果园苹果检测算法
DOI:
作者:
作者单位:

作者简介:

通讯作者:

基金项目:


Orchard Apple Detection Algorithm Based on Improved YOLOv5s
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
    摘要:

    目的 针对果园环境下现有苹果检测识别精度低以及模型参数量大、计算量高的问题,提出一种基于 Yolov5s 的改进型果园苹果检测算法。 方法 首先在 YOLOv5s 算法中使用 Soft_NMS 替换模型原有的 NMS,改变原模型对预 测框的处理方式,从而减少果园环境下苹果遮挡重叠带来的错检、漏检情况,提升检测精度;然后再使用 OTA 优化 原模型的标签分配方式,将标签分配看作最优传输问题,并完整利用上下文信息减少模糊框的数量,更好地处理苹 果密集遮挡的问题,进一步提高模型对果园苹果的检测性能;最后使用 FasterNet 块替换主干网络中的卷积模块作 为新的特征提取网络,降低模型的参数量和计算量,从而实现模型的轻量化。 结果 实验结果显示:在果园苹果数据 集上,改进的算法模型相对于原模型,mAP 提升了 2. 8%,参数量和计算量分别下降了 21%和 29%,与其他同系列 主流检测模型相比,改进模型具有更高的检测精度,更低的参数量和计算量。 结论 改进的模型更适合果园场景下 的苹果检测,并可作为一种解决方案来参考并应用于相关领域。

    Abstract:

    Objective To address the problems of low detection accuracy high model parameter count and high computational cost in existing apple detection in orchard environments an improved orchard apple detection algorithm based on YOLOv5s is proposed. Methods First Soft-NMS was used in the YOLOv5s algorithm to replace the original NMS of the model and the way the original model processed prediction boxes was changed so as to reduce false detections and missed detections caused by the overlapping and occlusion of apples in the orchard environment and improve the detection accuracy. Then OTA was used to optimize the label assignment of the original model. The label assignment was regarded as an optimal transport problem and the context information was fully utilized to reduce the number of ambiguous boxes so as to better handle the problem of dense apple occlusion and further improve the model?? s detection performance for orchard apples. Finally the convolution module in the backbone network was replaced with FasterNet blocks which served as a new feature extraction network. This replacement reduced the parameter count and the computational cost of the model thereby achieving a lightweight model. Results The experimental results showed that on the orchard apple dataset compared with the original model the mAP of the improved algorithm model increased by 2. 8% and the parameter count and the computational cost decreased by 21% and 29% respectively. Compared with other mainstream detection models in the same series the improved model had higher detection accuracy fewer parameters and lower computational cost. Conclusion The improved model is more suitable for apple detection in orchard scenarios and can be used as a solution for reference and application in related fields.

    参考文献
    相似文献
    引证文献
引用本文

王彦辉,汪 军.基于改进 Yolov5s 的果园苹果检测算法[J].重庆工商大学学报(自然科学版),2026,43(4):53-58
WANG Yanhui WANG Jun. Orchard Apple Detection Algorithm Based on Improved YOLOv5s[J]. Journal of Chongqing Technology and Business University(Natural Science Edition),2026,43(4):53-58

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-07-07
×
2025年《中国学术期刊影响因子年报》发布