基于改进YOLOv4的葡萄叶片病害检测与识别算法.pdf

第 18 期2023 年 9 月无线互联科技Wireless Internet TechnologyNo.18September,2023作者简介:金彬(1995),男,江苏南通人,助教,硕士;研究方向:机器视觉,人工智能。基于改进 YOLOv4 的葡萄叶片病害检测与识别算法金 彬,解祥新(南通理工学院 计算机与信息工程学院,江苏 南通 226002)摘要:为了降低葡萄果园的管理成本,及时发现并预防葡萄病害,文章提出了一种基于改进 YOLOv4模型的葡萄叶片病害检测与识别算法。该算法对传统 YOLOv4 模型进行了改进,针对细粒度、多尺度的葡萄叶片早期疾病检测优化了检测速度和准确性,并应用于真实环境中的实时检测。在检测时间为 18.31 ms 时,该检测模型的平均准确率(mAP)和 F1 得分分别达到 90.4%和 94.8%。总体检测结果表明,当前算法的性能显著优于现有的检测模型,精度提高了 7.8%,F1 分数提高了 6.6%。该模型可作为一种检测复杂现实情景下葡萄叶片病害的有效方法。关键词:实时目标检测;葡萄叶病;卷积神经网络;计算机视觉中图分类号:TP311 文献标志码:A0 引言 植物病虫害是造成生态和农业损失的重要原因,及时发现和预防各种植物病害是提升农场和果园农业收益的关键。当下仍有不少果园采用传统人工排查诊断的方法,效率低下,管理成本高1。近年来,随着计算机视觉在精准农业技术中的进步,机器视觉的病害检测应用大大提高了作物病害检测的效率,为作物产量的提高作出了重要贡献。1 现有模型的应用局限 植物病害的早期识别和预防是作物收获的重要环节,可以有效减少生长障碍,从而最大限度地减少农药的使用,实现无公害作物生产。王权顺等2将传统机器学习方法用于植物和疾病分类和检测。然而,此类方法在实时疾病检测中的性能和速度较低,也无法适应具有不均匀复杂背景的现实生活检测场景,而深度学习可以很好地解决机器视觉在这方面的问题,并且已被应用在诸如作物和水果的分类、图像分割及品种检测中。其中,CNN 模型可以对输入的图像直接提取特征,从而避免了复杂的预处理流程,因此性能比较突出,使用范围较广3。基于 CNN 的目标检测模型大致可以分为两类:一阶检测器和二阶检测器。二阶检测器需要做两件事:分类和定位,也因此检测时间较长。而一阶检测器 YouOnlyLookOnce(YOLO)算法,将目标分类和定位统一为回归问题,通过回归来检测目标,检测速度大幅提升。本文采用改进的 YOLOv4算法进行葡萄叶片的病害检测,具有较高的精度和准确度。葡萄是一种经济价值较高的水果产物,是不少果园经营者的主要经济来源之一。然而,葡萄病害是葡萄生产过程中的一个主要问题,其严重影响了葡萄的产量和质量。常见的病害可以通过葡萄叶片的现象反映出来,在染病初期发现病害并及时采取措施可以有效防治葡萄的多数病害。然而,由于细粒度的多尺度分布、病害与背景颜色纹理的相似性、病害形态的多样性,葡萄叶片病害的早期实时检测仍然具有挑战性。此外,复杂的背景,包括重叠的叶子和土壤、真实环境中光线的变化以及其他一些因素,导致高精度检测葡萄叶片病害的任务十分艰巨4。现有的检测模型无法实现高准确性的同时还具备较高的实时检测速度。2 改进的 YOLO 模型提出 本研究提出一种基于改进 YOLOv4 的葡萄叶片病害检测与识别算法,以解决传统方法存在的问题,同时提高检测的准确性和效率。通过引入 DenseNet块,将 CSPDarkNet53 修改为 Dense-CSPDarkNet53,以改进小目标检测的特征传输和重用。为了优化冗余并降低计算成本,通过修改卷积块来减少网络层数。改进的路径聚合网络(PANet)也被用来保留细粒度的定位信息并增强多尺度语义信息的特征融合。此921第 18 期2023 年 9 月无线互联科技研究创新No.18September,2023外,所提出的模型中空间金字塔池块的集成增强了感受野。更改网络主要激活函数,提高特征学习能力进而提高识别准确率。为了防止训练过程中的过度拟合并提高鲁棒性,采用了数据增强,将数据集进行了扩充5。该模型可以在复杂的农业环境下自动检测不同大小的葡萄叶片出现的不同病害的判别特征。实验结果表明,该算法优于原始 YOLOv4 模型,具有较高的准确率和鲁棒性,可以为葡萄病害的预防和控制提供有力的支持。3 改进的 YOLO 模型建立 YOLOv4 是一种高精度的单阶段目标检测模型,通过生成边界框坐标和每个类对应的概率,将目标检测任务转化为回归问题。在目标检测过程中,输入图像被分为 NN 均匀相等的网格。该模型将生成 B 个预测边界框和相应的置信度。当目标类的真实值的中心落在指定的网格内时,它会检测到特定对象类的目标。每个网格用每个目标类的置信度分数和相应的 C 类条件概率来预测 B 个边界框。置信度得分可以表示为:confidence=pr(object)IoUtruthpred,pr(object)0,1(1)当目标类落入 YOLO 网格内时,pr(object)=1,否则,pr(object)=0。参考和预测边界框之间的重合由下式描述:IoUtruthpred。这里,IoU 是交集与并集的比值,称为交并比。pr(object)的值表示在网格内检测到目标类时边界框预测的准确性。在获得最终边界框之前,通过非极大值抑制算法过滤每个尺度的最佳边界框预测。然而,原始 YOLOv4 模型在检测葡萄叶片的不同病害时,存在病害密集、细粒度、多尺度分布、病区几何形态不规则、病害同时发生等问题6。同一片叶子和复杂的背景,极大地影响了检测精度,导致大量漏检和错误的目标预测。为了解决上述问题,本文提出了基于 YOLOv4 算法的改进版本,以提高葡萄叶片病害的检测效率和准确性。改进的网络模型架构如图1 所示,主干网络用于特征提取,颈部用于提取特征的语义表示,头部用于预测。图 1 改进的 YOLOv4 主干网络 在目标检测过程中,YOLOv4 算法减少了神经网络中的特征映射。为了结合更多的特征信息,在传播过程中采用密集块网络结构实现了特征的保存与前级特征的重用,密集块网络结构使得每一层都以前馈模式连接到其他层。网络块的主要优点是第 n 层能够从所有输入接收所需的特征信息 Xn,可以表示为Xn=HnX0,X1,Xn-1,由于图像数据集的复杂性,密集的块在整个神经网络中促进了更好的特征传输和梯度,也能在一定程度上减轻过拟合。因此,在提出的模型中,用 DB1-CSP1、DB2-CSP2、DB3-CSP4、DB4-CSP4 和 DB5-CSP2 替换原 CSPDarknet53 中CSP1、CSP2、CSP8、CSP8 和 CSP4,通过增加密集的连接块,以增强特征的传播,减少卷积块,以减少冗余特征操作的数量,提高计算速度。所提出的密集块网络结构示意图如图 2 所示。目标检测模型的一个重要方面是为特定问题选择适当的激活函数,以提高神经网络的准确性和性能。为了增强网络梯度流的稳定性,帮助在检测模型中学习更多的表达特征,提出的模型使用了 Mish 激活函数7,可以表示为:f(x)=xtanh(softplus(x)=x031第 18 期2023 年 9 月无线互联科技研究创新No.18September,2023图 2 密集块网络结构tanh(ln(1+ex)。此外,由于 Mish 具有独特的无界性和下有界性,它有助于消除输出神经元的饱和问题,改进网络正则化。经过验证,使用 Mish 作为主要激活函数,在自定义模型数据集中精度有明显的提高。4 验证结果 原始数据集中,共有1 600 张,包括3 种葡萄病害的叶片图片各 400 张,健康葡萄叶片 400 张。利用图像增强程序,将数据集扩展到 4 800 张,作为本研究的自定义数据集。针对自定义数据集,使用 LabelImg脚本进行标注,文件包含了训练数据集中图像标注时 的目标类信息和相应的边界坐标。从自定义数据集中,按照 3 1 1 的比例构建训练集、验证集和测试集。为了提高所提出的检测模型对葡萄不同生长阶段的病害识别准确性,规定输入 512512 大小的 3 通道图像,设定 batch 大小为 16,学习率最低 0.001。为了比较所提出的检测模型的总体性能,本研究将 IoU、F1-score、mAP、最终验证损失和平均检测时间与 YOLOv3 和 YOLOv4 进行了比较,如表 1 所示。对比 IoU,发现该模型的 IoU 值最高,为0.915,比原始YOLOv4 模型高出 6.2%。因此,与其他两种模型相比,所提出的检测模型具有更好的边界框检测精度。该模型的 F1 评分为0.948,mAP 为0.904,比 YOLOv4提高了 6.6%和 7.8%。此外,比较了 3 种模型的平均检测时间,测试结果表明 YOLOv4 的检测时间最低,为 15.721 ms。该模型的检测时间高于 YOLOv4 模型,检测时间为 18.313 ms。表 1 模型之间的 IoU、F1 得分、平均检测精度、Loss、用时与帧率对比ModelIoUF1-scoremAPLoss用时/ms帧率/FPSYOLOv30.7630.7970.75712.1116.34661.12YOLOv40.8530.8820.8264.9715.72163.61New-Model0.9150.9480.9041.8318.31354.61 当 PR 曲线下的面积在所有 3 个模型中最高时,所提出的模型对于特定召回的精度值更高。与YOLOv3 和 YOLOv4 检测模型相比,所提出的模型显著提高了测试数据集的总体精度、查全率和 F1 得分,在精度和准确度上都明显优于 YOLOv3 和 YOLOv4,代价仅是降低少许检测速度。由此,可以得出所提模型的性能和精度得到了显著提高。5 结语 综上所述,本研究基于改进的 YOLOv4 算法开发了一个实时目标检测框架,并将其应用于葡萄叶的病害检测。本研究对该模型进行了改进,以优化其准确性,并通过在复杂的果园情景下检测疾病进行验证。在检测帧率为 54.6FPS 的情况下,该算法的平均精度(mAP)值为 91.5%,F1 分数为 94.8%。与原 YOLOv4模型相比,该模型的精度提高了 7.8%,F1 分数提高了 6.6%,表明在实时现场应用中具有较强的检测性能潜力。参考文献1陈道怀,汪杭军.基于改进 YOLOv4 的林业害虫检测J.浙江农业学报,2022(6):1306-1315.2王权顺,吕蕾,黄德丰,等.基于改进 YOLOv4 算法的苹果叶部病害缺陷检测研究J.中国农机化学报,2022(11):182-187.3傅云龙,梁丹,梁冬泰,等.基于机器视觉与 YOLO算法的马铃薯表面缺陷检测J.机械制造,2021(8):82-87.4雷建云,叶莎,夏梦,等.基于改进 YOLOv4 的葡萄叶片病害检测J.中南民族大学学报(自然科学版),2022(6):712-719.5储鑫,李祥,罗斌,等.基于改进 YOLOv4 算法的番茄叶部病害识别方法J.江苏农业学报,2023(5):1199-1208.6刘阗宇,冯全,杨森.基于卷积神经网络的葡萄叶片病害检测方法J.东北农业大学学报,2018(3):73-83.7邹翔翔.基于深度卷积神经网络的面部表情识别研究D.南京:南京邮电大学,2023.(编辑 王永超)131第 18 期2023 年 9 月无线互联科技研究创新No.18September,2023Grape leaf disease detection and recognition algorithm based on improved YOLOv4Jin Bin Xie Xiangxin Computer and Information Engineering Department Nantong Institute of Technology Nantong 226002 China Abstract In order to reduce the management cost of grape orchards and timely detect and prevent grape diseases this paper proposes a grape leaf disease detection and recognition algorithm based on an improved YOLOv4 model.The traditional YOLOv4 model has been improved to optimize detection speed and accuracy for fine-grained and multi-scale early disease detection of grape leaves and applied to real-time detection in real environments.At a detection time of 18.31 ms the average accuracy mAP and F1-score of the detection model reached 90.4%and 94.8%respectively.The overall detection results indicate that the current algorithm performs significantly better than existing detection models with an accuracy improvement of 7.8%and an F1-score improvement of 6.6%.This model can serve as an effective method for detecting grape leaf diseases in complex real-world scenarios.Key words real-time object detection grape leaf disease convolutional neural network computer vision(上接第 108 页)Power outage research and judgment based on data fusion of multiple heterogeneous modelsYe Ning Jiang Hui Guodian Nanjing Automation Co.Ltd.Nanjing 211100 China Abstract Distribution network outage fault is one of the main factors affecting the reliability of power supply.In order to improve the accuracy and efficiency of fault diagnosis this paper proposes a distribution network outage fault diagnosis method based on data fusion of multiple heterogeneous models.This method integrates different types of data and models and obtains more comprehensive and accurate fault diagnosis results through data cleaning feature extraction data alignment and model fusion.Specifically this paper adopts support vector machine naive Bayes decision tree and other models combined with the actual situation of the field to study and judge the distribution network outage fault.Experimental results show that the proposed method can effectively improve the accuracy and efficiency of fault diagnosis.Key words multivariate heterogeneous model data fusion power failure research and judgment231