基于YOLOv3的多类烟草叶部病害检测研究

合集下载

基于YOLOv3的自然环境下茶叶嫩芽目标检测方法研究

基于YOLOv3的自然环境下茶叶嫩芽目标检测方法研究

本栏目责任编辑:唐一东本期推荐基于YOLOv3的自然环境下茶叶嫩芽目标检测方法研究施莹莹,李祥瑞,孙凡(江苏大学计算机科学与通信工程学院,江苏镇江212013)摘要:以自然环境下手机拍摄的多天气情况、多品种的茶叶嫩芽图像为实验样本数据集,研究茶叶嫩芽目标检测方法。

采用基于深度神经网络的YOLOv3算法构建模型,将经过数据增强处理的多品种茶叶嫩芽图像作为模型输入,通过单一的特征提取网络,进行多尺度目标检测,对输入图像进行粗、中、细的网格划分,将茶叶嫩芽目标检测任务看作目标区域预测和类别预测的回归问题。

实验结果表明,基于YOLOv3的目标检测算法对自然环境下多品种的茶叶嫩芽目标具有较高的召回率及检测精度,为农业机械化、智能化提供了思路。

关键词:目标检测;YOLOv3;茶叶嫩芽识别;深度学习中图分类号:TP3文献标识码:A文章编号:1009-3044(2021)03-0014-03开放科学(资源服务)标识码(OSID ):中国茶文化源远流长,而茶叶作为茶文化的中心,同时也是中国与世界文化沟通的桥梁之一,其种植与采摘自然是不可忽视的方面。

目前,主要的茶叶采摘方式分为人工采摘和机械采摘2种。

一方面,人工采摘虽然能较准确地识别嫩芽,但效率低下、成本高昂,且在采茶洪峰时期务工难[1];另一方面,使用机械采摘代替人工手采虽能提高采摘速度,但采摘时老叶、嫩叶、嫩芽一起采,缺乏选择性,并有部分叶片遭破损现象,降低了原料品质[2]。

因此,研究一种面向自动采茶装置的高效、低损伤的茶叶嫩芽目标检测方法十分必要。

目前,茶叶嫩芽的识别与检测主要依赖于茶叶形状和颜色特征的提取,且大多是基于传统机器学习和数字图像处理技术的。

如杨福增[3]等针对白色背景下的茶叶图像,利用其在RGB 颜色空间中的G 分量,通过双阈值方法进行分割,再根据茶叶嫩芽的形状特征进行边缘检测,展开嫩芽识别的研究。

吴雪梅[2]等提出基于茶叶嫩芽图像在Lab 颜色模型中a 分量、b 分量信息的K -means 聚类法,为智能采摘技术的研究提供了支持。

基于YOLO算法的农作物病虫害识别研究综述

基于YOLO算法的农作物病虫害识别研究综述

2024年2月Feb.2024第48卷第1期Vol.48,No.1热带农业工程TROPICAL AGRICULTURAL ENCINEERING基于YOLO 算法的农作物病虫害识别研究综述万应霞1燕振刚2(1甘肃农业大学理学院甘肃兰州730070;2甘肃农业大学信息科学技术学院甘肃兰州730070)摘要农作物病虫害是农业生产管理的关键,为及时防控病虫害,人们通过各种技术手段识别和监测病虫害。

本文通过介绍目标检测算法YOLO (You Only Look Once )的发展历程及其在农作物病虫害识别中的应用,着重分析了YOLO 算法在提高农作物病虫害识别准确度和缩短识别时间的优势,以期为农业生产提供科学指导。

关键词YOLO 算法;农作物病虫害;目标检测;深度学习中图分类号S43Research Review of Crop Diseases and Pests Identification basedon YOLO AlgorithmWAN Yingxia 1YAN Zhengang 2(1College of Science,Gansu Agricultural University,Lanzhou,Gansu 730070;2College of Information Science and Technology,Gansu Agricultural University,Lanzhou,Gansu 730070)AbstractCrop pests and diseases are the key to agricultural production management.In order to preventand control pests and diseases in time,scientists identify and monitor pests and diseases through various technical means.This paper introduces the development process of the target detection algorithm YOLO (You Only Look Once),its application in the identification of crop pests and diseases.It focuses on the analysis of the advantages of YOLO algorithm in improving the accuracy of crop pest identification,and reducing the recognition time,and provides scientific guidance for agricultural production.KeywordsYOLO algorithm;crop pests and diseases;object detection;deep learning病虫害防控是农业生产和农户收入提升的重要保障。

基于YOLOv5_改进的咖啡叶片病虫害检测研究

基于YOLOv5_改进的咖啡叶片病虫害检测研究

识别方法,通过将 ConvNext 融入主干网络,并加入 ECA 注意
目标识别精度,如 FPN,SPPNet,R-FCN 等。 徐会杰等
可以使咖啡叶片病虫害高精度识别;通过及时发现病虫害种
然后通过卷积神经网络对候选区域分类,具有较高的定位和
[5]

出采用 Darknet-53 作为特征提取网络,将特征图进行拼接,
测精度。 王远 志 等[11] 基 于 Faster R - CNN 网 络 模 型,进 行
HSV 颜色空间转换及 Retinex 算法处理,对 3 种苹果叶部病
害识别。 这些方法由于针对特定环境检测的自身局限性,导
国内外针对自然环境下病虫害识别问题提出许多解决
stage 和 Two-stage
络的残差结构,不使用锚框和 NMS 后处理,减少了网络参
图 2 数据增强后图片
Fig. 2 Images after data enhancement
2 试验方法
在主干网络进行特征提取,提取到的特征称作特征层,根据
2. 1 YOLOv5 网络结构 YOLOv5 是一种单精度目标检测
获取的 3 个有效特征层进行下一步网络的构建。
特征点对应的物体情况,在数据集上可快速收敛,模型可定
。 One-stage 检测器有较高的推理速
度,不需要区域建议步长,如 YOLO、SSD 等可用于实时设备。
致在复杂背景环境下病斑相似性的叶片识别效果较差,大多
针对静态图片的单叶片病虫害分类,并且优化后的模型所占
内存空间资源较大。
鉴于此,笔者提出一种基于 YOLOv5 的改进咖啡病虫害
Two-stage 检测器将问题划分为 2 个阶段,先进行区域生成,

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

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

