基于改进YOLOv5的葡萄果穗检测算法

GRAPE BERRY DETECTION ALGORITHM BASED ON IMPROVED YOLOv5

  • 摘要: 针对现代化农场中葡萄果穗分布密集、背景复杂导致其检测精度较低的问题,提出一种基于改进YOLOv5的葡萄果穗快速精确检测算法。以YOLOv5为基础目标检测算法,使用坐标法来为机制对特征提取网络进行改进,增强其特征表达能力,利用Bi-FPN对图像特征的高效融合,增强网络整体预测能力。实验结果表明,该模型检测精度可达83.1%,可以在复杂环境中有效地检测葡萄果穗。

     

    Abstract: To address the problem of low detection accuracy due to the dense distribution and complex background of grape clusters in modern farms, a fast and accurate grape cluster detection method based on improved YOLOv5 is proposed. YOLOv5 was used as the base target detection model, and the feature extraction network was improved by using the coordinate attention mechanism to enhance its feature representation capability, and Bi-FPN was used for efficient fusion of image features to enhance the overall prediction capability of the network. The experimental results show that the detection accuracy of the model can reach 83.1%, which can effectively detect grape clusters in complex environments.

     

/

返回文章
返回