基于YOLOv5的苹果目标识别方法研究

摘要

苹果采摘机器人的图像识别系统必须对苹果目标能够快速准确地识别,针对现有的苹果目标识别方法存在准确率不高的问题,基于YOLOv5算法在数据增强和主干网络层两个方面对YOLOv5算法进行改进。改进后的算法经过优化参数训练后,每一张图片的识别速度为0.042s,mAP达到了0.989,训练权重大小约为50mb,可以准确快速识别苹果目标。