江苏农业学报(JiangsuJ.ofAgr.Sci.)ꎬ2023ꎬ39(5):1199 ̄1208http://jsnyxb.jaas.ac.cn储㊀鑫ꎬ李㊀祥ꎬ罗㊀斌ꎬ等.基于改进YOLOv4算法的番茄叶部病害识别方法[J].江苏农业学报ꎬ2023ꎬ39(5):1199 ̄1208.doi:10.3969/j.issn.1000 ̄4440.2023.05.012基于改进YOLOv4算法的番茄叶部病害识别方法储㊀鑫1ꎬ2ꎬ3ꎬ㊀李㊀祥1ꎬ㊀罗㊀斌2ꎬ3ꎬ㊀王晓冬2ꎬ3ꎬ㊀黄㊀硕2ꎬ3(1.东华理工大学信息工程学院ꎬ江西南昌330013ꎻ2.国家农业信息化工程技术研究中心ꎬ北京100097ꎻ3.北京市农林科学院智能装备技术研究中心ꎬ北京100097)收稿日期:2022 ̄08 ̄31基金项目:江苏省科技计划重点及面上项目(BE2021379)ꎻ江西省核地学数据科学与系统工程技术研究中心开放基金项目(JETRCNGDSS201801)作者简介:储㊀鑫(1998-)ꎬ女ꎬ安徽安庆人ꎬ硕士研究生ꎬ研究方向为计算机视觉ꎮ(E ̄mail)416940305@qq.comꎮ罗斌为共同第一作者通讯作者:李㊀祥ꎬ(Tel)0791 ̄83897395ꎻ(E ̄mail)tom_lx@126.com㊀㊀摘要:㊀为快速准确识别自然环境下的番茄叶片病害ꎬ提出一种基于改进YOLOv4算法的轻量化番茄叶部病害识别方法ꎮ该方法根据番茄病害特征采用K均值聚类算法调整先验框的维度ꎬ并使用宽度因子为0.25的Mo ̄bileNetv1代替YOLOv4原有的主干网络CSPDarknet53进行特征提取ꎬ并在特征融合网络PANet中引入深度可分离卷积代替原有的3ˑ3标准卷积ꎬ同时在主干网络的2个输出特征层和空间金字塔池化输出层分别嵌入卷积块注意力模块(CBAM)ꎬ提高模型识别精度ꎮ试验结果表明ꎬ改进后的模型对8类番茄叶片整体检测精准性(mAP)为98 76%ꎬ参数量为12 64Mꎬ传输帧数为1s101 76帧ꎬ相较于原YOLOv4模型ꎬ模型参数量减少80%ꎬ每秒传输帧数比原始YOLOv4模型提高了130%ꎮ关键词:㊀YOLOv4ꎻMobileNetꎻ轻量化ꎻ注意力机制ꎻ病害中图分类号:㊀S436.412㊀㊀㊀文献标识码:㊀A㊀㊀㊀文章编号:㊀1000 ̄4440(2023)05 ̄1199 ̄10IdentificationmethodoftomatoleafdiseasesbasedonimprovedYOLOv4algorithmCHUXin1ꎬ2ꎬ3ꎬ㊀LIXiang1ꎬ㊀LUOBin2ꎬ3ꎬ㊀WANGXiao ̄dong2ꎬ3ꎬ㊀HUANGShuo2ꎬ3(1.CollegeofInformationEngineeringꎬEastChinaUniversityofTechnologyꎬNanchang330013ꎬChinaꎻ2.NationalEngineeringTechnologyResearchCenterforAgriculturalInformatizationꎬBeijing100097ꎬChinaꎻ3.ResearchCenterofIntelligentEquipmentTechnologyꎬBeijingAcademyofAgricultureandForestryꎬBeijing100097ꎬChina)㊀㊀Abstract:㊀InordertoidentifytomatoleafdiseasesinnaturalenvironmentquicklyandaccuratelyꎬalightweighttomatoleafdiseaseidentificationmethodbasedonimprovedYOLOv4algorithmwasproposed.ThemethodusedK ̄meansclusteringalgorithmtoadjustthedimensionsofthepriorboxaccordingtothecharacteristicsoftomatodiseaseꎬandusedMobileNetv1withawidthfactorof0.25insteadoftheoriginalbackbonenetworkCSPDarknet53ofYOLOv4forfeatureextractionꎬandintroduceddeepseparableconvolutioninplaceoftheoriginal3ˑ3standardconvolutioninthefeaturefusionnetworkPANet.Atthesametimeꎬtheconvolutionalblockattentionmodule(CBAM)wasembed ̄dedinthetwooutputfeaturelayersandthespatialpyramidpoolingoutputlayerofthebackbonenetworktoimprovethemodelrecognitionaccuracy.Theresultsshowedthatꎬtheoveralldetectionaccuracy(mAP)oftheimprovedmodelforeighttypesoftomatoleaveswas98 76%ꎬtheparameterquantitywas12 64Mꎬandthetransmissionframenumberwas101 76f/sꎬwhichwas80%lowerthanthatoftheoriginalYOLOv4modelꎬandthenumberoftransmittedframespersecondwas130%higherthanthatoftheoriginalYOLOv4model.Keywords:㊀YOLOv4ꎻMobileNetꎻlightweightꎻattentionalmechanismꎻdiseases9911㊀㊀番茄是重要的经济作物之一ꎬ其营养价值和经济价值非常高ꎮ随着番茄植株的生长ꎬ大多数番茄叶部病害的发生也呈上升趋势[1]ꎮ因此ꎬ及时防治番茄叶部病害是提高番茄质量和产量的关键举措ꎬ而精准识别番茄叶部病害种类是番茄病虫害精准防控的前提ꎮ现阶段ꎬ中国大部分地区主要还是依赖人工排查番茄叶部病害ꎬ这不仅耗时耗力ꎬ而且仅仅通过肉眼分辨叶部病害种类ꎬ难免产生错误ꎬ从而不利于病害的及时防治[2]ꎮ对此ꎬ需要研究高效率㊁低成本的病害图像智能识别算法ꎬ为番茄叶部病害精准识别提供技术支撑ꎮ随着深度学习的不断发展和其在农业方面越来越多的应用[3]ꎬ有学者针对作物叶片病害检测识别进行了深入研究ꎮ刘文波等[4]将改进后的SOLOv2算法应用于番茄叶部病害检测ꎬ平均精度达42.3%ꎬ与其他文献的模型相比ꎬ精度不太理想ꎮ文斌等[5]在YOLOv3的基础上采用注意力特征金字塔和双瓶颈层构建新的网络ꎬ实现对三七叶片病害的识别ꎬ但检测速度仅为23fpsꎮHu等[6]提出将多卷积神经网络集成用于茶树叶部病害的识别ꎬ结果显示ꎬ该算法可以有效区分病叶与健康叶ꎬ但检测类别较少ꎮQi等[7]在YOLOv5模型中增加了1个挤压 ̄激发(SE)模块ꎬ构建新的模型ꎬ实现对番茄病害的识别ꎬ平均检测精度为91 07%ꎬ但模型参数量较大ꎬ高达42 796Mꎮ刘延鑫等[8]构建5种烟草叶部病害数据集ꎬ选用YOLOv3模型对烟草叶部病害进行识别检测ꎬ采用K ̄means++聚类方法对锚框(Anchorbox)的宽和高进行计算ꎬ提高了模型检测精度ꎬ得到整体检测精准性(mAP)为0 77ꎬ算法结果优于SSD(单步多框目标检测)算法检测的mAP值(0 69)ꎬ但模型单样本耗时比SSD算法多2sꎮ王超学等[9]分别以一张葡萄叶片和一串葡萄果实为单位做标注ꎬ将YOLOv3算法中的主干网络DarkNet ̄53替换为Ef ̄ficientNet ̄B3作为模型的主干特征提取网络ꎬ对12种葡萄病害㊁健康葡萄叶片和健康葡萄果实进行识别ꎬ识别精度为98 60%ꎬB3网络结构模型与其他以EfficientNet作为主干特征提取网络的模型训练时间和模型体积相近ꎬ且相对于YOLOv3网络模型ꎬ参数量缩减50%ꎬ模型体积缩减2/3的同时ꎬ模型平均识别精度提升3%ꎬ最终参数量仍然较大ꎬ约为29Mꎮ周维等[10]利用GhostNet网络代替YOLOv4中主干网络CSPDarkNet ̄53ꎬ并以深度可分离卷积代替普通卷积的方式改进路径聚合网络(PANet)ꎬ对网上获取的6种水稻病虫害数据集进行检测识别ꎬ识别准确率为79 36%ꎮ王权顺等[11]利用PlantVillage中苹果叶部黑星病图片与网上另外3种苹果叶部病害图片构建数据集ꎬ通过将YOLOv4中主干网络CSPdarknet53替换为DenseNet121网络ꎬ在保证特征提取能力的同时减少计算量ꎬ针对YOLOv4的预设锚框不适合应用于病斑检测的情况ꎬ采用二分K均值聚类算法得到一组新的锚框优化算法ꎬ相较于原YOLOv4算法mAP值提高0 89%ꎬ模型权重大小为62 71MBꎬ相较于其他文献报道ꎬ该算法识别病害种类较少ꎬ仅有4种ꎮ考虑到上述国内外学者对作物叶片病害检测识别的研究中存在的一些问题ꎬ例如ꎬ精确度低㊁检测速度慢㊁模型参数大等ꎬ本研究针对番茄叶片多类病害病斑差别较小㊁病斑形状复杂且无规律㊁检测识别方法的检测速率低和模型参数大等问题ꎬ提出一种改进的YOLOv4轻量化番茄叶部病害识别方法ꎬ使用K均值聚类算法得到9个新的先验框ꎬ同时利用迁移学习ꎬ将预训练YOLOv4算法得到的最优参数与在COCO数据集训练得到的MobileNetv1权重进行融合ꎬ并对主干网络㊁路径聚合网络进行改进ꎬ达到降低模型参数和减小计算量的目的ꎮ在主干网络的2个输出特征层和空间金字塔池化输出层分别嵌入坐标注意力机制CBAM(卷积块注意力模块)ꎬ以此避免改进网络后减少模型参数带来的精度损失ꎮ1㊀YOLOv4模型改进1.1㊀YOLOv4模型YOLOv4[12 ̄13]作为YOLO系列第4代单阶段目标检测算法ꎬ相较于YOLOv3算法ꎬ其在检测精度与检测速度方面均有较大的提升ꎮYOLOv4模型主要包括4个方面:输入端㊁主干网络(Backbone)㊁颈部网络(Neck)以及头部网络(Head)ꎮYOLOv4在输入端采用Mosaic数据增强[14]ꎬ丰富了数据的多样性ꎬ提高了模型的泛化能力ꎮ主干网络在DarkNet53的基础上融合了跨阶段局部网络的思想ꎬ形成特征提取效果更强的CSPDarkNet53结构ꎮCSPDarkNet53主干网络主要由5层残差网络Resblock_Body组成ꎬ同时通过下0021江苏农业学报㊀2023年第39卷第5期采样输出3个有效特征层ꎬ以此保证提取更完备的图像语义信息ꎮ颈部网络主要由空间金字塔池化(SpacepyramidpoolꎬSPP)[15]和PANet[16]组成ꎬ空间金字塔池化通过5个大小为5ˑ5㊁9ˑ9和13ˑ13的最大池化层来融合尺度大小不同的特征图ꎬ起到增加感受野的作用ꎮ路径聚合网络可以提高信息流在网络中的传递效率[17]ꎬ其对特征图进行上采样与下采样ꎬ借此融合网络提取到的特征信息ꎮ最后YOLOv4生成3个检测头用于检测目标[18]ꎮYOLOv4网络结构如图1所示ꎮ为提高模型检测效率ꎬ降低模型参数量ꎬ本研究提出一种改进的YOLOv4算法的番茄叶部病害识别方法ꎮInputs:输入图像ꎻSPP:空间金字塔池化ꎻCSPDarknet53:YOLOv4主干网络ꎻResblock_Body:残差网络ꎻPANet:路径聚合网络ꎻYoloHead:头部网络ꎮ图1㊀YOLOv4网络结构Fig.1㊀YOLOv4networkstructurediagram1.2㊀改进YOLOv4模型1.2.1㊀运用迁移学习㊀在原先的YOLOv4模型中ꎬ输入图片尺寸大小为416ˑ416ꎬ冻结层与非冻结层批次大小(Batchsize)均为16ꎬ模型检测番茄叶片的mAP为98 16%ꎮ由于本研究采用的番茄叶部病害数据集尺寸大小为256ˑ256ꎬ因此ꎬ在模型输入端将尺寸改为256ˑ256ꎬ并将冻结层与非冻结层batchsize均改为8ꎬ最终得到模型检测番茄叶片的mAP为98 58%ꎮ因此ꎬ在下文其他模型训练时均采用上述参数进行训练ꎮ由于COCO数据集包含的数据特征丰富ꎬ因此ꎬ本研究用到的所有模型的训练权重均为对应模型在COCO数据集训练所得ꎬ利用这些训练权重有利于得到更高精度的结果ꎮ由此可见ꎬ运用迁移学习有利于检测精度的提升ꎬ迁移学习改进后参数量由64 36M降低为63 79Mꎬ图2为参数改进前后mAP的对比ꎮ1021储㊀鑫等:基于改进YOLOv4算法的番茄叶部病害识别方法左图为修改参数前ꎬ右图为修改参数后ꎮmAP:整体检测精准性ꎮ图2㊀修改参数前后mAP对比Fig.2㊀Comparisonofoveralldetectionaccuracy(mAP)beforeandaftertheparameterwasmodified1.2.2㊀使用MobileNetv1为主干网络㊀MobileNetv1是2017年由谷歌公司提出的一种轻量化深层神经网络ꎬ与传统的卷积神经网络模型相比ꎬ该模型需要的计算工作量相对较少[19]ꎬ主要是用于嵌入各类硬件和软件设备ꎬ其最大优势是提出了深度可分离卷积(Depthwiseseparableconvolution)ꎬ并且Mobile ̄Netv1采用ReLU6代替ReLU作为激活函数ꎮ图3为普通卷积和深度可分离卷积结构的对比ꎮ采用多个ReLU6激活函数ꎬ可以在减少计算量的同时避免带来精度损失ꎮ图3㊀普通卷积(左)和深度可分离卷积结构(右)对比Fig.3㊀Ordinaryconvolution(left)anddepthseparableconvo ̄lutionstructure(right)㊀㊀为了使模型参数量更小ꎬ检测速度更快ꎬMo ̄bileNetv1在输入通道上引入可以使每一层网络均匀细化的宽度因子(a)ꎬ其取值为0 25㊁0 50㊁0 75㊁1 00ꎮ当a取1 00时ꎬ代表原始的网络ꎬ只有当a<1 00时ꎬ才有细化网络层的功能ꎮ本研究选取a为0 25的MobileNetv1网络作为主干网络对YOLOv4模型进一步压缩ꎬ在只给精度仅带来0 01%的损失的同时减少了23 23M参数量ꎬ因此ꎬ选取a为0 25的MobileNetv1网络代替YOLOv4的主干网络CSPDarkNet ̄53ꎬ大大地减少了模型参数量且几乎不带来识别精度的损失ꎮ1.2.3㊀改进PANet㊀深度可分离卷积的核心思想是将普通卷积运算分为逐深度卷积(Depthwiseconvo ̄lution)与逐点1ˑ1卷积(Pointwiseconvolution)2步进行[20]ꎮ逐深度卷积不同于普通卷积的是1个通道由1个卷积核负责ꎬ每个通道独立进行空间卷积运算ꎮ不同深度卷积的输出由逐点1ˑ1卷积重新组合输出ꎬ从而得到新的特征图ꎮ图4A为普通卷积运算ꎬ图4B为逐深度卷积运算ꎬ图4C为逐点卷积运算ꎮ㊀㊀由图4A可知ꎬ普通卷积计算量为:HˑWˑMˑNˑDkˑDk(1)由图4B可知ꎬ逐深度卷积计算量为:HˑWˑMˑDkˑDk(2)由图4C可知ꎬ逐点卷积计算量为:HˑWˑMˑN(3)式中ꎬW㊁H表示输入特征图ꎬM表示输入通道ꎬN表示输出通道ꎬDk表示卷积核ꎮ因此ꎬ采用深度可分离卷积代替普通卷积后的计算量与普通卷积计算量之比如公式(4)所示ꎬPA ̄Net采用深度可分离卷积代替3ˑ3普通卷积可以大2021江苏农业学报㊀2023年第39卷第5期A:普通卷积ꎻB:逐深度卷积ꎻC:逐点卷积ꎮM表示输入通道ꎬN表示输出通道ꎬDk表示卷积核ꎮ图4㊀深度可分离卷积Fig.4㊀Depthseparableconvolution幅度降低计算量ꎬ有利于模型的压缩ꎮHˑWˑMˑDkˑDk+HˑWˑMˑNHˑWˑMˑNˑDkˑDk=1N+1D2k(4)1.2.4㊀改进Anchorbox㊀YOLOv4是一种根据锚框对目标物体进行预测回归的算法ꎬ其原先Anchorbox是根据COCO数据集训练得到的固定数值ꎮ而本研究的对象是COCO数据集中鲜少出现的番茄叶片病害ꎬ并且数据集中番茄叶片病害种类繁多ꎬ导致数据集标注边界框尺寸相差甚远ꎮ为了使Anchorbox更接近数据集标注尺寸ꎬ本研究采用K均值聚类算法对本研究数据集的Anchorbox进行聚类分析ꎬ得到合适的锚框尺寸ꎮK均值聚类算法具有计算复杂度低㊁收敛速度快和可解释性强等特点ꎮ根据K均值聚类算法随机初始化得到9个中心点ꎬ计算每个坐标到中心点的欧式距离ꎬ将每个样本分配给离它最近的中心点ꎬ不断重复此步骤直到9个中心点不再移动ꎬ从而得到9个新的Anchorboxꎬ分别为(132ꎬ198)㊁(162ꎬ222)㊁(182ꎬ200)㊁(186ꎬ234)㊁(203ꎬ224)㊁(226ꎬ212)㊁(209ꎬ240)㊁(223ꎬ237)㊁(237ꎬ242)ꎬ将新的Anchorbox运用到算法中ꎬ并采用mAP值和GFLOPs值(每秒浮点运算次数)作为优化前后的评价指标ꎮ最终结果表明ꎬ优化后的mAP值增长了0 02%ꎬGFLOPs值也比优化前高出1 36Gꎮ图5是K均值聚类分布ꎮ1.2.5㊀添加CBAM㊀注意力机制CBAM模块是一种在通道和空间2个维度上对特征图像进行权重分配ꎬ从而提取到更有用的特征信息的模块ꎮ特ˑ:质心ꎻ圆点:数据集标注框样本ꎮ图5㊀先验框K均值聚类分布Fig.5㊀ClusterdistributionofpriorboxK ̄means征图像首先进入通道注意力模块ꎬ通过平均池化和最大池化得到各个通道的信息ꎬ并将得到的参数通过多层感知进行叠加ꎬ再经过激活函数处理ꎬ从而得到通道注意力特征图[21]ꎮ空间注意力模块主要关注空间层面上特征图的内在联系ꎬ将通道注意力模块的输出作为空间注意力模块的输入ꎬ经过卷积和激活操作后获得空间注意力模块的特征图[22]ꎮ为提取到番茄病害更丰富的有效特征信息ꎬ本研究在YOLOv4改进后的主干特征提取网络Mo ̄bileNetv1的2个输出特征层以及加强特征提取网络SPP的输出特征层上添加CBAM模块ꎬ以CBAM模块输出的有效特征图作为PANet的输入ꎬ传入PA ̄Net结构中进行卷积和上采样ꎬ最终提取到包含更多关键信息的特征图ꎮ改进后YOLOv4的网络结构如图6所示ꎮ2㊀结果与分析为验证本研究算法在多类番茄病害识别与分类方面的性能ꎬ首先构建数据集ꎬ然后分别对YOLOv4不同主干网络和在主干网络输出层加入的不同注意力机制模块进行训练ꎬ并选取经典的单阶段目标检测算法与本研究改进的YOLOv4算法进行对比ꎮ试验的硬件配置为IntelCoreTMi7 ̄10700K处理器和NVIDIAGeForceRTX3060显卡ꎮ操作系统是Ubuntu20.04.2LTS64位系统ꎮPython的版本为3.8.5ꎬPytorch的版本为1.8.1ꎮ在训练过程中ꎬ首先对输入的图片进行Mosaic(马赛克)数据增强ꎬ并使用余弦退火算法对学习率进行调整ꎬ每轮共设置300次迭代ꎬ冻结层和非冻结层batchsize设置为8ꎮ3021储㊀鑫等:基于改进YOLOv4算法的番茄叶部病害识别方法MobileNet:MobileNet轻量化网络ꎻSPP:空间金字塔池化网络ꎻCBAM:注意力机制ꎻPANet:路径聚合网络ꎻYOLOHead:头部网络ꎮ图6㊀改进后YOLOv4的网络结构Fig.6㊀ImprovednetworkstructureofYOLOv42.1㊀数据集的构建本研究所用数据集选自PlantVillage公开数据集中的番茄叶片集ꎬ其中包括细菌性斑点病㊁叶霉病㊁晚疫病㊁叶斑病㊁二斑叶螨病㊁黄曲叶病㊁早疫病叶片以及健康叶片ꎬ共8个类别ꎮ由于样本分布不均匀会影响模型训练结果ꎬ因此ꎬ每个类别的叶片选取相同的数量ꎬ均为1250张ꎮ数据集中早疫病图片仅有1000张ꎬ通过编写程序将其中250张图片旋转270ʎ后与原有图像整合ꎬ使其与其他类别图像达到相同数量ꎬ最终得到所有数据集10000张ꎬ数据集如图7所示ꎮ使用MakeSense工具对上述10000张番茄叶片病害图像进行标注ꎬ标注时利用最小外接矩形框标记出叶片位置并选择对应的类别标签ꎮ标注完成后ꎬ每张图片会产生1个对应的包含病害类别标签和矩形框坐标信息的txt文件ꎬ图8为数据集标注示例ꎮ通过编写随机抽样的程序ꎬ从10000张图片中随机选择8001张作为训练集ꎬ随机选择999张图片作为验证集ꎬ随机选择1000张图片作为测试集ꎮ2.2㊀评价指标为了评估算法的性能优劣ꎬ本研究以对所有类别的番茄叶片整体检测精准性(mAP)㊁每秒所执行的浮点运算次数㊁每秒帧数(FPS)㊁平均每张图片检测时间以及模型参数量作为评价模型大小㊁检测速度和精度的指标ꎮmAP由所有类别的平均精度(AP)取平均值得到ꎬAP是精确度(P)和召回率(R)组成的曲线的面积ꎮP㊁R的计算方法分别见公式(5)㊁公式(6)ꎬmAP的计算公式如公式(7)所示ꎮGFLOPs用于衡量模型的计算量ꎬ其值越大ꎬ表明模型计算速度越高ꎮFPS用于衡量模型的实时性能ꎬ其值越大ꎬ表明模型实时检测的性能越好ꎮP=TPTP+FP(5)R=TPTP+FN(6)mAP=1nðni=1AP(7)式中ꎬTP表示正确检测到的病害类别的数量ꎻFP表示误检病害类别的数量ꎻFN表示未检测到的病害类别的数量ꎻn表示番茄病害样本类别数ꎮ2.3㊀不同主干网络的性能比较针对YOLOv4参数量多和网络结构复杂等问题ꎬ分别选取不同的轻量化主干网络代替原来的CSPDarknet网络ꎬ对8001张训练集进行训练后ꎬ得到相应权重再分别检测验证集和测试集ꎮ为验证MobileNetv1轻量化网络检测番茄病害的优越性ꎬ选取的网络分别是MobileNetv2㊁MobileNetv3以及CSPDarknetꎬ预测结果如表1所示ꎮ由表1可以看出ꎬMobileNetv2和MobileNetv3虽然检测时间优于原YOLOv4模型ꎬ但识别精度(整体检测精准性)却低于原YOLOv4模型ꎮMo ̄bileNetv1不仅比其他轻量化网络识别模型的整体检测精准性高ꎬ检测速度也具有较大的优势ꎬ检测每张图片平均仅需9msꎬ比原YOLOv4模型快了13msꎬ满足实时检测的条件ꎮ4021江苏农业学报㊀2023年第39卷第5期图7㊀番茄叶片数据集Fig.7㊀Datasetoftomatoleaves图8㊀番茄叶片数据集标注示例Fig.8㊀Labelingsamplesoftomatoleafdataset表1㊀不同主干网络的性能比较Table1㊀Performancecomparisonofdifferentbackbonenetworks模型mAP(%)GFLOPsFPS(帧ꎬ1s)Time(ms)YOLOv498.5822.6544.2422YOLOv4+MobileNetv398.332.6870.4714YOLOv4+MobileNetv298.402.8881.1812YOLOv4+MobileNetv198.6810.01102.319mAP:模型对8类叶片整体检测精准性ꎻGFLOPs:每秒所执行的浮点运算次数ꎻFPS:每秒传输帧数ꎻTime:模型检测时间ꎮ2.4㊀不同注意力机制的比较为验证本研究在主干网络2个输出层和SPP结构的输出层添加CBAM模块的有效性ꎬ本研究设计了在同等条件下添加不同注意力模块的对比试验ꎬ结果如表2所示ꎮ由表2可以看出ꎬ在设定参数以及其他改进均相同的情况下ꎬ添加CBAM模块比添加CA模块和ECA模块的平均检测精度都具有优势ꎬ并且没有带来检测速度方面的损失ꎮ原因可能是CBAM模块属于轻量级模块ꎬ其插入5021储㊀鑫等:基于改进YOLOv4算法的番茄叶部病害识别方法网络层后并不会给网络带来运行负担ꎬ并且相较于ECA模块与CA模块只关注通道注意力ꎬCBAM模块可以在关注通道注意力的同时又关注空间注意力ꎬ而空间注意力对图像中的有效语义信息更为敏感ꎬ因此ꎬ添加CBAM模块使得空间注意力与通道注意力相辅相成ꎬ从而有效地提高网络的特征提取能力ꎮ表2㊀不同注意力机制的性能比较Table2㊀Performancecomparisonofdifferentattentionmechanisms模型㊀㊀㊀㊀mAP(%)FPS(帧ꎬ1s)Time(ms)YOLOv498.68102.319YOLOv4+ECA98.58102.309YOLOv4+CA98.6499.1410YOLOv4+CBAM98.76101.769mAP㊁FPS㊁Time见表1注ꎮECA㊁CA㊁CBAM表示不同的注意力机制模块ꎮ2.5㊀消融试验为验证本研究提出的每个改进对试验结果的积极影响ꎬ本研究设计了消融对比试验ꎬ试验内容包括运用迁移学习ꎬ以MobileNetv1为主干网络ꎬ改进PANetꎬ采用K均值聚类Anchorbox和添加CBAM模块ꎬ试验结果如表3所示ꎮ由表3可知ꎬ采用参数迁移学习后ꎬ番茄病害平均检测精度有所提高ꎬ而在采用MobileNetv1替换主干网络后ꎬ模型参数量减少了23 23Mꎬ在此基础上ꎬ进一步对PANet进行改进ꎬ使得模型参数量大幅度减少了28 26Mꎮ由此可以看出ꎬ对YOLOv4算法主干网络和路径聚合网络的改进ꎬ有效地减少了模型参数量ꎮ并且在网络中添加CBAM模块后ꎬ使得mAP有所提升的同时参数量仅有微小提升ꎮ因此ꎬ改进后的YOLOv4算法在保障精度有所提升的前提下ꎬ大大提高了模型检测病害的速度ꎮ表3㊀消融试验的性能比较Table3㊀Performancecomparisonofablationtests迁移学习引入MobileNetv1改进PANetK均值添加CBAM模型参数量(M)mAP(%)64.3698.16ɿ 63.7998.58ɿɿ 40.5698.57ɿɿɿ 12.3098.68ɿɿɿɿ 12.3098.70ɿɿɿɿɿ12.6498.76 表示未做改进ꎻ ɿ 表示已做改进ꎻmAP㊁Time见表1注ꎮ2.6㊀与其他单阶段目标检测算法比较当前的目标检测算法分为2类ꎬ一类是以Fas ̄terR ̄CNN[23]为代表的Two ̄stage检测算法ꎬTwo ̄stage检测算法识别精度较高ꎬ但对检测速度有所限制ꎬ显然不适用于实时检测场景ꎮ另一类是One ̄stage检测算法ꎬ这类算法直接使用网络检测目标边界框和类概率分数ꎬ检测速度高ꎬ识别准确率较高ꎬ适用于农业病害实时检测场景ꎮ因此ꎬ本研究选择对以SSD[24]和YOLO为代表的One ̄stage目标检测算法与改进的YOLOv4轻量化算法进行对比ꎬ表4为不同单阶段算法对比ꎮ由表4可以看出ꎬ本研究提出的算法平均每张图片检测速度分别比SSD和RetinaNet算法快6ms和5msꎬmAP比SSD和Reti ̄naNet算法高ꎮ本研究提出的算法模型参数量更是远远小于SSD㊁RetinaNet以及原YOLOv4模型ꎬ分别比SSD㊁RetinaNet以及原YOLOv4模型小9 64M㊁9 77M和51 33Mꎮ因此ꎬ本研究提出的算法在不影响识别精度的前提下大大减少了模型大小ꎬ更适合在嵌入式系统上部署ꎮ㊀㊀为验证改进后的YOLOv4算法是否可以较好地应用于自然场景ꎬ选择现实场景里番茄相对感染较多的4种病害ꎬ分别是早疫病㊁晚疫病㊁细菌性斑点病和黄曲叶病ꎬ使用改进后的YOLOv4算法对4种病害进行检测识别ꎮ图9为使用改进后的YOLOv4算法对病害识别的结果ꎬ可以看出该算法可以准确识别自然环境下的番茄叶部病害ꎮ6021江苏农业学报㊀2023年第39卷第5期表4㊀不同单阶段目标检测算法的性能比较Table4㊀Performancecomparisonofdifferentsingle ̄stageobjectdetectionalgorithms模型㊀㊀㊀mAP(%)GFLOPsFPS(帧ꎬ1s)参数量(M)Modelsize(MB)Time(ms)SSD98.5062.7966.3622.2898.815YOLOv498.1622.6544.2463.97256.422RetinaNet98.4834.3567.7222.41146.214本研究算法98.763.79101.7612.6453.79Modelsize:模型权重大小ꎻmAP㊁GFLOPs㊁FPS㊁Time见表1注ꎮA:早疫病ꎻB:晚疫病ꎻC:细菌性斑点病ꎻD:黄曲叶病ꎮ图9㊀自然场景病害识别结果Fig.9㊀Diseaseidentificationresultsinnaturalscenes3㊀结论针对不同种类番茄叶片病害误检率高㊁实时检测速度低等问题[25 ̄26]ꎬ本研究提出一种改进的YOLOv4轻量化检测算法ꎬ应用于8种番茄病害识别ꎬ改进后的模型在平均识别精确度㊁检测速度以及参数内存大小方面均取得了良好的效果ꎮ将YOLOv4的主干网络CSPDarkNet替换成轻量化的MobileNetv1网络ꎬ并采用深度可分离卷积取代PA ̄Net加强特征提取网络中的3ˑ3普通卷积ꎬ使得模型的计算量得到进一步压缩ꎮ通过有效调整YOLOv4原始模型参数ꎬ结合迁移学习策略ꎬ将Mo ̄bileNetv1在COCO数据集上训练的权重运用到改进的YOLOv4模型ꎬ并在主干网络的2个输出特征层和空间金字塔池化输出层分别嵌入坐标注意力机制ꎬ以此避免减小模型参数带来精度损失ꎮ应用K均值聚类算法对Anchorbox进行重新优化ꎬ得到新的Anchorbox值ꎬ取代原始YOLOv4的盲目搜索方法ꎬ从而提升了模型检测能力ꎮ本研究提出的模型传输帧数为101 76f/sꎬ比原始YOLOv4模型提高了130%ꎬ而模型参数量仅为12 64Mꎬ比原YOLOv4模型参数量减少了80%ꎬ在1000张测试集图片上整体检测精准性为98 76%ꎬ比原始YOLOv4模型提高了0 6个百分点ꎬ平均每张病害图片检测时间为9msꎮ因此ꎬ改进后的轻量化YOLOv4模型实时检测性能更好ꎬ精确度更高并且更适合在硬件上部署ꎮ下一步的研究重点是将本研究算法应用到各种复杂的自然场景ꎬ将深度学习应用到更广泛的农业工程中ꎮ参考文献:[1]㊀WANGXWꎬLIUJꎬZHUXN.Earlyreal ̄timedetectionalgorithmoftomatodiseasesandpestsinthenaturalenvironment[J].PlantMethodsꎬ2021ꎬ17(1):1 ̄17.[2]㊀XUCꎬDINGJQꎬQIAOYꎬetal.Tomatodiseaseandpestdiagno ̄sismethodbasedontheStackingofprescriptiondata[J].Comput ̄ersandElectronicsinAgricultureꎬ2022ꎬ197:106997. [3]㊀吕盛坪ꎬ李灯辉ꎬ冼荣亨.深度学习在我国农业中的应用研究现状[J].计算机工程与应用ꎬ2019ꎬ55(20):24 ̄33ꎬ51. [4]㊀刘文波ꎬ叶㊀涛ꎬ李㊀颀.基于改进SOLOv2的番茄叶部病害检测方法[J].农业机械学报ꎬ2021ꎬ52(8):213 ̄220. [5]㊀文㊀斌ꎬ曹仁轩ꎬ杨启良ꎬ等.改进YOLOv3算法检测三七叶片病害[J].农业工程学报ꎬ2022ꎬ38(3):164 ̄172. [6]㊀HUGSꎬYANGXWꎬZHANGYꎬetal.Identificationoftealeafdiseasesbyusinganimproveddeepconvolutionalneuralnetwork[J].SustainableComputing:InformaticsandSystemsꎬ2019ꎬ24:100353.[7]㊀QIJTꎬLIUXNꎬLIUKꎬetal.AnimprovedYOLOv5modelbasedonvisualattentionmechanism:applicationtorecognitionoftomatovirusdisease[J].ComputersandElectronicsinAgricultureꎬ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]REDMONJꎬDIVVALASꎬGIRSHICKRꎬetal.Youonlylookonce:unifiedꎬreal ̄timeobjectdetection[C].LasVegas:IEEECom ̄puterSocietyꎬ2016.7021储㊀鑫等:基于改进YOLOv4算法的番茄叶部病害识别方法[13]BOCHKOVSKIYAꎬWANGCYꎬLIAOHY.YOLOv4:optimalspeedandaccuracyofobjectdetection[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]HEKꎬZHANGXꎬRENSꎬetal.Spatialpyramidpoolingindeepconvolutionalnetworksforvisualrecognition[J].IEEEtransactionsonpatternanalysisandmachineintelligenceꎬ2015ꎬ37(9):1904 ̄1916.[16]LIUSꎬQILꎬQINHFꎬetal.Pathaggregationnetworkforin ̄stancesegmentation[C].SaltLakeCityꎬUTꎬUSA:IEEEPressꎬ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]SRINIVASUPNꎬSIVASAIJGꎬIJAZMFꎬetal.ClassificationofskindiseaseusingdeeplearningneuralnetworkswithMobileNetV2andLSTM[J].Sensorsꎬ2021ꎬ21(8):2852.[20]WANGHTꎬLUFYꎬTONGXꎬetal.Amodelfordetectingsafetyhazardsinkeyelectricalsitesbasedonhybridattentionmecha ̄nismsandlightweightMobilenet[J].EnergyReportsꎬ2021ꎬ7(S7):716 ̄724.[21]郝㊀帅ꎬ张㊀旭ꎬ马㊀旭ꎬ等.基于CBAM ̄YOLOv5的煤矿输送带异物检测[J].煤炭学报ꎬ2022ꎬ47(11):4149 ̄4158. [22]姚齐水ꎬ别帅帅ꎬ余江鸿.一种结合改进InceptionV2模块和CBAM的轴承故障诊断方法[J].振动工程学报ꎬ2022ꎬ35(4):949 ̄957.[23]RENSQꎬHEKMꎬGIRSHICKRꎬetal.FasterR ̄CNN:towardsreal ̄timeobjectdetectionwithregionproposalnetworks[J].IEEETransactionsonPatternAnalysisandMachineIntelligenceꎬ2017ꎬ39(6):1137 ̄1149.[24]LIUWꎬANGUELOVDꎬERHANDꎬetal.SSD:singleshotmultiboxdetector[M].AmsterdamꎬTheNetherland:ProceedingsoftheEuropeanConferenceonComputerVisionꎬ2016. [25]胡㊀政ꎬ张㊀艳ꎬ尚㊀静ꎬ等.高光谱图像在农作物病害检测识别中的研究进展[J].江苏农业科学ꎬ2022ꎬ50(8):49 ̄55. [26]牛学德ꎬ高丙朋ꎬ南新元ꎬ等.基于改进DenseNet卷积神经网络的番茄叶片病害检测[J].江苏农业学报ꎬ2022ꎬ38(1):129 ̄134.(责任编辑:陈海霞)8021江苏农业学报㊀2023年第39卷第5期。

