基于改进YOLOv4算法的番茄叶部病害识别方法

2023-09-19 19:30储鑫李祥罗斌王晓冬黄硕
江苏农业学报 2023年5期
关键词:注意力机制轻量化病害

储鑫 李祥 罗斌 王晓冬 黄硕

摘要:为快速准确识别自然环境下的番茄叶片病害,提出一种基于改进YOLOv4算法的轻量化番茄叶部病害识别方法。该方法根据番茄病害特征采用K均值聚类算法调整先验框的维度,并使用宽度因子为0.25的MobileNetv1代替YOLOv4原有的主干网络CSPDarknet53进行特征提取,并在特征融合网络PANet中引入深度可分离卷积代替原有的3×3标准卷积,同时在主干网络的2个输出特征层和空间金字塔池化输出层分别嵌入卷积块注意力模块(CBAM),提高模型识别精度。试验结果表明,改进后的模型对8类番茄叶片整体检测精准性(mAP)为98.76%,参数量为12.64 M,传输帧数为1 s 101.76帧,相较于原YOLOv4模型,模型参数量减少80%,每秒传输帧数比原始YOLOv4模型提高了130%。

关键词:YOLOv4;MobileNet;轻量化;注意力机制;病害

中图分类号:S436.412文献标识码:A文章编号:1000-4440(2023)05-1199-10

Identification method of tomato leaf diseases based on improved YOLOv4 algorithmCHU Xin LI Xiang LUO Bin WANG Xiao-dong HUANG Shuo

(1.College of Information Engineering, East China University of Technology, Nanchang 330013, China;2.National Engineering Technology Research Center for Agricultural Informatization, Beijing 100097, China;3.Research Center of Intelligent Equipment Technology, Beijing Academy of Agriculture and Forestry, Beijing 100097, China)

Abstract:In order to identify tomato leaf diseases in natural environment quickly and accurately, a lightweight tomato leaf disease identification method based on improved YOLOv4 algorithm was proposed. The method used K-means clustering algorithm to adjust the dimensions of the prior box according to the characteristics of tomato disease, and used MobileNetv1 with a width factor of 0.25 instead of the original backbone network CSPDarknet53 of YOLOv4 for feature extraction, and introduced deep separable convolution in place of the original 3×3 standard convolution in the feature fusion network PANet. At the same time, the convolutional block attention module (CBAM) was embedded in the two output feature layers and the spatial pyramid pooling output layer of the backbone network to improve the model recognition accuracy. The results showed that, the overall detection accuracy (mAP) of the improved model for eight types of tomato leaves was 98.76%, the parameter quantity was 12.64 M, and the transmission frame number was 101.76 f/s, which was 80% lower than that of the original YOLOv4 model, and the number of transmitted frames per second was 130% higher than that of the original YOLOv4 model.

Key words:YOLOv4;MobileNet;light weight;attentional mechanism;diseases

番茄是重要的經济作物之一,其营养价值和经济价值非常高。随着番茄植株的生长,大多数番茄叶部病害的发生也呈上升趋势[1]。因此,及时防治番茄叶部病害是提高番茄质量和产量的关键举措,而精准识别番茄叶部病害种类是番茄病虫害精准防控的前提。现阶段,中国大部分地区主要还是依赖人工排查番茄叶部病害,这不仅耗时耗力,而且仅仅通过肉眼分辨叶部病害种类,难免产生错误,从而不利于病害的及时防治[2]。对此,需要研究高效率、低成本的病害图像智能识别算法,为番茄叶部病害精准识别提供技术支撑。

