基于深度学习的番茄病害检测识别算法
- 1、下载文档前请自行甄别文档内容的完整性,平台不提供额外的编辑、内容补充、找答案等附加服务。
- 2、"仅部分预览"的文档,不可在线预览部分如存在完整性等问题,可反馈申请退款(可完整预览的文档不适用该条件!)。
- 3、如文档侵犯您的权益,请联系客服反馈,我们会尽快为您处理(人工客服工作时间:9:00-18:30)。
引言
农作物生长周期易遭受到病害的侵袭,而我国是农业病害多发重发的国家,农作物每年遭受病害侵蚀经济损失程度巨大,根据联合国农粮组织估计,全世界范围内每年农作物病虫害自然损失率在37%以上。
由此可见,病虫害的防治对我国的农业发展具有至关重要的作用。
病虫害种类繁多,而传统的病虫害检测方法依靠人工凭借经验判断种类,难以准确诊断。
改变传统的病虫害检测方式,能更有效的满足病虫害的防治需求。
近年来,智能检测技术不断应用于农业上。
刘洋等提出一种轻量级 CNN 的植物病害识别方法,并将其移植到 Android 手机端,并在识别精度、运算速度和网络尺寸之间取得不错的效果。
孙俊等通过改进卷积神经网络的初始化参数和尝试不同类型的激活函数对多种植物叶片病虫害进行识别,平均测试识别准确率达到 99.56%。
魏丽冉等在Lab彩色空间模型下的a、b分量上进行叶片分割并提取特征,然后采用基于核函数的支持向量机多分类方法对病害进行检测识别并分类,识别率最高达到89.5%。
赵立新等基于迁移学习的棉花叶部病虫害识别,通过改进 AlexNet模型的迁移学习,源领域学习到的知识迁移到目标领域,数据增强技术能有效缓解过拟合,能取得不错的效果。
Sumita Mishra等将通过预训练的卷积神经网络模型部署到专门的CNN硬件模块中,使玉米叶疾病的识别模型准确率达到了88.46%。
刘君等针对背景环境的要求较高,同时不能实现病虫害的定位的问题,提出基于YOLO卷积神经网络的番茄病虫害检测算法并建立数据库,对8类番茄病虫害的检测平均精度高达 85.09%。
本研究采用的是YOLOv4目标识别算法对番茄病害进行识别与定位,YOLOv4是YOLOv3的改进版,在结合YOLOv3的基础上提出一系列新的方法。
YOLOv4在YOLOv3的基础上,在FPS不下降的情况下,mAP达到了44,提高非常明显。
通过调整原始 YOLOv4 网络参数来适应番茄病害检测任务,研究优化改进模型效果,相对于深度学习病害图像识别来说,YOLOv4目标检测算法对病害的检测效果有较大提升,并能对发病位置进行精确定位,同时还拥有不错的速度和精度。
1.YOLOv4算法结构框架
YOLOv4是在YOLOv3基础上进行了改进,主要改进的地方有将YOLOv3的主干特征提取网络DarkNet53改为CSPDarkNet53,其结构图如表1所示,加入SPP、PAN 特征金字塔,使用mish激活函数代替LeakyReLU,使用CutMix数据增强和马赛克(Mosaic)数据增强。
YOLOv4相比于YOLOv3检测速度更快,精度更高,对小目标检测更敏感。
在FPS一样的情况下YOLOv4比YOLOv3的AP 大概高出了31.25%。
表1 CSPDarkNet网络结构图
Type Filters Size Output
Convolutional323×3256×256
Convolutional643×3/2128×128
CrossStagePartial
Convolutional321×1
1×Convolutional643×3
Residual128×128
Convolutional1283×3/264×64
CrossStagePartial
Convolutional641×1
2×Convolutional643×3
Residual64×64
Convolutional2563×3/232×32
CrossStagePartial
Convolutional1281×1
8×Convolutional1283×3
Residual32×32
Convolutional5123×3/216×16
CrossStagePartial
Convolutional2561×1
8×Convolutional2563×3
Residual16×16
Convolutional10243×3/28×8
CrossStagePartial
Convolutional5121×1
4×Convolutional5123×3
Residual8×8
Avgpool Global
Connected1000
Softmax
CSPDarkNet一共有5大残差块,每个残差块分别包含小的残差块为1、2、8、8、4。
在每组Residual block加
基于深度学习的番茄病害检测识别算法
陶国柱 廖义奎 董力量
(广西民族大学电子信息学院,广西 南宁 530006)
摘 要:本文以番茄病害图像为研究对象,基于深度学习的思想提出了一种基于YOLOv4的番茄病害检测识别算法,通过优化网络模型参数进一步提高模型识别精度,提高模型的鲁棒性,使其能更好的适应番茄病害的识别。
对6类番茄病害的检测平均精度高达 86.23%。
并且拥有不错的检测速度,结果表明,该算法能够有效地提升番茄病害检测的精度和速度,得到番茄病害的精准定位,检测效果优于 Faster R-CNN 和 SSD 等农业病害检测方法。
本研究为农作物病害识别,农药定位喷洒技术的发展及提高番茄重要病症分类和病理检测效率提供参考依据。
关键词:番茄;图像处理;病害检测;YOLOv4;目标检测
文章编号:ISSN2096-0743/2021-05-0141
上一个Cross Stage Partial结构。
并且CSP-DarkNet中也取消了Bottleneck的结构,减少了参数使其更容易训练。
相比于普通的残差卷积块具有加强CNN的学习能力、消除计算瓶颈、降低内存成本的优点。
SPP传统的CNN结构中,由于全连接层的特征数是固定的,所以输入的图片大小是不变的。
而SPP使用多个不同大小的窗口对特征图进行pooling,然后将分别得到的结果进行合并就会得到固定长度的输出。
2.番茄病害检测算法
本实验提出的YOLOv4病害检测算法首先对病害图像进行预处理,图片仍然保持原来的颜色3通道以便保留更多的原始病害信息,将病害图像resize到416×416像素大小尽量保证其病害像素值在合理范围内,然后通过特征提取主干网络对病害进行特征提取,将提取到的特征图进行网格划分。
最后将提取三个大小为52×52、26×26、13×13的特征层进行预测。
不同大小的网格可以感受不同大小的病害信息,13×13的网格用来感受大目标,而52×52的网格感受小目标,能逐步提取更高更丰富的病害语义特征信息,方便检测到不同大小的病害病斑,本实验算法流程如图1。
对于每个相应的网格都有一定数量的anchors,然后微调每个anchors形成预测框。
对于每个预测框模型需要预测输出P0=(t x,t y,t w,t h,P obj,P1,P2,...,P C) 其中P obj是anchors是否包含病害的概率,P1,P2,...,P C则是锚框包含的病害属于每个类别的概率。
tx,ty,tw,th则是预测框需要调整病害发病的位置。
对于一个预测框,网络需要输出(5+C)个实数来表征它是否有病害、发病位置和形状尺寸以及属于那个病害类别的概率。
由于我们在每个anchors都生成了K个预测框,则所有预测框一共需要网络输出的预测值数目是:
[K(5+C)]×m×n (1)
m,n为划分的网格数。
接下来将病害表征输出和预测框关联起来,也就是要建立起损失函数跟网络输出之间的关系。
CIOU将检测物体和每个锚框之间的距离、重叠率、尺度以及惩罚项都考虑进去,使得预测框波动更小,不会像IOU和GIOU一样出现训练过程中发散等问题。
而惩罚因子a,v 把预测框长宽比拟合目标框的长宽比考虑进去。
(2)
其中,分别代表了预测框和真实框的中心点的欧式距离。
c代表的是能够同时包含预测框和真实框的最小闭包区域的对角线距离。
a与v 的公式如下:
(3)
(4)
相应的为:
(5)
总损失函数LOSS为计算xy和wh上的LOSS ciou、计算置信度的LOSS obj、计算预测种类的LOSS class三者之和。
LOSS=LOSS ciou+LOSS obj+LOSS class (6)
LOSS obj表示obj中是否包含病虫害情况预测:
LOSS obj2
00
2
00
)
(
)
(ˆ
1
ˆ
12
2
c c
c c i i
i
k
j
obj
i j
noobj
i
i
i
k
j
obj
i j
s
s
−
+
−
=∑∑
∑∑
==
==
λ
(7)LOSS class表示对病虫害类别的预测:
LOSS class=
2
)(
)(
(
1ˆ
2
∑
∑
∈
=
−
class
c
i
i
i
obj
i j
c
c
s p
p (8)
考虑到在真实自然环境下病害的发生概率并没有那么高,由此可知大多数网格anchors置信度都为0,为了解决正负样本不平衡的问题,将正样本,即有病害的样本IOU阈值设置为0.5,该目的是为了平衡正负样本。
大于这个阈值的样本都会被算为正样本,大于阈值但是置信度为0的样本obj被标记为-1,不会被计算在LOSS内。
因此,加大其中存在病害目标的网格的权重,将权重更改为 λciou=5,λnoobj =0.5。
图1 病害检测算法流程图
3.实验数据
本文从Plantvillge开源数据集中选取不同种病害番茄图像各500张,共计2000张,因农业领域数据图片较少不好获取,且病害的种类发病与天气及种植环境等多种因素相关,通过筛选网上权威网站获取部分由各种植户上传的自然光下拍摄的番茄早疫病415张、晚疫病413张、白粉病426张、细菌性斑点病417张、叶霉病423张、病毒叶435张。
针对数据集中图片相对较少而产生的样本失衡的问题,本文利用随机的裁剪、旋转、色彩变换、抑制噪声等方法对部分数据进行样本增强,将病害数据集扩展到 5574 张,这使得样本多样性增加,提高了样本质量,同
时有效缓解训练过程中的过拟合现象。
按照9:1的比例划分训练集和测试集。
利用 LabelImg 软件进行了病害信息的人工标注,生成包含病虫害目标的种类和位置的 xml 文件,并将其转换成训练使用的 Label 文件,从而为试验做好数据准备。
样本数据如表2所示。
表2 番茄病害数据统计
细菌性斑点病
早疫病 晚疫病 病毒病 叶霉病 白粉病 总计 训练集 841 824 840 935 825 847 5112 测试集 94 92 94 93 92 95 462 总计
935
916
934
928
917
942
5574
4.实验结果分析
本实验系统为win7,CPU 为i7-9400,显卡GPU
Nvidia GTX1080,8GB 内存,程序使用python 编写,调用Keras、OpenCV 等库并在Pycharm 上运行。
4.1评价指标
在深度学习中为了能更准确衡量模型对病害检测的检测性能,本文选用Precision (精准度,P)和Recall (召回率,R)两者结合作为模型的评价指标。
具体计算如式(9)(10)所示,其中:TP 表示分类器认为是病害样本并且分对的例子,即被正确检测到的病害样本数;TN 表示分类器认为是正确样本但分错的例子,即没有框选的病害的部分;FN 是分类器认为不是病害样本且分错了的例子,即没有检测到病害数;FP 是分类器认为不是病害样本但分对了的例子,即检测到了病害的其他区域。
Pr TP
ecision TP FP =
+ (9)
Re TP
call TP FN =
+ (10)
P r e c i s i o n
图2 各算法对病害的总体检测情况
分别取召回率R 和精确率P 的值作为横坐标和纵坐标,绘制一条 P-R 曲线,如图2,精准度 P 表示模型分类器认为是正类并且确实是正类的部分占所有分类器认为是正类的比例;而召回率 R 表示模型分类器认为是正类并且确实是正类的部分占所有确实是正类的比例,可以反映网络模型的整体性能。
本试验的番茄病害检测结果通过Precision 与Recall 曲线来描述。
其中,Precision 和Recall 定义如下。
Precision=正确检测病害样本数/(正确检测病害样本数+错误检测病害样本数)
Recall=正确检测病害样本数/(正确检测病害样本数+没有检测到的病害样本数)
4.2病害检测精度
表3 各算法对病害的平均检测精度 %
病害类型
Faster-Rcnn SSD 本实验算法 a. 番茄白粉病 71.20 79.30 84.80 b. 番茄病毒病 66.80 80.40 86.50 c. 番茄晚疫病
73.80 83.10 88.50 d. 番茄细菌性斑点病 77.20 82.30 84.60 e.
番茄叶霉病
83.00 78.60 82.30 f. 番茄早疫病 75.10 81.20 90.70 平均值
74.52
80.82
86.23
图3 本实验算法对各病害检测效果4.3病害检测速度
表4 各算法对病害检测速度
算法类型检测速度(FPS)
Faster-Rcnn
10SSD 45本实验算法
75
对各算法对6种病害的检测精度统计如表3所示,检测效果如图3所示,与其他算法相比,本算法对番茄白粉病、番茄病毒病、番茄晚疫病、番茄细菌性斑点病、番茄叶霉病、番茄早疫病的检测精度分别为84.8%、86.5%、88.5%、84.6%、82.3%、90.7%。
平均精度达86.23%,在提升病害检测精度的同时也拥有不错的检测速度,FPS 达75,能够应用于实时病害检测中。
5.结语
本文利用 YOLOv4 卷积神经网络实现对番茄病害识别与定位检测,并与两个主流深度学习目标检测算法做了对比,通过试验测试,基于的 YOLOv4 网络的番茄定位识别
模型平均精确度达到 86.23%,相对于Faster-Rcnn、SSD 两种网络,平均精确度分别提高11.71%、5.41% ,本试验方法较前人的方法在病害检测准确率上有较大提升,同时能实现图像中病害位置的精准定位,能够对病害实行实时监测,说明该方法具有一定的实用价值。
参考文献:
[1]刘 洋,冯全王书志.基于轻量级CNN的植物病害识别方法及移动端应用[J].农业工程学报,2019
[2]孙 俊,谭文军,毛罕平,武小红,陈 勇,汪 龙.基
于改进卷积神经网络的多种植物叶片病害识别[J].农业工程学报,2017
[3]魏丽冉,岳 峻,李振波,寇光杰,曲海平.基于核函
数支持向量机的植物叶部病害多分类检测方法[J].农业机械学报,2017
[4]赵立新,侯发东,吕正超,朱慧超,丁筱玲.基于迁
移学习的棉花叶部病虫害图像识别[J].农业工程学报,2020
[5]Sumita Mishra, Rishabh Sachan, Diksha Rajpal. Deep
Convolutional Neural Network based Detection System for Real-time Corn Plant Disease Recognition. 2020 [6]刘 君,王学伟.基于YOLO的番茄病虫害识别算法[J].
中国瓜菜,2020
[7]Bochkovskiy, A., Wang, C. Y., & Liao, H. Y. M.
(2020). YOLOv4: Optimal Speed and Accuracy of Object Detection.arXiv.preprint arXiv:2004
[8]Wang, C. Y., Mark Liao, H. Y., Wu, Y. H.,
Chen, P. Y., Hsieh, J. W., & Yeh, I. H. (2020).
CSPNet: A new backbone that can enhance learning capability of cnn. In Proceedings of the IEEE/CVF
Conference on Computer Vision and Pattern Recognition Workshops (pp. 390-391).
[9]He, K., Zhang, X., Ren, S., & Sun, J. (2015).
Spatial pyramid pooling in deep convolutional networks for visual recognition. IEEE transactions on pattern analysis and machine intelligence, 37(9), 1904-1916.
[10]Zheng, Z., Wang, P., Liu, W., Li, J., Ye,
R., & Ren, D. (2020). Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression. In AAAI (pp. 12993-13000).
[11]王献锋,张传雷,张善文,等. 基于自适应判别深度
置信网络的棉花病虫害预测[J]. 农业工程学报,2018 基金项目:南宁市青秀区科技计划项目(项目编号2018008);中国-东盟研究中心开放课(TD201405);广西创新驱动发展专项资金科技重大专项(桂科AA17202015-5);
作者简介:陶国柱(1997-),男,广西横县人,硕士,研究方向:人工智能、物联网。
通讯作者:廖义奎(1970-),男,硕士,副教授,研究方向:计算机视觉、自动控制。
地方名片,不仅可以避免千篇一律的“流水线”操作,还可以促进当地经济、政治、文化的相互融合、共同繁荣,在保留原本朴素的美感与构造的同时,剔除消极因素,彰显其最大亮点。
3.3坚持普遍性与特殊性相结合,走出美丽庭院建设的特色中发展道路。
坚持矛盾普遍性与特殊性的统一,即坚持矛盾的共性与个性共性与个性相统一,是唯物辩证法的一对重要范畴,是对存在于客观事物和过程中的一种最普遍的联系和关系的抽象。
共性,是一类事物的共同的本质和规律;而个性,则是事物之间相互区别的特殊本质和规律。
坚持普遍性与特殊性相结合,能够使得美丽庭院建设既有一套固定完整的体系,方便美丽庭院建设的扩大与推广,同时具有较高的灵活性与适应性,使取得的成果与经验可以相互转化、相互促进,真正做到物尽其用、人尽其能,高效有序,最终实现“一处美促进整体美,整体美促进到处美”。
如何坚持普遍性与特殊性相结合,关乎基层治理的水平,也关乎居民的参与度与积极性。
统筹规划方针如何制定,领导干部不能独断空想,要深入到居民之中、深入到村庄、街道之中,认真调查、反复研究,以当地特色引领发展,通过建设美丽宜居乡村,大力发展乡村旅游、乡村旅游品牌,促进农村产业转型升级;根据当地风俗习惯、文化传统以及村民的喜好,开展活动,这是把握美丽庭院建设的“特殊个性”;以经验事实为基础,综合分析、科学总结,归纳出美丽庭院建设的基本发展规律与基本路线,这是提炼规律所必须依据的“普遍共性”。
此即共性寓于个性、个性共性相互融合、相互促进的生动平衡。
坚持普遍性与特殊性相统一,统筹规划下保持各地特色,促进美丽庭院建设的发展更加完善。
参考文献:
[1]李 丹.新公共服务理论视角下的美丽乡村建设研
究——以书院镇为例[D].上海.华东政法大学公共管理学院,2019
[2]朱志平,姚科艳,鞠 萍.乡村文化现代转型及其路径
选择——基于马庄经验[J].中国农业大学学报(社会科学版), 2020
基金项目:本文系2020年全国大学生创新创业训练计划项目成果,项目号202010327025Z
作者简介:鹿咏楠(2001—),女,在读本科,江苏徐州,南京财经大学法学院;李骏(1977—),女,汉族,江苏,艺术学博士,副研究员,硕士生导师;现任外国语学院党委书记
(上接第15页)。