基于YOLOv3_改进算法的烟叶原料烟草甲识别方法研究

基于YOLOv3_改进算法的烟叶原料烟草甲识别方法研究

河南农业科学,2023,52(11):157⁃166Journal of Henan Agricultural Sciencesdoi:10.15933/ki.1004-3268.2023.11.017基于YOLOv3改进算法的烟叶原料烟草甲识别方法研究张卫正1,陈赛越扬1,王艳玲1,帖金鑫2,丁佳3,李萌4,李灿林1,苏晓珂1,甘勇1(1.郑州轻工业大学计算机科学与技术学院,河南郑州450001;2.浙江中烟工业有限责任公司宁波卷烟厂,浙江宁波315040;3.河北中烟有限责任公司,河北石家庄050051;4.郑州轻工业大学烟草科学与工程学院,河南郑州450001)摘要:为了解决烟叶原料中烟草甲人工识别效率低、不准确的问题,实现烟草甲的精准、智能识别,提出一种基于YOLOv3改进算法的烟草甲识别模型。

首先采用Random Mix 数据增强技术扩充烟草甲数据集;然后,通过引入K-means++算法重新聚类锚框,改进YOLOv3对于烟草甲的识别能力;使用SIoU Loss 改进YOLOv3的边界框损失函数,提高模型定位的准确性,加速模型收敛;最后加入特征细化模块过滤冲突信息,优化模型对烟草甲数据集中小目标识别的精准度。

