首页 | 本学科首页   官方微博 | 高级检索  
     检索      

基于改进YOLOv5柑橘成熟果实检测算法
引用本文:蹇川,郑永强,刘艳梅,马岩岩,易时来,吕强,谢让金.基于改进YOLOv5柑橘成熟果实检测算法[J].中国果业信息,2024,53(1).
作者姓名:蹇川  郑永强  刘艳梅  马岩岩  易时来  吕强  谢让金
作者单位:西南大学工程技术学院,西南大学柑桔研究所/国家柑橘工程技术研究中心/国家数字种植业(柑橘)创新分中心,重庆市北碚区经济作物技术推广站,西南大学柑桔研究所/国家柑橘工程技术研究中心/国家数字种植业(柑橘)创新分中心,西南大学柑桔研究所/国家柑橘工程技术研究中心/国家数字种植业(柑橘)创新分中心,西南大学柑桔研究所/国家柑橘工程技术研究中心/国家数字种植业(柑橘)创新分中心,西南大学柑桔研究所/国家柑橘工程技术研究中心/国家数字种植业(柑橘)创新分中心
摘    要:为提高柑橘果园产量巡检作业效率和质量,巡检机器人精准检测成熟柑橘果实至关重要。考虑到成熟果实特有的颜色空间、果实遮挡导致的小目标以及巡检机器人有限的硬件资源,本文提出一种简单有效的基于改进YOLOv5的柑橘成熟果实检测方法。本工作主要设计一个由三层Context Aggregation Block (CAB)组成的特征金字塔并将其插入到YOLOv5网络的Head部分。此改进的YOLOv5模型具有如下优点:(1)集成的底层CAB模块通过特征通道注意力机制和空间注意力机制更好更快地学习小目标局部成熟果实的颜色和纹理特征、重叠果实的边缘特征;(2)多层CAB模块构建的特征金字塔能够有效地避免小目标随网络深度增加而消失,从而降低小目标果实漏检率。柑橘成熟果实识别验证试验结果表明,改进YOLOv5网络模型的精确度和平均精度分别为98.21%和98.07%,较原始YOLOv5分别提升了0.31%和0.17%,较Faster-RCNN分别提升了8.41%和8.31%。同时,单张成熟柑橘果实图像的平均检测时间32.5 ms,模型占用内存15.8 mb。该结果表明改进 YOLOv5模型可实现果园自然环境下柑橘成熟果实快速准确的检测识别,可为柑橘果园巡检机器人产量巡检提供技术支持。

关 键 词:柑橘  成熟果实  YOLOv5  Context  Aggregation  Block  目标检测  产量预估
收稿时间:2023/9/8 0:00:00
修稿时间:2023/12/5 0:00:00
点击此处可从《中国果业信息》浏览原始摘要信息
点击此处可从《中国果业信息》下载免费的PDF全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号