随着深度学习的不断发展和其在农业方面越来越多的应用[3],有学者针对作物叶片病害检测识别进行了深入研究。刘文波等[4]将改进后的SOLOv2算法应用于番茄叶部病害检测,平均精度达42.3%,与其他文献的模型相比,精度不太理想。文斌等[5]在YOLOv3的基础上采用注意力特征金字塔和双瓶颈层构建新的网络,实现对三七叶片病害的识别,但检测速度仅为23 fps。Hu等[6]提出将多卷积神经网络集成用于茶树叶部病害的识别,结果显示,该算法可以有效区分病叶与健康叶,但检测类别较少。Qi 等[7]在YOLOv5模型中增加了1个挤压-激发( SE )模块,构建新的模型,实现对番茄病害的识别,平均检测精度为91.07%,但模型参数量较大,高达42.796 M。刘延鑫等[8]构建5种烟草叶部病害数据集,选用YOLOv3模型对烟草叶部病害进行识别检测,采用K-means++聚类方法对锚框(Anchor box)的宽和高进行计算,提高了模型检测精度,得到整体检测精准性(mAP)为0.77,算法结果优于SSD(单步多框目标检测)算法检测的mAP值(0.69),但模型单样本耗时比SSD算法多2 s。王超学等[9]分别以一张葡萄叶片和一串葡萄果实为单位做标注,将YOLOv3算法中的主干网络DarkNet-53替换为EfficientNet-B3作为模型的主干特征提取网络,对12种葡萄病害、健康葡萄叶片和健康葡萄果实进行识别,识别精度为98.60%,B3网络结构模型与其他以EfficientNet作为主干特征提取网络的模型训练时间和模型体积相近,且相对于YOLOv3网络模型,参数量缩减50%,模型体积缩减2/3的同时,模型平均识别精度提升3%,最终参数量仍然较大,约为29 M。周维等[10]利用GhostNet网络代替YOLOv4中主干网络CSPDarkNet-53,并以深度可分离卷积代替普通卷积的方式改进路径聚合网络(PANet),对网上获取的6种水稻病虫害数据集进行检测识别,识别准确率为79.36%。王权顺等[11]利用PlantVillage中苹果叶部黑星病图片与网上另外3种苹果叶部病害图片构建数据集,通过将YOLOv4中主干网络CSPdarknet53替换为DenseNet121网络,在保证特征提取能力的同时减少计算量,针对YOLOv4的预设锚框不适合应用于病斑检测的情况,采用二分K均值聚类算法得到一组新的锚框优化算法,相较于原YOLOv4算法mAP值提高0.89%,模型权重大小为62.71 MB,相较于其他文献报道,该算法识别病害种类较少,仅有4种。

考虑到上述国内外学者对作物叶片病害检测识别的研究中存在的一些问题,例如,精确度低、检测速度慢、模型参数大等,本研究针对番茄叶片多类病害病斑差别较小、病斑形状复杂且无规律、检测识别方法的检测速率低和模型参数大等问题,提出一种改进的YOLOv4轻量化番茄叶部病害识别方法,使用K均值聚类算法得到9个新的先验框,同时利用迁移学习,将预训练YOLOv4算法得到的最优参数与在COCO数据集训练得到的MobileNetv1权重进行融合,并对主干网络、路径聚合网络进行改进,达到降低模型参数和减小计算量的目的。在主干网络的2个输出特征层和空间金字塔池化输出层分别嵌入坐標注意力机制CBAM(卷积块注意力模块),以此避免改进网络后减少模型参数带来的精度损失。

1YOLOv4模型改进

1.1YOLOv4模型

YOLOv4[12-13]作为YOLO系列第4代单阶段目标检测算法,相较于YOLOv3算法,其在检测精度与检测速度方面均有较大的提升。YOLOv4模型主要包括4个方面:输入端、主干网络(Backbone)、颈部网络(Neck)以及头部网络(Head)。YOLOv4在输入端采用Mosaic数据增强[14],丰富了数据的多样性,提高了模型的泛化能力。主干网络在DarkNet53的基础上融合了跨阶段局部网络的思想,形成特征提取效果更强的CSPDarkNet53结构。CSPDarkNet53主干网络主要由5层残差网络Resblock_Body组成,同时通过下采样输出3个有效特征层,以此保证提取更完备的图像语义信息。颈部网络主要由空间金字塔池化(Space pyramid pool,SPP)[15]和PANet[16]组成,空间金字塔池化通过5个大小为5×5、9×9和13×13的最大池化层来融合尺度大小不同的特征图,起到增加感受野的作用。路径聚合网络可以提高信息流在网络中的传递效率[17],其对特征图进行上采样与下采样,借此融合网络提取到的特征信息。最后YOLOv4生成3个检测头用于检测目标[18]。YOLOv4网络结构如图1所示。为提高模型检测效率,降低模型参数量,本研究提出一种改进的YOLOv4算法的番茄叶部病害识别方法。

1.2改进YOLOv4模型

1.2.1运用迁移学习在原先的YOLOv4模型中,输入图片尺寸大小为416×416,冻结层与非冻结层批次大小(Batch size)均为16,模型检测番茄叶片的mAP为98.16%。由于本研究采用的番茄叶部病害数据集尺寸大小为256×256,因此,在模型输入端将尺寸改为256×256,并将冻结层与非冻结层batch size均改为8,最终得到模型检测番茄叶片的mAP为98.58%。因此,在下文其他模型训练时均采用上述参数进行训练。由于COCO数据集包含的数据特征丰富,因此,本研究用到的所有模型的训练权重均为对应模型在COCO数据集训练所得,利用这些训练权重有利于得到更高精度的结果。由此可见,运用迁移学习有利于检测精度的提升,迁移学习改进后参数量由64.36 M降低为63.79 M,图2为参数改进前后mAP的对比。