结果表明,改进的YOLOv3网络模型对测试集中烟草甲的平均检测精确率、召回率、F1分数和平均精度均值(mAP )分别达到93.26%、88.25%、0.90和94.59%,分别比现有的YOLOv3算法提高了12.21百分点、11.79百分点、0.12、12.40百分点,可为烟草甲的有效识别提供支撑。

关键词:烟叶;烟草甲;YOLOv3改进算法;数据增强;损失函数;目标识别中图分类号:S126;TP391.4文献标志码:A文章编号:1004-3268(2023)11-0157-10收稿日期:2023-06-03基金项目:河南省科技攻关项目(222102210037);河南省高等教育教学改革研究与实践重点项目(2021SJGLX189);浙江中烟工业有限责任公司科技项目(ZJZY2023D020)作者简介:张卫正(1982-),男,河南滑县人,讲师,博士,主要从事烟草信息学、数字农业与农业物联网研究。

基于YOLO_v5s目标检测算法的烤烟鲜叶成熟度识别方法

基于YOLO_v5s目标检测算法的烤烟鲜叶成熟度识别方法

杜鹏程,蒋笃忠,向 阳,等.基于YOLOv5s目标检测算法的烤烟鲜叶成熟度识别方法[J].江苏农业科学,2023,51(19):158-165.doi:10.15889/j.issn.1002-1302.2023.19.024基于YOLOv5s目标检测算法的烤烟鲜叶成熟度识别方法杜鹏程1,蒋笃忠2,向 阳1,敬礼恒2,伍守贵2,骆君华2(1.湖南农业大学机电工程学院,湖南长沙410128;2.湖南省烟草公司永州市公司,湖南永州425099) 摘要:新鲜烟叶的成熟度对烤后烟叶的质量有显著影响,可于烘烤前完成不同成熟度的新鲜烟叶分选,并对各成熟度采用特定烘烤工艺,以提高烘烤后的烟叶质量。

