Faculty of Mechanical Engineering, Zhejiang Sci-Tech University, Hangzhou 310018, China.
Special Equipment Institute, Hangzhou Vocational & Technical College, Hangzhou 310018, China.
Sensors (Basel). 2023 Feb 1;23(3):1562. doi: 10.3390/s23031562.
An improved YOLOv5 algorithm for the efficient recognition and detection of asparagus with a high accuracy in complex environments was proposed in this study to realize the intelligent machine harvesting of green asparagus. The coordinate attention (CA) mechanism was added to the backbone feature extraction network, which focused more attention on the growth characteristics of asparagus. In the neck part of the algorithm, PANet was replaced with BiFPN, which enhanced the feature propagation and reuse. At the same time, a dataset of asparagus in complex environments under different weather conditions was constructed, and the performance variations of the models with distinct attention mechanisms and feature fusion networks were compared through experiments. Experimental results showed that the mAP of the improved YOLOv5 model increased by 4.22% and reached 98.69%, compared with the YOLOv5 prototype network. Thus, the improved YOLOv5 algorithm can effectively detect asparagus and provide technical support for intelligent machine harvesting of asparagus in different weather conditions and complex environments.
本研究提出了一种改进的 YOLOv5 算法,用于在复杂环境中高效识别和检测芦笋,以实现绿芦笋的智能机器采摘。该算法在骨干特征提取网络中添加了坐标注意力(CA)机制,使其更加关注芦笋的生长特征。在算法的颈部部分,用 BiFPN 替换了 PANet,增强了特征的传播和复用。同时,构建了一个在不同天气条件下复杂环境中芦笋的数据集,并通过实验比较了具有不同注意力机制和特征融合网络的模型的性能变化。实验结果表明,与 YOLOv5 原型网络相比,改进后的 YOLOv5 模型的 mAP 提高了 4.22%,达到 98.69%。因此,改进后的 YOLOv5 算法可以有效地检测芦笋,为不同天气条件和复杂环境下的芦笋智能机器采摘提供技术支持。