1.2.2使用MobileNetv1为主干网络MobileNetv1是2017年由谷歌公司提出的一种轻量化深层神经网络,与传统的卷积神经网络模型相比,该模型需要的计算工作量相对较少[19],主要是用于嵌入各类硬件和软件设备,其最大优势是提出了深度可分离卷积(Depthwise separable convolution),并且MobileNetv1采用ReLU6代替ReLU作为激活函数。图3为普通卷积和深度可分离卷积结构的对比。采用多个ReLU6激活函数,可以在减少计算量的同时避免带来精度损失。

为了使模型参数量更小,检测速度更快,MobileNetv1在输入通道上引入可以使每一层网络均匀细化的宽度因子(a),其取值为0.25、0.50、0.75、1.00。当a取1.00时,代表原始的网络,只有当a<1.00时,才有细化网络层的功能。本研究选取a为0.25的MobileNetv1网络作为主干网络对YOLOv4模型进一步压缩,在只给精度仅带来0.01%的损失的同时减少了23.23 M参数量,因此,选取a为0.25的MobileNetv1网络代替YOLOv4的主干网络CSPDarkNet-53,大大地减少了模型参数量且几乎不带来识别精度的损失。

1.2.3改进PANet深度可分离卷积的核心思想是将普通卷积运算分为逐深度卷积(Depthwise convolution)与逐点1×1卷积(Pointwise convolution)2步进行[20]。逐深度卷积不同于普通卷积的是1个通道由1个卷积核负责,每个通道独立进行空间卷积运算。不同深度卷积的输出由逐点1×1卷积重新组合输出,从而得到新的特征图。图 4A为普通卷积运算,图4B为逐深度卷积运算,图4C为逐点卷积运算。

1.2.4改进Anchor boxYOLOv4是一种根据锚框对目标物体进行预测回归的算法,其原先Anchor box是根据COCO数据集训练得到的固定数值。而本研究的对象是COCO数据集中鲜少出现的番茄叶片病害,并且数据集中番茄叶片病害种类繁多,导致数据集标注边界框尺寸相差甚远。为了使Anchor box更接近数据集标注尺寸,本研究采用K均值聚类算法对本研究数据集的Anchor box进行聚类分析,得到合适的锚框尺寸。K均值聚类算法具有计算复杂度低、收敛速度快和可解释性强等特点。根据K均值聚类算法随机初始化得到9个中心点,计算每个坐标到中心点的欧式距离,将每个样本分配给离它最近的中心点,不断重复此步骤直到9个中心点不再移动,从而得到9个新的Anchor box,分别为(132,198)、(162,222)、(182,200)、(186,234)、(203,224)、(226,212)、(209,240)、(223,237)、(237,242),将新的Anchor box运用到算法中,并采用mAP值和GFLOPs值(每秒浮点运算次数)作为优化前后的评价指标。最终结果表明,优化后的mAP值增长了0.02%,GFLOPs值也比优化前高出1.36 G。图5是K均值聚类分布。

1.2.5添加CBAM注意力机制CBAM模块是一种在通道和空间2个维度上对特征图像进行权重分配,从而提取到更有用的特征信息的模块。特征图像首先进入通道注意力模块,通过平均池化和最大池化得到各个通道的信息,并将得到的参数通过多层感知进行叠加,再经过激活函数处理,从而得到通道注意力特征图[21]。空间注意力模块主要关注空间层面上特征图的内在联系,将通道注意力模块的输出作为空间注意力模块的输入,经过卷积和激活操作后获得空间注意力模块的特征图[22]。

为提取到番茄病害更丰富的有效特征信息,本研究在YOLOv4改进后的主干特征提取網络MobileNetv1的2个输出特征层以及加强特征提取网络SPP的输出特征层上添加CBAM模块,以CBAM模块输出的有效特征图作为PANet的输入,传入PANet结构中进行卷积和上采样,最终提取到包含更多关键信息的特征图。改进后YOLOv4的网络结构如图6所示。

2结果与分析