以云烟87的新鲜烟叶为试验样本,将采集的新鲜烟叶分为上部叶、中部叶、下部叶,并依据不同部位成熟度划分为欠熟、成熟、过熟。

采用YOLOv5s目标检测算法,利用卷积神经网络提取新鲜烟叶的不同成熟度特征,完成不同部位新鲜烟叶的成熟度识别试验,并与Faster-RCNN、Mask-RCNN检测模型的测试结果进行对比。

结果表明,YOLOv5s对云烟87不同部位新鲜烟叶成熟度检测的总体mAP值分别为99 1%、97.5%、98.5%,单幅图像检测平均时间为29~30ms。

3种模型的对比试验中,Mask-RCNN、Faster-RCNN测试的mAP值均低于YOLOv5s,单幅图像检测平均时间均远高于YOLOv5s。

采用YOLOv5s训练的不同部位新鲜烟叶的检测模型,在检测速度与精准度方面基本满足对新鲜烟叶成熟度识别的生产要求。

关键词:新鲜烟叶;深度学习;YOLOv5;成熟度 中图分类号:TP391.41 文献标志码:A 文章编号:1002-1302(2023)19-0158-07收稿日期:2023-02-12基金项目:湖南省自然科学基金(编号:2021JJ30363);湖南省烟草公司永州市公司科技计划项目。

