An improved YOLOv5s model for assessing apple graspability in automated harvesting scene

文献类型： 外文期刊

第一作者： Li, Huibin

作者： Li, Huibin;Qian, Jianping;Yu, Qiangyi;Shi, Yun;Yang, Peng;Liu, Huaiyang;Liu, Xiang;Geng, Changxing

作者机构：

关键词： apple harvesting; BYOL; attention mechanism; occlusion detection; YOLOv5S

期刊名称：FRONTIERS IN PLANT SCIENCE （ 影响因子：5.6； 五年影响因子：6.8 ）

ISSN： 1664-462X

年卷期： 2023 年 14 卷

页码：

收录情况： SCI

摘要： IntroductionWith continuously increasing labor costs, an urgent need for automated apple- Qpicking equipment has emerged in the agricultural sector. Prior to apple harvesting, it is imperative that the equipment not only accurately locates the apples, but also discerns the graspability of the fruit. While numerous studies on apple detection have been conducted, the challenges related to determining apple graspability remain unresolved.MethodsThis study introduces a method for detecting multi-occluded apples based on an enhanced YOLOv5s model, with the aim of identifying the type of apple occlusion in complex orchard environments and determining apple graspability. Using bootstrap your own atent(BYOL) and knowledge transfer(KT) strategies, we effectively enhance the classification accuracy for multi-occluded apples while reducing data production costs. A selective kernel (SK) module is also incorporated, enabling the network model to more precisely identify various apple occlusion types. To evaluate the performance of our network model, we define three key metrics: APGA, APTUGA, and APUGA, representing the average detection accuracy for graspable, temporarily ungraspable, and ungraspable apples, respectively.ResultsExperimental results indicate that the improved YOLOv5s model performs exceptionally well, achieving detection accuracies of 94.78%, 93.86%, and 94.98% for APGA, APTUGA, and APUGA, respectively.DiscussionCompared to current lightweight network models such as YOLOX-s and YOLOv7s, our proposed method demonstrates significant advantages across multiple evaluation metrics. In future research, we intend to integrate fruit posture and occlusion detection to f]urther enhance the visual perception capabilities of apple-picking equipment.

分类号：

  • 相关文献
作者其他论文 更多>>

微信小程序