为验证本研究算法在多类番茄病害识别与分类方面的性能,首先构建数据集,然后分别对YOLOv4不同主干网络和在主干网络输出层加入的不同注意力机制模块进行训练,并选取经典的单阶段目标检测算法与本研究改进的YOLOv4算法进行对比。试验的硬件配置为Intel CoreTMi7-10700K处理器和NVIDIA GeForce RTX 3060显卡。操作系统是Ubuntu 20.04.2 LTS 64位系统。Python的版本为3.8.5,Pytorch的版本为1.8.1。在训练过程中,首先对输入的图片进行Mosaic(马赛克)数据增强,并使用余弦退火算法对学习率进行调整, 每轮共设置300次迭代,冻结层和非冻结层batch size设置为8。

2.1数据集的构建

本研究所用数据集选自Plant Village公开数据集中的番茄叶片集,其中包括细菌性斑点病、叶霉病、晚疫病、叶斑病、二斑叶螨病、黄曲叶病、早疫病叶片以及健康叶片,共8个类别。由于样本分布不均匀会影响模型训练结果,因此,每个类别的叶片选取相同的数量,均为1 250张。数据集中早疫病图片仅有1 000张,通过编写程序将其中250张图片旋转270°后与原有图像整合,使其与其他类别图像达到相同数量,最终得到所有数据集10 000张,数据集如图7所示。使用Make Sense工具对上述10 000张番茄叶片病害图像进行标注,标注时利用最小外接矩形框标记出叶片位置并选择对应的类别标签。标注完成后,每张图片会产生1个对应的包含病害类别标签和矩形框坐标信息的txt文件,图8为数据集标注示例。通过编写随机抽样的程序,从10 000张图片中随机选择8 001张作为训练集,随机选择999张图片作为验证集,随机选择1 000张图片作为测试集。

2.2评价指标

为了评估算法的性能优劣,本研究以对所有类别的番茄叶片整体检测精准性(mAP)、 每秒所执行的浮点运算次数、每秒帧数(FPS)、平均每张图片检测时间以及模型参数量作为评价模型大小、检测速度和精度的指标。mAP由所有类别的平均精度(AP)取平均值得到,AP是精确度(P)和召回率(R)组成的曲线的面积。P、R的计算方法分别见公式(5)、公式(6),mAP的计算公式如公式(7)所示。GFLOPs用于衡量模型的计算量,其值越大,表明模型计算速度越高。FPS用于衡量模型的实时性能,其值越大,表明模型实时检测的性能越好。

2.3不同主干网络的性能比较

针对YOLOv4参数量多和网络结构复杂等问题,分别选取不同的轻量化主干网络代替原来的CSPDarknet网络,对8 001张训练集进行训练后,得到相应权重再分别检测验证集和测试集。为验证MobileNetv1轻量化网络检测番茄病害的优越性,选取的网络分别是MobileNetv2、MobileNetv3以及CSPDarknet,预测结果如表1所示。

由表1可以看出,MobileNetv2和MobileNetv3虽然检测时间优于原YOLOv4模型,但识别精度(整体检测精准性)却低于原YOLOv4模型。MobileNetv1不仅比其他轻量化网络识别模型的整体检测精准性高,检测速度也具有较大的优势,检测每张图片平均仅需9 ms,比原YOLOv4模型快了13 ms,满足实时检测的条件。

2.4不同注意力机制的比较

为验证本研究在主干网络2个输出层和SPP结构的输出层添加CBAM模块的有效性,本研究设计了在同等条件下添加不同注意力模块的对比试验,结果如表2所示。由表2可以看出,在设定参数以及其他改进均相同的情况下,添加CBAM模块比添加CA模块和ECA模块的平均检测精度都具有优势,并且没有带来检测速度方面的损失。原因可能是CBAM模块属于轻量级模块,其插入网络层后并不会给网络带来运行负担,并且相较于ECA模块与CA模块只关注通道注意力,CBAM模块可以在关注通道注意力的同时又关注空间注意力,而空间注意力对图像中的有效语义信息更为敏感,因此,添加CBAM模块使得空间注意力与通道注意力相辅相成,从而有效地提高网络的特征提取能力。

2.5消融试验

