基于改进的 YOLOv5s 架构的实时草莓检测在开放环境中的机器人采摘

💡 原文中文,约500字,阅读约需2分钟。
📝

内容提要

该研究提出了一种基于YOLOv5的自定义目标检测模型,用于户外环境中的草莓检测。通过改进YOLOv5s的架构,提高了特征梯度流,并结合空间金字塔池化和Cross Stage Partial Net,提升了在草莓数据集上的泛化能力。该模型在平均精度上达到了80.3%,超过了其他竞争模型。同时,该模型具有快速的推理速度和适用于实时草莓检测和定位的网络参数。

🎯

关键要点

  • 该研究提出了一种基于YOLOv5的自定义目标检测模型,用于户外环境中的草莓检测。
  • 修改了YOLOv5s的原始架构,通过将C3模块替换为C2f模块,提供了更好的特征梯度流。
  • 结合空间金字塔池化和Cross Stage Partial Net,提高了在草莓数据集上的泛化能力。
  • 该模型在平均精度上达到了80.3%,超过了YOLOv3-tiny、YOLOv5s、YOLOv5s-C2f和YOLOv8s等竞争模型。
  • YOLOv3-tiny、YOLOv5s、YOLOv5s-C2f和YOLOv8s的平均精度分别为73.4%、77.8%、79.8%和79.3%。
  • 该模型具有快速的推理速度,适用于实时草莓检测和定位。
➡️

继续阅读