基于改进YOLO v8-Pose的红熟期草莓识别和果柄检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金项目(32001419)和山东省重点研发计划(重大科技创新工程)项目(2022CXGC020701)


Red Ripe Strawberry Recognition and Stem Detection Based on Improved YOLO v8-Pose
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对高架栽培模式下的大棚草莓,借鉴人体姿态检测算法,建立了改进YOLO v8-Pose模型对红熟期草莓进行识别与果柄关键点检测。通过对比YOLO v5-Pose、YOLO v7-Pose、YOLO v8-Pose模型,确定使用YOLO v8-Pose模型作为对红熟期草莓识别与关键点预测的模型。以YOLO v8-Pose为基础,对其网络结构添加Slim-neck模块与CBAM注意力机制模块,提高模型对小目标物体的特征提取能力,以适应草莓数据集的特点。改进YOLO v8-Pose能够有效检测红熟期草莓并准确标记出果柄关键点,P、R、mAP-kp分别为98.14%、94.54%、97.91%,比YOLO v8-Pose分别提高5.41、5.31、8.29个百分点。模型内存占用量为22MB,比YOLO v8-Pose的占用量小 6MB。此外,针对果园非结构化的特征,探究了光线、遮挡与拍摄角度对模型预测的影响。对比改进前后的模型在复杂环境下对红熟期草莓的识别与果柄预测情况,改进YOLO v8-Pose在受遮挡、光线和角度影响情况下的mAP-kp分别为94.52%、95.48%、94.63%,较YOLO v8-Pose分别提高8.9、10.75、5.17个百分点。改进YOLO v8-Pose可在保证网络模型精度的同时对遮挡、光线和拍摄角度等影响均具有较好的鲁棒性,能够实现对复杂环境下红熟期草莓识别与果柄关键点预测。

    Abstract:

    The improved YOLO v8-Pose model was established to identify red ripe strawberries and detect the key points of the stem in greenhouse strawberries under elevated cultivation mode. By comparing the YOLO v5-Pose, YOLO v7-Pose and YOLO v8-Pose models, the YOLO v8-Pose model was determined to be used as the model to identify and predict the key points of red ripe strawberries. Based on YOLO v8-Pose, Slim-neck module and CBAM attention mechanism module were added to its network structure to improve the feature extraction ability of the model for small target objects, so as to adapt to the characteristics of strawberry data set. The P, R and mAP-kp of the improved YOLO v8-Pose were 98.14%, 94.54% and 97.91%, respectively, which can effectively detect red ripe strawberries and accurately mark the key points of the fruit stalk, which was 5.41, 5.31 and 8.29 percentage points higher than that of YOLO v8-Pose. The model memory footprint was 22MB, which was 6MB less than that of the YOLO v8-Pose footprint.In addition, according to the unstructured characteristics of the orchard, the influence of light, occlusion and shooting angle on the model prediction was explored. Compared with the recognition and stem prediction of the improved YOLO v8-Pose model in the complex environment, the mAP-kp of the improved YOLO v8-Pose under the influence of occlusion, light and angle was 94.52%, 95.48% and 94.63%, respectively. Compared with YOLO v8-Pose, it was 8.9, 10.75 and 5.17 percentage points higher, respectively. The improved YOLO v8-Pose can ensure the accuracy of the network model, and at the same time, it had good robustness to the effects of occlusion, light and shooting angle, etc., which can realize the identification of red ripe strawberries in complex environments and the prediction of key points of fruit stalk.

    参考文献
    相似文献
    引证文献
引用本文

刘莫尘,褚镇源,崔明诗,杨庆璐,王金星,杨化伟.基于改进YOLO v8-Pose的红熟期草莓识别和果柄检测[J].农业机械学报,2023,54(s2):244-251. LIU Mochen, CHU Zhenyuan, CUI Mingshi, YANG Qinglu, WANG Jinxing, YANG Huawei. Red Ripe Strawberry Recognition and Stem Detection Based on Improved YOLO v8-Pose[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(s2):244-251.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-06-28
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-08-29
  • 出版日期: