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

文献类型： 中文期刊

第一作者： 刘莫尘

作者： 刘莫尘;褚镇源;崔明诗;杨庆璐;王金星;杨化伟

作者机构：

关键词： 红熟期草莓识别;关键点预测;YOLO v8-Pose;注意力机制

期刊名称： 农业机械学报

ISSN： 1000-1298

年卷期： 2023 年 54 卷 0z2 期

页码： 244-251

收录情况： EI ; 北大核心 ; CSCD

摘要： 针对高架栽培模式下的大棚草莓,借鉴人体姿态检测算法,建立了改进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个百分点.模型内存占用量为22 MB,比YOLO v8-Pose的占用量小6 MB.此外,针对果园非结构化的特征,探究了光线、遮挡与拍摄角度对模型预测的影响.对比改进前后的模型在复杂环境下对红熟期草莓的识别与果柄预测情况,改进YOLO v8-Pose在受遮挡、光线和角度影响情况下的mAP-kp 分别为 94.52%、95.48%、94.63%,较 YOLO v8-Pose 分别提高 8.9、10.75、5.17 个百分点.改进 YOLO v8-Pose可在保证网络模型精度的同时对遮挡、光线和拍摄角度等影响均具有较好的鲁棒性,能够实现对复杂环境下红熟期草莓识别与果柄关键点预测.

分类号： TP391.4

  • 相关文献

[1]改进YOLOv4的温室环境下草莓生育期识别方法. 龙洁花,郭文忠,林森,文朝武,张宇,赵春江. 2021

[2]基于YOLOX改进模型的茶叶嫩芽识别方法. 俞龙,黄楚斌,唐劲驰,黄浩宜,周运峰,黄永权,孙佳琪. 2022

[3]融合注意力机制的开集猪脸识别方法. 王荣,高荣华,李奇峰,刘上豪,于沁杨,冯璐. 2023

[4]基于双线性注意力网络的农业灯诱害虫细粒度图像识别研究. 姚青,姚波,吕军,唐健,冯晋,朱旭华. 2021

[5]基于改进YOLO v5的复杂环境下桑树枝干识别定位方法. 李丽,卢世博,任浩,徐刚,周永忠. 2024

[6]改进YOLOv5在胡瓜钝绥螨品质管控系统中的设计与应用. 李建兴,刘振宇,马莹,张艳璇,宋江,纪茂源,旷树森. 2023

[7]基于多维间注意力机制的水稻病害识别模型. 王忠培,谢成军,董伟,管博伦. 2024

[8]基于Attention_DenseCNN的水稻问答系统问句分类. 王郝日钦,吴华瑞,冯帅,刘志超,许童羽. 2021

[9]基于改进YOLOv5s的硬核期葡萄簇检测. 冯晓,张辉,刘运超,张微,李小红,马中杰. 2024

[10]无人机遥感影像在油菜品种识别中的应用. 李婕,李毅,张瑞杰,李俐俐,李礼,姚剑,乔江伟. 2022

[11]基于注意力机制及多尺度特征融合的番茄叶片缺素图像分类方法. 韩旭,赵春江,吴华瑞,朱华吉,张燕. 2021

[12]基于注意力机制的农业文本命名实体识别. 赵鹏飞,赵春江,吴华瑞,王维. 2021

[13]基于注意力机制和多尺度残差网络的农作物病害识别. 黄林生,罗耀武,杨小冬,杨贵军,王道勇. 2021

[14]简单三维注意力机制水稻病害识别模型. 王忠培,董伟,朱静波,谢成军. 2023

[15]基于改进UperNet的结球甘蓝叶球识别方法. 朱轶萍,吴华瑞,郭旺,吴小燕. 2024

[16]基于改进YOLOV5s网络的奶牛多尺度行为识别方法. 白强,高荣华,赵春江,李奇峰,王荣,李书琴. 2022

[17]基于CycleGAN和注意力增强迁移学习的小样本鱼类识别. 刘世晶,刘阳春,钱程,郑浩君,周捷,张成林. 2023

[18]基于改进YOLOv4算法的番茄叶部病害识别方法. 储鑫,李祥,罗斌,王晓冬,黄硕. 2023

[19]基于双阶段注意力机制和LSTM的鸡舍氨气浓度预测算法. 郭祥云,连京华,李慧敏,孙凯. 2021

[20]基于多尺度和注意力机制的番茄病害识别方法. 张宁,吴华瑞,韩笑,缪祎晟. 2021

作者其他论文 更多>>

微信小程序