为验证本研究提出的每个改进对试验结果的积极影响,本研究设计了消融对比试验,试验内容包括运用迁移学习,以MobileNetv1为主干网络,改进PANet,采用K均值聚类Anchor box和添加CBAM模块,试验结果如表3所示。由表3可知,采用参数迁移学习后,番茄病害平均检测精度有所提高,而在采用MobileNetv1替换主干网络后,模型参数量减少了23.23 M,在此基础上,进一步对PANet进行改进,使得模型参数量大幅度减少了28.26 M。由此可以看出,对YOLOv4算法主干网络和路径聚合网络的改进,有效地减少了模型参数量。并且在网络中添加CBAM模块后,使得mAP有所提升的同时参数量仅有微小提升。因此,改进后的YOLOv4算法在保障精度有所提升的前提下,大大提高了模型检测病害的速度。

2.6与其他单阶段目标检测算法比较

当前的目标检测算法分为2类,一类是以Faster R-CNN[23]为代表的Two-stage检测算法,Two-stage检测算法识别精度较高,但对检测速度有所限制,显然不适用于实时检测场景。另一类是One-stage检测算法,这类算法直接使用网络检测目标边界框和类概率分数,检测速度高,识别准确率较高,适用于农业病害实时检测场景。因此,本研究选择对以SSD[24]和YOLO为代表的One-stage目标检测算法与改进的YOLOv4轻量化算法进行对比,表4为不同单阶段算法对比。由表4可以看出,本研究提出的算法平均每张图片检测速度分别比SSD和RetinaNet算法快6 ms和5 ms,mAP比SSD和RetinaNet算法高。本研究提出的算法模型参数量更是远远小于SSD、RetinaNet以及原YOLOv4模型,分别比SSD、RetinaNet以及原YOLOv4模型小9.64 M、9.77 M和51.33 M。因此,本研究提出的算法在不影响识别精度的前提下大大减少了模型大小,更适合在嵌入式系统上部署。

为验证改进后的YOLOv4算法是否可以较好地应用于自然场景,选择现实场景里番茄相对感染较多的4种病害,分别是早疫病、晚疫病、细菌性斑点病和黄曲叶病,使用改进后的YOLOv4算法对4种病害进行检测识别。图9为使用改进后的YOLOv4算法对病害识别的结果,可以看出该算法可以准确识别自然环境下的番茄叶部病害。

3结论

针对不同种类番茄叶片病害误检率高、实时检测速度低等问题[25-26],本研究提出一种改进的YOLOv4轻量化检测算法,应用于8种番茄病害识别,改进后的模型在平均识别精确度、检测速度以及参数内存大小方面均取得了良好的效果。将YOLOv4的主干网络CSPDarkNet替换成轻量化的MobileNetv1网络,并采用深度可分离卷积取代PANet加强特征提取网络中的3×3普通卷积,使得模型的计算量得到进一步压缩。通过有效调整YOLOv4原始模型参数,结合迁移学习策略,将MobileNetv1在COCO数据集上训练的权重运用到改进的YOLOv4模型,并在主干网络的2个输出特征层和空间金字塔池化输出层分别嵌入坐标注意力机制,以此避免减小模型参数带来精度损失。应用K均值聚类算法对Anchor box进行重新优化,得到新的Anchor box值,取代原始YOLOv4的盲目搜索方法,从而提升了模型检测能力。本研究提出的模型传输帧数为101.76 f/s,比原始YOLOv4模型提高了130%,而模型参数量仅为12.64 M,比原YOLOv4模型参数量减少了80%,在1 000张测试集图片上整体检测精准性为98.76%,比原始YOLOv4模型提高了0.6個百分点,平均每张病害图片检测时间为9 ms。因此,改进后的轻量化YOLOv4模型实时检测性能更好,精确度更高并且更适合在硬件上部署。下一步的研究重点是将本研究算法应用到各种复杂的自然场景,将深度学习应用到更广泛的农业工程中。

参考文献:

[1]WANG X W,LIU J, ZHU X N. Early real-time detection algorithm of tomato diseases and pests in the natural environment[J]. Plant Methods,2021,17(1):1-17.

[2]XU C,DING J Q,QIAO Y,et al. Tomato disease and pest diagnosis method based on the Stacking of prescription data[J]. Computers and Electronics in Agriculture,2022,197:106997.

[3]吕盛坪,李灯辉,冼荣亨. 深度学习在我国农业中的应用研究现状[J].计算机工程与应用,2019,55(20):24-33,51.

[4]刘文波,叶涛,李颀. 基于改进SOLO v2的番茄叶部病害检测方法[J].农业机械学报,2021,52(8):213-220.

[5]文斌,曹仁轩,杨启良,等. 改进YOLOv3算法检测三七叶片病害[J].农业工程学报,2022,38(3):164-172.