作者简介:杜鹏程(1999—),男,湖南常德人,硕士研究生,主要从事农业电气化与自动化、机器视觉、深度学习、近红外检测、图像识别等研究。

基于改进的YOLOv3农作物目标检测算法

基于改进的YOLOv3农作物目标检测算法

基于改进的YOLOv3农作物目标检测算法
郭蓓;王贝贝;张志红;吴苏;李鹏;胡莉婷
【期刊名称】《农业大数据学报》
【年(卷),期】2024(6)1
【摘要】农作物图像在进行目标检测时,由于作物种植较密集、成像质量不佳等原因严重影响目标检测算法的检测精度。

针对存在的问题,提出一种基于YOLOv3的改进算法优化在农作物目标检测的检测性能:对YOLOv3的主干特征提取网络进行优化,利用原网络中输出的4倍降采样特征图对目标进行检测,并且在算法原网络残差块的基础上增加残差单元,以检测目标较小的农作物位置信息;提出高斯衰减函数,对图像中高度重叠的农作物候选框的衰减较强,在有效抑制冗余框的同时也可以有效地降低漏检率;对回归损失函数进行优化改进,用CIOU Loss作为损失函数,使得目标检测过程中最终的目标定位更加精确。

将改进的YOLOv3算法和原YOLOv3算法、Faster R-CNN算法在实拍的玉米作物图像数据集上进行对比实验,结果表明改进后的YOLOv3算法能有效检测农作物小目标,算法检测的平均准确率均值和检测速度都有明显的提升。

