基于改进SPP-x的YOLOv5神经网络水稻叶片病害识别检测.pdf

1 9 0 中国农机化学报2 0 2 3年D O I:1 0.1 3 7 3 3/j.j c a m.i s s n.2 0 9 55 5 5 3.2 0 2 3.0 9.0 2 7杨波,何金平,张立娜.基于改进S P P-x的YO L O v 5神经网络水稻叶片病害识别检测J.中国农机化学报,2 0 2 3,4 4(9):1 9 0-1 9 7Y a n g B o,H e J i n p i n g,Z h a n g L i n a.I d e n t i f i c a t i o n a n d d e t e c t i o n o f r i c e l e a f d i s e a s e s b y YO L O v 5 n e u r a l n e t w o r k b a s e d o n i m p r o v e d S P P-x J.J o u r n a l o f C h i n e s e A g r i c u l t u r a l M e c h a n i z a t i o n,2 0 2 3,4 4(9):1 9 0-1 9 7基于改进S P P-x的Y O L O v 5神经网络水稻叶片病害识别检测*杨波1,何金平1,张立娜2(1.长春财经学院信息工程学院,长春市,1 3 0 1 2 2;2.吉林农业大学信息技术学院,长春市,1 3 0 1 1 8)摘要:针对YO L O v 5模型 在水 稻 病害 叶片 检测 计 算复 杂度 高,计 算速 度 慢的 问题,提 出 一种 基于 改 进S P P-x的YO L O v 5模型水稻病害叶片识别检测方法。首先,将原主干网中S P P模块中3个不同尺寸(55、99、1 31 3)的M a x P o o l层替换为3个尺寸相同的55 M a x P o o l层连接,后面通过11卷积层来调整输出特征维数,再将YO L O v 5网络中优化器替换为A d a m,从而构建新的YO L O v 5网络结构。通过试验比较S G D和A d a m优化器在训练集上的收敛速度,结果 表 明:改 进 后 的S P P-x模 块 在 运 算 时 间 上 仅 是 原S P P的5 0%,计 算 精 度 值 达 到9 7%,mA P_0.5和m A P_0.5:0.9 5两项指标分别收 敛 于0.9 8 3和0.8 2 2。试 验发 现 改进S P P-x的YO L O v 5模 型 单张 图像 检测 速 度0.3 4 s,效果良好,能够有效地辅助水稻病害识别。关键词:水稻叶片病害;卷积神经网络;YO L O v 5;R e s N e t;优化器中图分类号:S 4 3 5.1 1:T P 3 9 1.4 1 文献标识码:A 文章编号:2 0 9 55 5 5 3(2 0 2 3)0 9 0 1 9 00 8收稿日期:2 0 2 2年4月2 7日 修回日期:2 0 2 2年7月1 8日*基金项目:吉林省科技发展计划重点研发项目(2 0 2 1 0 2 0 3 2 1 1 S F)第一作者:杨波,男,1 9 7 6年生,吉林长春人,硕士,教授;研究方向为机器视觉和深度学习。E-m a i l:y a n g b o c c u f e.e d u.c n通讯作者:张立娜,女,1 9 8 2年生,吉林长春人,硕士,副教授;研究方向为大数据和模式识别。E-m a i l:z h a n g l n j l a u.e d u.c nI d e n t i f i c a t i o n a n d d e t e c t i o n o f r i c e l e a f d i s e a s e s b y Y O L O v 5 n e u r a l n e t w o r k b a s e d o n i m p r o v e d S P P-xY a n g B o1,H e J i n p i n g1,Z h a n g L i n a2(1.S c h o o l o f I n f o r m a t i o n E n g i n e e r i n g,C h a n g c h u n U n i v e r s i t y o f F i n a n c e a n d E c o n o m i c s,C h a n g c h u n,1 3 0 1 2 2,C h i n a;2.S c h o o l o f I n f o r m a t i o n T e c h n o l o g y,J i l i n A g r i c u l t u r a l U n i v e r s i t y,C h a n g c h u n,1 3 0 1 1 8,C h i n a)A b s t r a c t:A i m i n g a t t h e p r o b l e m s o f h i g h c o m p u t a t i o n a l c o m p l e x i t y a n d s l o w c o m p u t a t i o n a l s p e e d o f Y O L O v 5 m o d e l i n r i c e d i s e a s e l e a f d e t e c t i o n,a n i m p r o v e d m e t h o d f o r Y O L O v 5 m o d e l i d e n t i f i c a t i o n a n d d e t e c t i o n o f r i c e d i s e a s e l e a f b a s e d o n S P P-x w a s p r o p o s e d.F i r s t l y,t h r e e M a x P o o l l a y e r s o f d i f f e r e n t s i z e s(5 5,9 9,1 3 1 3)i n t h e S P P m o d u l e o f t h e o r i g i n a l B a c k B o n e n e t w o r k w e r e r e p l a c e d w i t h t h r e e 55 M a x P o o l l a y e r s o f t h e s a m e s i z e,a n d t h e n t h e o u t p u t f e a t u r e d i m e n s i o n w a s a d j u s t e d b y a 1 1 c o n v o l u t i o n a l l a y e r.T h e n t h e o p t i m i z e r i n t h e Y O L O v 5 n e t w o r k w a s r e p l a c e d w i t h A d a m.T h u s,a n e w Y O L O v 5 n e t w o r k s t r u c t u r e w a s c o n s t r u c t e d.B y c o m p a r i n g t h e c o n v e r g e n c e r a t e o f S G D a n d A d a m o p t i m i z e r o n t h e t r a i n i n g s e t,t h e r e s u l t s s h o w e d t h a t t h e o p e r a t i o n t i m e o f t h e i m p r o v e d S P P-x m o d u l e w a s o n l y 5 0%o f t h e o r i g i n a l S P P,a n d t h e c a l c u l a t i o n a c c u r a c y r e a c h e d 9 7%.T h e t w o i n d e x e s o f m A P_0.5 a n d m A P_0.5:0.9 5 c o n v e r g e d t o 0.9 8 3 a n d 0.8 2 2,r e s p e c t i v e l y.T h e e x p e r i m e n t f o u n d t h a t t h e d e t e c t i o n s p e e d o f s i n g l e i m a g e o f Y O L O v 5 m o d e l i m p r o v e d S P P-x w a s 0.3 4 s,a n d t h e e f f e c t w a s g o o d,w h i c h c o u l d e f f e c t i v e l y a s s i s t r i c e d i s e a s e r e c o g n i t i o n.K e y w o r d s:r i c e l e a f d i s e a s e s;c o n v o l u t i o n a l n e u r a l n e t w o r k;YO L O v 5;R e s N e t;o p t i m i z e r0 引言水稻是禾本科一年生水生草本植物,原产于中国,是亚洲热带种植最广泛的重要谷物。全球一半以上的人口以稻米为主要食物来源,近年来消费者对优质稻米的需求正在不断增长。但是水稻在种植过程中极易第4 4卷 第9期2 0 2 3年9月中国农机化学报J o u r n a l o f C h i n e s e A g r i c u l t u r a l M e c h a n i z a t i o nV o l.4 4 N o.9S e p.2 0 2 3第9期杨波 等:基于改进S P P-x的YO L O v 5神经网络水稻叶片病害识别检测1 9 1 发生病害,水稻病害是水稻降低产量的主要因素之一13,水稻每年因病害所造成的经济损失巨大4,因此快速识别发现水稻病害类型并提供后续的解决治疗方案极其重要。目前基于深度学习且应用比较广泛的目标检测算法51 5分为两类:(1)双步目标检测算法t w o-s t a g e:如F a s t R-C N N5、F a s t e r-R C N N68、M a s k R-C N N91 0等这些模型都是将目标检测分为两个阶段,首先使用区域候选网络(R P N)提取候选目标区域信息,再利用高质量的分类器进行分类检测,最后使用检测网络完成对候选区域类别和位置的预测与识别,此类算法虽在检测精度上有优势,但是计算机消耗资源大,不利于在嵌入式等设备上进行实时检测。(2)单步目标检测算法o n e-s t a g e:如R-F C N1 1、S S D1 2、YO L O1 3、YO L O v 31 41 5、YO L O v 41 6等此类算法将提取候选区域和进行分类两个任务融合到一个网络中,不使用R P N和预定的边界框来进行寻找目标物体,通过卷积核直接计算卷积网络中特征的类别分数和位置偏差,此类算法在计算效率上有优势,具备更快的检测速度。目前国内在植物分类识别领域的研究已经取得一定 的 进 展,吕 石 磊 等1 7提 出 一 种 基 于 改 进Y O L O v 3-L I T E轻量级神经网络的柑橘识别方法,使用M o b i l e N e t-v 2作为该模型的骨干网络并使用混合训练与迁移学习结合的预训练方式来提高模型的泛化能力。曹跃腾等1 8提出一种基于改进R e s N e t模型的轻量化植物叶片病虫害识别算法S i m p l i f y-R e s N e t,使用55卷积和残差块的瓶颈结构以及模型剪枝来处理训练后的模型,识别准确率为9 2.4 5%。鲍文霞等1 9提出一种基于轻量型残差网络LW-R e s N e t的自然场景水稻害虫识别方法,LW-R e s N e t网络在特征提取模块减少了残差块的数量,在注意力子模块中采用深度可分离卷积,减少了浮点运算量,实现了模型的轻量化。孙俊等2 0提出一种空洞卷积与全局池化相结合的多尺度特征融合卷积神经网络识别模型,通过对卷积层的输入数据进行批归一化处理,加速了网络收敛,进一步缩减特征图数目,并采用全局池化的方法减少特征数,上述研究为本文的水稻病害叶片的识别提供了参考。本文以单步目标检测算法的深度卷积神经网络Y O L O v 5为基础,对Y O L O v 5的结构进行改进。将水稻病害数据集中的病害叶片作为训练和检测的目标,探究改进后的Y O L O v 5模型在实际场景下对不同病害类别叶片的识别率与准确率;将改进后的Y O L O v 5网络结构与R e s N e t-3 4网络模型进行对比试验。1 数据采集与处理本文采用佳能E O S 1 5 0 0 d单镜头反光照相机,针对同一个品种水稻进行数据采集,并对采集的样本进行人工筛选,所采集图像均为自然正常生长状态下的水稻叶片图像。为了满足水稻病害数据的普遍性和多样性,选 取 了 水 稻 常 见 的 三 种 病 害:白 叶 枯 病(B a c t e r i a l l e a f b l i g h t)、褐斑病(B r o w n s p o t)、稻瘟病(R i c e b l a s t)和健康(H e a l t h y)叶片4种对象来构建数据集。原始图像共4 7 6张,图像分辨率为3 0 0像素3 0 0像素、3 0 0像素5 0 0像素。由于水稻病害随着时间的变化,在真实自然环境下其病害叶片的纹理、颜色等特征会呈现出季节性、随机性的特点,因此难以搜集齐完整的病害叶片样本。同时,为了扩大样本数量以提高模型训练的精度并防止卷积神经网络过拟合线性,本文采用H S V颜色处理、改变色温、高斯模糊和椒盐噪声等基于几何变换的数据增强。将样本数据集按1 021比例划分为训练集、验证集和测试集。对白枯叶病样本数据增强后图像如图1所示。经过数据增强的数据集分布情况如表1所示。(a)原图(b)H S V(c)不等比例缩放(d)等比例缩放(e)改变色温(f)高斯模糊(g)高斯噪声(h)椒盐噪声(i)平均池化(j)水平垂直翻转(k)随机矩形遮挡(l)自适应直方图均衡图1 数据增强处理F i g.1 D a t a e n h a n c e m e n t p r o c e s s i n g1 9 2 中国农机化学报2 0 2 3年表1 数据集分布情况T a b.1 D a t a s e t d i s t r i b u t i o n张类型白叶枯病褐斑病稻瘟病健康总共训练集8 8 89 2 47 4 11 8 4 84 4 0 1验证集1 7 61 8 41 4 63 6 88 7 4测试集8 89 27 31 8 44 3 72 基于Y O L O v 5模型的水稻叶片病害识别2.1 Y O L O v 5模型构建YO L O v 3之前的YO L O模型采用K-m e a n s和遗传算法对自定义数据集进行分析,从而获得自定义数据集中目标对象的预设定锚框。而YO L O v 5的预设定框是基于在训练水稻数据集时自动学习的,由于目标框架需要缩放自定义水稻病害叶数据集中原始图片的尺寸,并且数据集中原始图片尺寸小于原YO L O的标准输入尺寸6 4 06 4 0,因此在YO L O v 5设置了自适应锚定框(A u t o L e a r n i n g B o u n d i n g B o x A n c h o r s)。YO L O v 5使用了跨阶段局部网C S P D a r k n e t 5 3作为B a c k b o n e网络,在借鉴了残差网络的思路:在部分层之间设置捷径连接(S h o r t c u t c o n n e c t i o n)的同时,还借鉴了D e n s e N e t的思想:叠加(C o n c a t)前面所有层的特征映射图,通过D e n s e b l o c k发送到后面的网络层,进行密集连接,从而将前层的特征映射图分离出来,实现特征重用,即减少模型参数数目、尺寸和F L O P S数值,还保证了推理速度和准确率,同时还有效缓解反向传播中的梯度消失、梯度爆炸等问题。在Y O L O v 5中使用基于M a s k R-C N N的F P N框架和路径聚合网络(P A N E T)作为N e c k来聚合特征,将原F P N结构改进为一种新增强的自下向上路径的F P N结构,并用3 3卷积来处理前一阶段的特征映射。同时使用自适应特征池化(A d a p t i v e f e a t u r e p o o l i n g)恢复每个候选区域和所有特征层次之间被破坏的信息路径,聚合每个特征层次上的候选区域,从而增强底层特征的传播。2.2 优化函数改进2.2.1 随机梯度下降S G DYO L O v 5默认的优化器(O p t i m i z e r)是随机梯度下降S G D(S t o c h a s t i c G r a d i e n t D e s c e n t),其更新规则:每次更新时对单个样本使用梯度下降更新,一次只进行一个样本的更新。S G D算法一次只进行随机单个样本的迭代更新,使得S G D并不是每次迭代都向着整体最优化方向,准确率会降低。同时S G D还需预先设置一个合适的学习率l r,当预设的l r过小,会导致网络在深层训练时收敛速度过慢;预设的l r过大,会导致在训练过程中优化的幅度跳出函数的范围(最优点),因此选择一个自适应学习算法来解决上述问题。2.2.2 自适应矩估计A d a m自 适 应 矩 估 计A d a m(A d a p t i v e M o m e n t E s t i m a t i o n)2 1作为S G D-M和RM S p r o p的结合体,融合了动量和自适应的优点,是一种计算每个参数的自适应学习率的优化方法。A d a m利用梯度的一阶矩估计和二阶矩估计来动态的调整学习率,来进行修正偏差,其计算过程如下。mt=1mt-1+(1-1)gt(1)vt=2vt-1+(1-2)gt2(2)式中:mt 一阶动量,梯度的第一矩,各个时刻梯度方向的指数移动平均值;vt 二阶动量,梯度第二矩(无中心方差)的估计值,代表所有梯度值的平方和;gt 当前参数的梯度。当mt和vt被初始化为0向量时,mt和vt就会向0偏置,所以需要经过式(3)和式(4)来偏差校正第一和第二阶矩估计值来抵消这些偏差。mt=mt1-1t(3)vt=vt1-2t(4)mt和vt都是偏差校正项(b i a s-c o r r e c t i o n t e r m),它们使用了EMA滑动平滑值的思想,去求得 mt 与gt、vt 与二阶动量 gt2 之间的关系。下面以vt为例。vt=(1-2)ti=1t-i2g2i =gt2(1-2)ti=12t-i+=gt2(1-2)+(5)即:gt2=vt-1-2t(6)在极限的情况下,忽略常数(是一个极小的常数值),可得vtvt1-t2。根据上述公式不断更新参数,得出A d a m参数的梯度更新规则,使得损失函数(目标函数)的各个参数向正确的方向更新合适的大小,同时让更新后的损失函数(目标函数)值不断向全局最小值逼近。t+1=t-vt+mt(7)式中:t 当前待优化的参数;平衡因子;学习率。超参数设定值:一阶动量衰减系数1=0.9;二阶第9期杨波 等:基于改进S P P-x的YO L O v 5神经网络水稻叶片病害识别检测1 9 3 动量衰减系数2=0.9 9 9;设置平衡因子=1 0-8来保持数值稳定。2.3 基于Y O L O v 5模型的优化S P P-x模块的改进策略:原Y O L O v 5中的S P P模块是将5 5、9 9、1 3 1 3三个不同尺寸的M a x P o o l层并行连接,并借鉴了残差网络中的捷径连接思想,然后做进一步融合,在一定程度上解决了目标多尺度问题。项目组将S P P模块三个不同尺寸大小的M a x P o o l层,改成三个尺寸相同的5 5 M a x P o o l层连接,最后借鉴卷积映射思想,连接一个11卷积层,调整输出尺寸大小,改进前后模块如图2所示。替换S P P-x模块后完整Y O L O v 5网络结构图如图3所示。优化后的S P P-x模块中两个55 M a x P o o l层与原来一个99 M a x P o o l层计算输出结果相同,三个连接的55 M a x P o o l层与原来一个1 31 3 M a x P o o l层计算输出结果相同。为保证试验公平以及验证试验结果的准确性,在试验中将S P P-x模块最后的11卷积层删除,只对比了只含有M a x P o o l层改进前后的两个模块。试验数据如表2所示。表2试验结果表明,原S P P模块的运算时间为0.4 8 3 6 s,S P P-x模块运算时间为0.2 2 6 4 s。S P P-x模块的平均运算时间减少了5 0%,计算速度比原模块更快。(a)原S P P模块结构(b)S P P-x模块结构图图2 S P P和S P P-x模块结构图F i g.2 S P P a n d S P P-x m o d u l e s t r u c t u r e d i a g r a m图3 Y O L O v 5整体网络结构F i g.3 YO L O v 5 o v e r a l l n e t w o r k s t r u c t u r e表2 模块参数设置与运算时间T a b.2 M o d u l e p a r a m e t e r s e t t i n g a n d c a l c u l a t i o n t i m e类别M a x P o o l模块尺寸数量/个运算时间/sS P P5519910.4 8 3 6S P P-x1 31 315530.2 2 6 43 R e s N e t-3 4模型3.1 映射机理传统优化深度卷积神经网络的思路几乎都是根据卷积计算公式原理来不断叠加网络层数,随着网络深度的增加,当深层次的网络开始收敛时,训练时产生的误差会使得:当平均精度mA P达到饱和后,会迅速下降,使得学习的方向偏离最优方向,并没有浅层的网络模型训练效果好。因此在不引入额外参数,也不引入计算复杂度的条件下,文献2 2 提出了直接映射和卷积映射2种映射方法,在理论上使得模型在训练时学习到的结果不会比原结果差,并且会优先训练浅层的网络模型,再逐步训练深层的网络模型。(1)直接映射:当输入尺寸和输出特征参数、尺寸相同时,可以直接进行相加映射。(2)卷积映射:当输入输出不同时,为满足输出特征个数、尺寸与输入相同,加上一个11卷积来增加维度,以匹配尺寸,使得输入输出一致。1 9 4 中国农机化学报2 0 2 3年3.2 卷积核参数改进为降低R e s N e t网络模型参数数目,对残差块进行优化,即将两个3 3串联的卷积层瓶颈结构替换成1 1、3 3、1 1串联的卷积层。在卷积核参数改进的基础上,结合直接映射和卷积映射构建出新的卷积模块,如图4所示。新结构中3 3卷积层首先通过第一层1 1卷积把2 5 6维c h a n n e l降到6 4维,然后在下一层的11卷积做出还原,从而保证精度。对比两种卷积核整体参数数目可知:修改前为332 5 62 5 6+332 5 62 5 6=1 1 7 9 6 4 8;修改后为1 1 2 5 6 6 4+3 3 6 4 6 4+1 1 6 4 2 5 6=6 9 6 3 2,数量相差1 6.9 4倍。(a)修改前模块(b)修改后模块图4 卷积核F i g.4 C o n v o l u t i o n k e r n e l m o d u l e基于卷积神经网络的基础架构以及卷积核的改进,通过2种映射方式交替更迭构建出了R e s N e t-3 4模型结构,改进后的模型网络结构图如5所示。图5 改进后的R e s N e t-3 4网络模型结构F i g.5 I m p r o v e d R e s N e t-3 4 n e t w o r k m o d e l4 试验与结果分析4.1 不同优化器性能对比本文分别在YO L O v 5中设置两种优化器S G D和A d a m,并随机选取水稻病害叶片数据集中7 5 4幅图像,在相同的网络层数下训练4 0 0轮e p o c h s=4 0 0。S G D初始学习率l r0=0.0 1;A d a m的l r0=0.0 0 1。为控制模型一致,保持训练数据b a t c h s i z e一致,并且关闭s h u f f l e,其余超参数循环学习率l rf=0.2,学习率动量m o m e n t u m=0.9 3 7等均设置相同,模型其余部分已经尽量保持一致。YO L O v 5的最终损失函数La l l=1Lo b j+2Lc l a+3Ll o c,主要由目标损失Lo b j、分类损失Lc l a和定位损失Ll o c三部分损失加权求和组成,其中Lo b j和Lc l a均采用B C EW i t h L o g i t s l o s s计算,Lo b j计算所有样本的目标损失,Lc l a只计算正样本的分类损失,其计算公式如式(8)所示。Lo b j=Lc l a=1nniyil o g(xi)+(1-yi)l o g(1-(xi)/(a)(8)式中:(a)s i g m o i d函数,(a)=11+e-a;xi、yi 将训 练 样 本 分 割 成 网 格 后 的 坐 标位置。其中定位损失Ll o c采用C I o U l o s s计算,只计算正样本的定位损失,计算公式如式(9)所示。Ll o c=C I o U_l o s s=1-I o U(Bp r e,Bg t)+2(Bp r e,Bg t)c2+v(9)v=4a r c t a nwg thg t-a r c t a nwp r ehp r e 2(1 0)=v1-I o U(Bp r e,Bg t)+v(1 1)式中:1、2、3 多任务损失函数权重;B、w、h 中心点、宽、高;c 预测框p r e和真实框g t最小闭包区域对角线距离;v 预测框和真实框长宽比例差值的归一化;平衡因子,长由于宽造成的损失和I o U部分造成的损失之间的比值。为了验证两种优化器的性能,项目组在p y t o r c h中的对同一网络YO L O v 5进行试验,并比较了两种模型的交叉熵损失函数和精准率,随着训练层数e p o c h收敛变化的情况,如图6所示。第9期杨波 等:基于改进S P P-x的YO L O v 5神经网络水稻叶片病害识别检测1 9 5 (a)训练集定位损失值(b)训练集分类损失值(c)训练集目标损失值(d)精准率图6 优化器的性能对比图F i g.6 P e r f o r m a n c e c o m p a r i s o n d i a g r a m o f t h e o p t i m i z e r试验用相同数量的超参数来调参S G D和A d a m模型,在本文的水稻病害叶训练集上,A d a m模型的定位损失Ll o c和分类损失Lc l a在第3轮就开始分别就收敛于0.0 7和0.0 2,虽然S G D模型对应的l o s s值收敛于0.0 5和0.0 1均比A d a m模型低,但却是从第1 0轮左右开始收敛;在训练前期S G D模型Ll o c、Lc l a一直单调减,出现了较大坡度的下降趋势,未出现收敛趋势,而A d a m模型的定位和分类l o s s值呈现上下波动,开始出现收敛趋势。虽然S G D在t r a i n集上的l o s s最终收敛值均分别比A d a m低1.5%、0.8%,但是A d a m的收敛速度明显优于S G D。并且从图6(c)可看出A d a m模型目标损失值Lo b j整体呈现收敛趋势,S G D依旧出现了先下降后收敛的现象,同时由图6(d)可看出A d a m模型的精准率最终收敛于9 7%,而S G D模型的精准率收敛于9 5%,A d a m比S G D模型高2%,所以当需要训练较小的自定义水稻病害叶片数据集,A d a m是更合适的选择。4.2 神经网络模型对比模型的P r e c i s i o n和R e c a l l数据结果如图7所示。(a)精准率(b)召回率图7 精准率和召回率F i g.7 P r e c i s i o n a n d R e c a l l试验将水稻叶片病害数据集中4种不同类别共4 4 0 1幅 图 像 一 起 进 行 训 练。通 过 对 比 改 进 后 的YO L O v 5和R e s N e t-3 4神经网络模型,来探究找出最合适进行水稻叶片病害识别的模型,并应用到实践当中。为科学地评估两个的模型的性能,引入精准1 9 6 中国农机化学报2 0 2 3年率P r e c i s i o n、召 回 率R e c a l l作 为 衡 量 改 进 后 的YO L O v 5和R e s N e t-3 4模型性能的主要指标,初始学习 率l r0=0.0 1,a n c h o r=4,采 用A d a m优 化 器(w e i g h t_d e c a y=51 0-5)。经 过4 0 0次 迭 代,当YO L O v 5的3个损失函数取值小幅波满足了阈值条件(I o U0.5),意味着网络模型已经稳定,然后输出稳定的YO L O v 5网络模型结构。对比精准度发现,Y O L O v 5在e p o c h=2 0 0开始收敛,收敛于9 5.6%左右,R e s N e t-3 4在e p o c h=2 4 0时开始收敛,收敛于8 9.8%附近,Y O L O v 5比R e s N e t-3 4模型高5.8%;对比召回率发现,Y O L O v 5和R e s N e t-3 4分别在2 0 0,2 8 0开始收敛,Y O L O v 5最终收敛于9 7.4%附近,而R e s N e t-3 4收敛于8 9.1%左右,Y O L O v 5整体比R e s N e t-3 4模型高8.3%,这表明改进后的Y O L O v 5在性能上得到了一定程度的改善。4.3 Y O L O v 5模型识别检测测试传统目标检测结果P r e c i s i o n和R e c a l l值呈现正相关分布,并且希望两者值越高越好,但在特殊情况下会出现二者呈现负相关的极端情况(P r e c i s i o n值极高,R e c a l l值极低或相反),致使无法判定神经网络的性能。为了更加科学的评价改进后的YO L O v 5模型,本文引进通过在水稻病害叶数据集计算得出的I o U设为0.5时 的 平 均 精 度mA P_0.5和I o U0.5,0.9 5,步长s t r i d e=0.0 5的mA P_0.5:0.9 5作为另外两种衡量改进后的YO L O v 5性能指标,其结果如图8所示。图8 m A P值图F i g.8 m A P o f d a t amA P_0.5和mA P_0.5:0.9 5分别收敛于0.9 8 3、0.8 2 2,二者相差0.1 6 1。主要原因为mA P_0.5在水稻数据 集 中 的4个 类 别 在 计 算 平 均 精 准 度 时,将P r e c i s i o n-R e c a l l曲线的横轴坐标平均分成1 0等份(精度低)和1 0 0等 份(精度 高),即 使 是 在 精 度 提 高 了1 0倍的情况下,mA P_0.5:0.9 5值依旧处于中上阶段,这表明本文的水稻数据集中不良数据很少且改进后的YO L O v 5对水稻数据集训练效果良好。本文将训练好的YO L O v 5模型对测试集中4个样本,标签名称设置为白叶枯病b l i g h t、褐斑病s p o t、稻瘟病b l a s t和健康h e a l t h y,共4 3 7幅图像样本进行识别,每张识别结果图给出类别置信度,单张图片识别速度为0.3 4 s,部分样本识别效果如图9所示。(a)稻瘟病(b)褐斑病(c)白叶枯病(d)健康状态图9 识别效果图F i g.9 R e c o g n i t i o n e f f e c t g r a p h i c s5 结论1)本文提出一种基于改进的YO L O v 5模型,对自然环境下的水稻病害叶片进行识别检测,首先通过将水稻原始图像进行数据增强操作,扩增数据集。然后对YO L O v 5模型中的S P P模块进行改进,并与原模块进行对比试验。同时替换YO L O v 5中的优化函数为A d a m,试验分别比较A d a m和S G D的YO L O v 5在训练集 上 损 失 函 数 的 收 敛 情 况。最 后 将 改 进 后 的YO L O v 5与R e s N e t-3 4模型进行对比试验。2)通过试验发现,改进后的S P P-x运算时间仅为原S P P模块的5 0%。同时,通过比较S G D和A d a m优化器在训练集上3种损失函数 在训练前期的收敛速度和P r e c i s i o n值,表明A d a m的收敛速度更快,且A d a m的精准率最终收敛值9 7%,比S G D优化器高2%。将改进后的YO L O v 5模型和R e s N e t-3 4在相同数据集里训练迭代4 0 0轮比较发现,基于S P P-x的YO L O v 5模型的精准率高5.8%,召回率高8.3%。同时,Y O L O v 5自身的m A P_0.5和m A P_0.5:0.9 5两项指标分别达到了0.9 8 3、0.8 2 2,表明样本数据集不良数据较少,且模型训练效果较好。通过对训练后的YO L O v 5模型进行实际应用发现,检测效果比较理想,能够达到辅助水稻病害识别诊断的目的。第9期杨波 等:基于改进S P P-x的YO L O v 5神经网络水稻叶片病害识别检测1 9 7 参 考 文 献1 温小红,谢明杰,姜健,等.水稻稻瘟病防治方法研究进展J.中国农学通报,2 0 1 3,2 9(3):1 9 0-1 9 5.2 虞玲锦,张国良,丁秀文,等.水稻抗白叶枯病基因及其应用研究进展J.植物生理学报,2 0 1 2,4 8(3):2 2 3-2 3 1.3 伏震,袁兆恩.水稻细菌性褐条病识别与防治J.农技服务,2 0 0 7,2 4(4):7 5.4 丁麟.基于文献计量的水稻三种主要病害研究水平的国际比较与实证分析D.北京:中国农业科学院,2 0 1 3.5 黄华毅,马晓航,扈丽丽,等.F a s t R-C NN深度学习和无人机遥 感相 结合 在 松 材 线 虫 病 监 测 中 的 初 步 应 用 研究J.环境昆虫学报,2 0 2 1,4 3(5):2 2 0-2 2 8.6 朱红春,李旭,孟炀,等.基于F a s t e r R-C NN网络的茶叶嫩芽检测J.农业机械学报,2 0 2 2,5 3(5):2 1 7-2 2 4.Z h u H o n g c h u n,L i X u,M e n g Y a n g.T e a b u d d e t e c t i o n b a s e d o n f a s t e r R-C NN n e t w o r k J.T r a n s a c t i o n s o