[6]HU G S,YANG X W,ZHANG Y,et al. Identification of tea leaf diseases by using an improved deep convolutional neural network[J]. Sustainable Computing: Informatics and Systems,2019,24: 100353.

[7]QI J T,LIU X N,LIU K,et al. An improved YOLOv5 model based on visual attention mechanism: application to recognition of tomato virus disease[J]. Computers and Electronics in Agriculture,2022,194:106780.

[8]刘延鑫,王俊峰,杜传印,等. 基于YOLOv3的多类烟草叶部病害检测研究[J].中国烟草科学,2022,43(2):94-100.

[9]王超学,祁昕,马罡,等. 基于YOLOv3的葡萄病害人工智能识别系统[J].植物保护,2022,48(6):278-288.

[10]周维,牛永真,王亚炜,等. 基于改进的YOLOv4-GhostNet水稻病虫害识别方法[J].江苏农业学报,2022,38(3):685-695.

[11]王权顺,吕蕾,黄德丰,等. 基于改进YOLOv4算法的苹果叶部病害缺陷检测研究[J].中国农机化学报,2022,43(11):182-187.

[12]REDMON J,DIVVALA S,GIRSHICK R,et al. You only look once:unified,real-time object detection[C].LasVegas:IEEE Computer Society,2016.

[13]BOCHKOVSKIY A,WANG C Y ,LIAO H Y. YOLOv4:optimal speed and accuracy of object detection[J/OL].arXiv,2020.

[14]胡文骏,杨莉琼,肖宇峰,等.识别安全帽佩戴的轻量化网络模型[J/OL].计算机工程与应用:1-9.[2023-03-15].http://kns.cnki.net/kcms/detail/11.2127.TP.20220524.1106.011.html.

[15]HE K ,ZHANG X , REN S ,et al. Spatial pyramid pooling in deep convolutional networks for visual recognition[J]. IEEE transactions on pattern analysis and machine intelligence, 2015, 37(9):1904-1916.

[16]LIU S, QI L, QIN H F, et al. Path aggregation network for instance segmentation[C]. Salt Lake City,UT,USA:IEEE Press, 2018: 8759-8768.

[17]陈道怀,汪杭军. 基于改进YOLOv4的林业害虫检测[J].浙江农业学报,2022,34(6):1306-1315.

[18]裴瑞景,王硕,王华英. 基于改进YOLOv4算法的水果识别检测研究[J/OL].激光技术:1-11[2023-03-15].http://kns.cnki.net/kcms/detail/51.1125.TN.20220518.1135.004.html.

[19]SRINIVASU P N,SIVASAI J G,IJAZ M F,et al. Classification of skin disease using deep learning neural networks with MobileNet V2 and LSTM[J]. Sensors,2021,21(8):2852.

[20]WANG H T,LU F Y,TONG X,et al. A model for detecting safety hazards in key electrical sites based on hybrid attention mechanisms and lightweight Mobilenet[J]. Energy Reports,2021,7(S7):716-724.

[21]郝帥,张旭,马旭,等. 基于CBAM-YOLOv5的煤矿输送带异物检测[J].煤炭学报,2022,47(11):4149-4158.

[22]姚齐水,别帅帅,余江鸿. 一种结合改进Inception V2模块和CBAM的轴承故障诊断方法[J].振动工程学报,2022,35(4):949-957.

[23]REN S Q,HE K M,GIRSHICK R, et al. FasterR-CNN: towards real - time object detection with region proposal networks [J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2017,39(6):1137-1149.

[24]LIU W, ANGUELOV D, ERHAN D, et al. SSD: single shot multibox detector [M].  Amsterdam,The Netherland:Proceedings of the European Conference on Computer Vision,2016.

[25]胡政,张艳,尚静,等. 高光谱图像在农作物病害检测识别中的研究进展[J].江苏农业科学,2022,50(8):49-55.

[26]牛学德,高丙朋,南新元,等. 基于改进DenseNet卷积神经网络的番茄叶片病害检测[J].江苏农业学报,2022,38(1):129-134.

(责任编辑:陈海霞)

猜你喜欢
注意力机制轻量化病害
早春养羊需防六大病害
汽车轻量化集成制造专题主编
小麦常见三种病害咋防治
葡萄病害周年防治历
果树休眠期咋防病害
基于深度学习的问题回答技术研究
一种轻量化自卸半挂车结构设计
一种轻量化自卸半挂车结构设计
基于LSTM?Attention神经网络的文本特征提取方法
基于注意力机制的双向LSTM模型在中文商品评论情感分类中的研究