【总页数】8页(P40-47)
【作者】郭蓓;王贝贝;张志红;吴苏;李鹏;胡莉婷
【作者单位】中国气象局·河南省农业气象保障与应用技术重点开放实验室;河南中原光电测控技术有限公司
【正文语种】中文
【中图分类】TP3
【相关文献】
1.基于改进YOLOv3的列车运行环境图像小目标检测算法
2.基于改进Yolov3的飞机目标检测算法
3.基于改进YOLOv3的轻量级目标检测算法
4.基于改进YOLOv3的遥感目标检测算法
因版权原因,仅展示原文概要,查看原文内容请购买。

基于改进YOLOv5s-ECA-ASFF算法的茶叶病害目标检测

基于改进YOLOv5s-ECA-ASFF算法的茶叶病害目标检测

基于改进YOLOv5s-ECA-ASFF算法的茶叶病害目标检测叶荣;马自飞;高泉;李彤;邵郭奇;王白娟【期刊名称】《中国农机化学报》【年(卷),期】2024(45)1【摘要】在自然场景下,茶叶病害形状各异、目标小,传统卷积神经网络不适用于复杂背景下的病害检测。

因此,提出一种改进的YOLOv5s-ECA-ASFF茶叶病害目标检测算法。

该算法通过引入ECA通道注意力模块来增强通道维度上的全局上下文信息,并使用自适应空间特征融合(ASFF)技术改进茶叶病害的多尺度特征融合,提高模型的背景抗干扰能力。

