基于改进YOLOv5复杂环境下的草莓果实识别检测

RECOGNITION AND DETECTION OF STRAWBERRY FRUIT BASED ON IMPROVED YOLOV5 COMPLEX ENVIRONMENT

  • 摘要: 针对目前草莓识别检测存在背景干扰、重叠遮挡、尺寸差异严重等问题,提出一种基于改进YOLOv5算法的检测方法。在YOLOv5的backbone与Neck中引入TC-SE模块,在PAN结构中添加特征融合通道,提高小目标检测性能;在模型输出的后处理阶段引入高斯加权,提高重叠遮挡目标的召回率;在位置损失函数中引入回归之间的向量角度,重新定义惩罚指标,提高复杂背景下的识别精度。改进后模型的MAP为93%,较原模型提高了3.7百分点,检测速度为31.15ms,对比实验证明了综合性能指标优于其他算法,并满足采摘的实时性要求。

     

    Abstract: In view of the existing problems in strawberry recognition and detection, such as background interference, overlapping occlusion, and serious size difference, this paper proposes a detection method based on improved YOLOv5 algorithm. TC-SE module was introduced into YOLOv5's backbone and Neck, and feature fusion channel was added to PAN structure to improve small target detection performance. In the post-processing stage of model output, Gaussian weighting was introduced to improve the recall rate of overlapping occluded targets. The vector angle between regressions was introduced into the position loss function, and the penalty index was redefined to improve the recognition accuracy under complex background. The MAP of the improved model was 93%, which was 3.7 percentage points higher than the original model, and the detection speed was 31.15ms. The comparative experiment proves that the comprehensive performance index is better than other algorithms, and meets the real-time requirements of picking.

     

/

返回文章
返回