基于改进YOLOv8的苹果检测方法.pdf

第 13 期2023 年 7 月无线互联科技Wireless Internet TechnologyNo.13July,2023基金项目:吉林省科技发展计划项目;项目名称:基于数据挖掘技术的全基因组选择方法研发及云计算平台体系构建;项目编号:YDZJ202201ZYTS692。作者简介:杜宝侠(1998),男,山东泰安人,硕士研究生;研究方向:计算机视觉。通信作者:唐友(1979),男,黑龙江龙江人,教授,博士;研究方向:生物信息学,农业信息化。基于改进 YOLOv8 的苹果检测方法杜宝侠1,2,唐 友1,2,辛 鹏1,杨 牧3(1.吉林化工学院,吉林 吉林 132022;2.吉林农业科技学院,吉林 吉林 132101;3.一汽东机工减振器有限公司,吉林 长春 130001)摘要:实现苹果自动检测对推动苹果生产管理自动化、智能化有着重要意义。为了解决当前苹果检测方法准确率低、鲁棒性差等问题,文章提出了一种基于改进 YOLOv8 算法的苹果检测模型,通过引入GAM 注意力机制提升模型的检测精度。在苹果检测数据集上进行验证,基于改进的 YOLOv8 算法和原始 YOLOv8 算法相比,mAP 0.5 提高了 1.7%,mAP 0.5 0.95 提高了 2.2%。实验结果表明,文章提出的改进 YOLOv8 算法更能满足实际情况中对苹果检测的要求。关键词:苹果检测;YOLOv8;注意力机制中图分类号:S661.1;TP391.41 文献标志码:A0 引言 我国是农业大国,水果生产和消费数量在世界上位居前列,在众多水果中,苹果的种植面积和产量均超过世界总量的 50%,均居世界首位1。苹果生产是我国农业经济发展的重要支柱,但我国的大部分苹果生产还处于传统的农业模式阶段,在苹果生产管理过程中需要耗费大量的人力物力,这也导致了生产效率的低下和农民收入水平的不稳定。因此,推动苹果生产过程的自动化、智能化发展具有重要意义。近年来,现代农业结合计算机视觉技术的各种智能设备在农业生产中的应用日益普及,并且成为农业领域的研究热点。这些智能设备的应用极大地提高了农业生产效率和精度,减少了对劳动力和资源的依赖,并降低了农业生产成本,促进了农业的可持续发展。具体到苹果的生产方面,计算机视觉技术可以通过拍摄的图片或视频进行产量预测、病虫害识别,也可以通过图像信息对苹果果实进行优劣分级等,这些技术极大地推动了苹果生产管理的智能化发展。其中,利用目标检测技术对苹果果实进行识别和定位,从而帮助智能机器人实现自动采摘是一个重要的研究方向。在对苹果的检测方面,可以分为传统算法和基于深度学习的算法两类。传统算法大多依赖于目标的颜色、形状等外部特征来区分目标和背景2,存在特征提取困难、模型泛化能力差、在环境变化时性能下降严重等问题。近年来,基于卷积神经网络的深度学习技术在目标检测领域取得了显著的成功,相较于传统的机器学习方法,基于深度学习的目标检测技术场景适应性强,精度更高。牟其松3简化了 YOLOv3 的主干网络,并通过改进损失函数,提高了苹果检测的精度;张中华等4提出了一种基于优化 FCOS 网络的绿色苹果检测模型,通过改进特征融合 FPN 提高了对绿色苹果的检测精度;Meili 等5针对复杂环境下的小苹果提出了一种平衡特征金字塔网络 BFP Net。上述研究通过对先进目标检测算法进行改进,增强其检测苹果的能力,但仍存在模型较复杂、检测速度慢、不容易部署等问题。在实际的果园场景中,要实现苹果的实时检测,不仅需要较好的检测精度,对检测的速度也有较高的要求。因此,本文以单阶段目标检测算法中最先进的YOLOv8 模型为基础并加以改进,来实现苹果果实的自动检测。1 YOLOv8 模型原理1.1 YOLOv8 模型结构 YOLOv8 是 YOLO 系列最新的模型,具有非常优秀的检测精度和速度。根据网络的深度与特征图的宽度大小,YOLOv8 算法分为:YOLOv8-n、YOLOv8-s、YOLOv8-m、YOLOv8-l、和 YOLOv8-x 5 个版本。按911第 13 期2023 年 7 月无线互联科技研究创新No.13July,2023照网络结构图,YOLOv8 可分为:Input、Backbone、Neck和 Head 4 部分。其中,Input 输入端负责将输入图片缩放到训练需要的尺寸,并包含缩放、改变图片色调和 Mosaic 数据增强等数据增强操作;Backbone 主干网络用来进行目标特征的提取,由卷积模块 Conv、C2f 结构和 YOLOv5 中使用的 SPPF 模块构成;Neck用来将不同维度的特征进行加强融合,其结构遵循特征金字塔(Feature Pyramid Network,FPN)和路径聚合结构(Path Aggregation Network,PAN),并删除了YOLOv5 上采样阶段的卷积操作;Head 部分将加强后的特征进行计算,最终得到不同目标的置信度和位置。考虑到模型检测的速度和精度,本次实验以检测速度最快的 YOLOv8-n 为基线模型,并在此基础上加以改进。1.2 数据增强 数据增强是在不改变图像重要信息的前提下,基于有限的数据生成更多同样有效的数据的方法。可以极大地丰富训练数据的多样性,从而使得训练后的模型具有更强的泛化能力。改变图片的颜色和形状是最基础和常见的数据增强方式,在本研究中,使用随机改变图片的色调、饱和度、明亮度,同时对图片进行随机缩放和平移这些传统的数据增强方式。除了这些基础的数据增强方式外,有些研究还提出了更高效的数据增强方式,如 Mosaic 等6。Mosaic 数据增强方法的主要思想是将 4 张图片随机裁剪缩放,再拼接到一张图上作为训练数据。在本次实验中使用传统数据增强和 Mosaic 方法相结合的数据增强方式。2 YOLOv8 模型改进方法2.1 GAM 注意力机制原理 注意力机制可以赋予空间中不同通道或区域以不同的权重,从而帮助模型专注于更重要信息的提取。全局注意力机制(Global Attention Mechanism,GAM)7,可以通过减少信息缩减和放大全局交互表示来提高深度神经网络的性能。该机制使用了通道-空间注意机制,在通道层面,GAM 首先将输入特征图进行维度转换,接着输入到 MLP 恢复原来的维度,最后通过 Sigmoid 输出。在空间层面,GAM 通过一系列的卷积处理再送入 Sigmoid 输出,最终实现两个层面注意力作用的叠加。GAM 的整体模块如图 1 所示。图 1 GAM 注意力机制原理2.2 改进的模型结构 在 YOLOv8 模型的改进方面,将 GAM 注意力添加在 Neck 和 Head 的连接部分,引入的 GAM 注意力模块可以对通道和空间维度中的特征图信息进行自适应学习,并对特征进行重新拟合和分配,帮助模型更聚焦于苹果目标,提高了特征图中小目标的权重占比,从而加强了模型对其特征的训练。最终改进的YOLOv8 网络模型如图 2 所示。3 实验结果与分析3.1 数据集本实验使用的数据集来源为 https:/ 张。其中,训练集、验证集和测试集的图像数量分别为 273 张、14 张和 10 张。如图 3 所示为数据集示例。3.2 实验环境 本文实验平台使用 Ubuntu 20.02 操作系统,CPU处理器为 2 个 Intel(R)Xeon(R)Silver 4110,GPU 为GeForce RTX 3090,显存为 24 GB。模型训练框架为Pytorch 1.12.0,具体训练参数设置如表 1 所示。3.3 评估方法 为验证改进方法的有效性,使用 P(Precision:精度)、R(Recall:召 回 率)、mAP 0.5(Mean Average Precision:各类别在 IOU=0.5 下的 AP 平均值)和 mAP 0.5 0.95(Mean Average Precision:各类别在不同 IOU下的 AP 平均值)作为模型评价指标。另外,TP(True Positive)表 示 将 正 类 预 测 为 正 类 数,FN(False Negative)表 示 将 正 类 预 测 为 负 类 数,FP(False Positive)表 示 将 负 类 预 测 为 正 类 数,TN(True Negative)表示将负类预测为负类数。对应的评价指标计算方法如下。精度是衡量模型识别出的真实阳性样本数占所有被模型识别为阳性样本数的比例。精度越高,说明模型识别出的阳性样本更准确,具有更好的准确性。021第 13 期2023 年 7 月无线互联科技研究创新No.13July,2023 图 2 改进后的 YOLOv8 结构图 3 数据集示例表 1 训练参数参数名参数值参数名参数值Learning Rate(学习率)0.01Epochs(迭代轮次)300Batch Size(批量)32Momentum(梯度动量)0.937Image Size(图片尺寸)640640Weight Decay(权重衰减度)0.000 5计算公式如下:P=TPTP+FP(1)召回率是衡量模型识别出的真实阳性样本数占所有真实样本数的比例。召回率越高,说明模型能够更准确地识别出真实阳性样本,具有更好的鉴别能力。计算公式如下:R=TPTP+FN(2)AP 的定义为 P-R 曲线下面的面积,AP 值越高,模型的性能越好,mAP 为各类别 AP 的平均值,可以通过下列公式得到:AP=n-1i=1(ri+1-ri)p(ri+1)(3)mAP=1mmi=1APi(4)3.4 实验结果与分析 为验证本文所用改进方法的有效性,通过使用相121第 13 期2023 年 7 月无线互联科技研究创新No.13July,2023同的训练参数设置在数据集上进行实验,改进后的YOLOv8 模型检测精度和召回率明显提高,具体的数值结果如表 2 所示,可以看出,通过添加 GAM 注意力机制,模型的性能得到了有效提高。和原模型相比,改进 后 的 YOLOv8 模 型 mAP 0.5 提 高 了 1.7%,mAP 0.5 0.95 提高了 2.2%。结果表明,改进后的YOLOv8 模型更能满足对苹果果实的检测要求。表 2 评价指标对比结果算法PRmAP 0.5mAP 0.5 0.95YOLOv80.9340.810.9410.498改进的 YOLOv80.870.920.9580.524 结语 为解决苹果果实的智能检测问题,提高智能设备对苹果的检测能力,本文使用先进的 YOLOv8 目标检测模型并加以改进,通过使用多种数据增强方法,增加了模型的泛化能力。同时引入 GAM 注意力机制,提升模型聚焦苹果的能力,并提高模型的检测性能。最终的实验结果可知,改进后的 YOLOv8 算法具有较高的检测精度,在 640 640 大小的图像输入下,mAP 0.5 达到了 0.958,比原模型提高了 1.7%,更加适合部署到设备。未来的工作目标是要丰富苹果检测场景,提升模型在复杂环境下的检测效果。参考文献1霍学喜,刘天军,刘军弟,等.2020 年度中国苹果产业发展报告(精简版)J.中国果菜,2022(2):1-6.2孙飒爽,吴倩,谭建昌,等.枝条遮挡下单个苹果目标识别与重建方法的研究J.西北农林科技大学学报(自然科学版),2017(11):138-46.3牟其松.基于深度学习的水果采摘通用检测模型研究D.天津:天津理工大学,2021.4张中华,贾伟宽,邵文静,等.优化 FCOS 网络复杂果园环境下绿色苹果检测模型J.光谱学与光谱分析,2022(2):647-53.5MEILI S,LIANCHENG X,XIUDE C,et al.BFP net:balanced feature pyramid network for small apple detection in complex orchard environment J.Plant Phenomics,2022(1):1-19.6BOCHKOVSKIY A,WANG C Y,LIAO H Y M.YOLOv4:optimal speed and accuracy of object detectionEB/OL.(2020-04-23)2023-05-04.https:/arxiv.org/abs/2004.10934.7LIU Y C,SHAO Z R,HOFFMANN N.Global attention mechanism:retain information to enhance channel-spatial interactionsEB/OL.(2021-12-10)2023-05-04.https:/arxiv.org/abs/2112.05561.(编辑 沈 强)Apple detection method based on improved YOLOv8Du Baoxia1 2 Tang You1 2 Xin Peng1 Yang Mu3 1.Jilin Institute of Chemical Technology Jilin 132022 China 2.Jilin Agricultural Science and Technology University Jilin 132101 China 3.FAW-Tokico Shock Absorber Co.Ltd.Changchun 130001 China Abstract The implementation of automatic apple detection is of great significance for promoting the automation and intelligence of apple production management.In order to solve the problems of low accuracy and poor robustness of current apple detection methods the article proposes an apple detection model based on the improved YOLOv8 algorithm which improves to the detection accuracy of the model by introducing the GAM attention mechanism.Verified on the apple detection dataset based on the improved YOLOv8 algorithm and the original YOLOv8 algorithm mAP 0.5 improved by 1.7%and mAP 0.50.95 improved by 2.2%.The experimental results show that the improved YOLOv8 algorithm proposed in this paper can better meet the requirements of apple detection in practical situations.Key words apple detection YOLOv8 attention mechanism221