Aug, 2023

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

TL;DR该研究提出了一种基于 YOLOv5 的自定义目标检测模型,用于在户外环境中检测草莓。修改了 YOLOv5s 的原始架构,通过在骨干网络中将 C3 模块替换为 C2f 模块,提供了更好的特征梯度流。通过将 YOLOv5s 的最终层中的空间金字塔池化快速与 Cross Stage Partial Net 相结合,提高了在草莓数据集上的泛化能力。验证了该方法在开放环境中检测草莓的优势,并与四个竞争检测模型进行了比较。结果表明,通过所提出的架构实现了最高 80.3% 的平均精度,而 YOLOv3-tiny、YOLOv5s、YOLOv5s-C2f 和 YOLOv8s 分别获得了 73.4%、77.8%、79.8% 和 79.3% 的平均精度。该模型具有快速的推理速度和适用于实时草莓检测和定位的充足网络参数。