同时,使用GIoU损失函数作为边界框损失函数,进一步提高回归目标的检测精度。

与原始YOLOv5s模型相比,改进YOLOv5s-ECA-ASFF模型在茶白星病、茶轮斑病、茶炭疽病、茶藻斑病的识别平均精度分别提高5%、4%、3%、2%,平均精度均值为92.1%。

此外,该模型的图像检测速度为64 f/s,并且综合性能也优于YOLOv4、SSD和Faster R-CNN模型。

因此,该模型为茶叶在自然生长环境下不同种类病害的检测提供参考,并为早期预测提供重要的技术支持。

【总页数】8页(P244-251)【作者】叶荣;马自飞;高泉;李彤;邵郭奇;王白娟【作者单位】云南农业大学食品科学技术学院;云南省作物生产与智慧农业重点实验室;云南农业大学大数据学院;云南农业大学茶学院【正文语种】中文【中图分类】S24;TP391.4【相关文献】1.基于改进SSD算法的航拍目标检测算法研究2.基于改进YOLOX的茶叶嫩芽目标检测研究3.基于改进单步多框目标检测的道路小目标检测算法4.基于YOLO-v5算法的航拍图像小目标检测改进算法因版权原因,仅展示原文概要,查看原文内容请购买。

  1. 1、下载文档前请自行甄别文档内容的完整性,平台不提供额外的编辑、内容补充、找答案等附加服务。
  2. 2、"仅部分预览"的文档,不可在线预览部分如存在完整性等问题,可反馈申请退款(可完整预览的文档不适用该条件!)。
  3. 3、如文档侵犯您的权益,请联系客服反馈,我们会尽快为您处理(人工客服工作时间:9:00-18:30)。

基于YOLOv3的多类烟草叶部病害检测研究
刘延鑫;王俊峰;杜传印;丁睿柔;高强;宗浩;姜红花
【期刊名称】《中国烟草科学》
【年(卷),期】2022(43)2
【摘要】烟草叶部病害种类繁多,病理复杂,严重影响烟草产量及品质,烟草病害精准检测是烟草病害及时防治的前提。

传统检测方式精准性差、效率低,基于深度学习的算法可提高烟草病害检测准确性。

本文以5种较为常见的烟草病害(普通花叶病、黄瓜花叶病毒病、赤星病、烟草野火病、气候性斑点病)为研究对象,构建基于YOLOv3的烟草病害检测模型,实现烟草多类病害的精准快速检测。

使用Darknet53特征网络提取烟叶病害特征并将不同尺度病害特征融合,并用K-means++算法对融合后特征进行分类和位置预测,通过非极大值抑制算法(NMS)去除冗余框,得到病害区域预测框。

用田间实际采集的烟草病害数据集,对构建的YOLOv3病害检测模型与SSD(Single Shot multibox Detector)模型对比测试。

结果表明,YOLOv3的mIoU为0.81,明显优于SSD的0.73,且YOLOv3模型的mAP为0.77,也高于SSD的0.69。

本研究构建的YOLOv3烟草病害检测模型能有效定位烟叶病害区域,实现多类烟草病害的检测,为精准病害防治提供参考。

【总页数】7页(P94-100)
【作者】刘延鑫;王俊峰;杜传印;丁睿柔;高强;宗浩;姜红花
【作者单位】山东农业大学信息科学与工程学院;山东潍坊烟草有限公司;山东临沂烟草有限公司
【正文语种】中文
【中图分类】S43
【相关文献】
1.甘草叶部病害研究进展简述
2.烟塔类特种结构病害鉴定及加固修复技术研究
3.基于改进YOLOv3的小目标检测方法研究
4.基于改进YOLOv3的机坪工作人员反光背心检测研究
5.基于GLCM纹理特征提取的黄瓜叶部病害检测算法研究
因版权原因,仅展示原文概要,查看原文内容请购买。

相关文档
最新文档