技术特征:
1.一种基于改进yolov5网络的广佛手病虫害识别方法,其特征在于,包括以下步骤:
获取广佛手病虫害图像,并进行标注,构建初始数据集;
引入yolov5网络模型,并对yolov5网络模型的骨干网络和Neck模块进行改进;
利用所述初始数据集对改进后的yolov5网络模型进行训练、验证和测试,得到最终的病虫害识别模型;
对待检测图像进行预处理;判断待检测图像为晴天拍摄或雨天拍摄,若为雨天拍摄,则利用Attentive GAN算法对待检测图像进行处理;若为晴天拍摄,则不进行处理;
基于所述病虫害识别模型对预处理后的待检测图像进行病虫害识别。
2.根据权利要求1所述的一种基于改进yolov5网络的广佛手病虫害识别方法,其特征在于,对yolov5网络模型的骨干网络进行改进的过程包括:
在原yolov5网络模型的骨干网络的基础上,添加4个Involution自卷积模块,并将骨干网络中的3个C3模块分别替换为两个ESPMIXConv2d模块和一个ESPConv2d模块;改进后的骨干网络包括:Focus模块、Conv模块、Involution自卷积模块、C3模块、ESPMIXConv2d模块、SPP模块和ESPConv2d模块;其中,所述ESPMIXConv2d模块依次对输入的特征图进行分割处理,分组卷积和空洞卷积处理;所述ESPConv2d模块对输入的特征图进行普通卷积和空洞卷积处理。
3.根据权利要求2所述的一种基于改进yolov5网络的广佛手病虫害识别方法,其特征在于,所述ESPMIXConv2d模块对输入的特征图进行处理的过程为:
将输入的特征图的输入通道进行分割操作,当输入特征图输入通道数为c时,将输入特征图分割成特征图1和特征图2;其中,所述特征图1和所述特征图2的通道数为c/2;
对所述特征图1进行卷积核为1*1,步长为1,分组数groups=c/2的分组卷积,并将分组卷积后的所述特征图1分别输入扩张率为1和扩张率为2的空洞卷积;
对所述特征图2进行卷积核为3*3,步长为1,分组数groups=c/2的分组卷积,并将分组卷积后的特征图2分别输入扩张率为1和扩张率为2的空洞卷积;
将使用扩张率为1的空洞卷积的特征图1和特征图2进行Concat连接;将使用扩张率为2的空洞卷积的特征图1和特征图2进行Concat连接;
将两个Concat的连接的输出进行Add相加操作,得到输出特征图。
4.根据权利要求1所述的一种基于改进yolov5网络的广佛手病虫害识别方法,其特征在于,对yolov5网络模型的Neck模块进行改进的过程为:
引入CBAM注意力模块,并将所述CBAM注意力模块与将原yolov5网络模型中Neck模块的Upsample上采样模块封装在一起,组成upCBAM模块;其中,所述CBAM注意力模块包括通道注意力模块和空间注意力模块;
将原yolov5网络模型的Upsample上采样模块替换为所述upCBAM模块。
5.根据权利要求1所述的一种基于改进yolov5网络的广佛手病虫害识别方法,其特征在于,所述upCBAM模块对输入的特征图进行处理的过程为:
将输入的特征图与经所述通道注意力模块的输出的特征图进行乘法操作,输出特征M;
将输出的特征M与经所述空间注意力模块输出的特征M进行乘法操作,得到特征M';
利用所述Upsample上采样模块对特征M'进行上采样操作,得到最终输出的特征图。
6.根据权利要求1所述的一种基于改进yolov5网络的广佛手病虫害识别方法,其特征在于,所述获取广佛手病虫害图像,并进行标注,构建初始数据集,包括:
对广佛手发病叶片进行东、南、西、北四个方向拍摄,得到广佛手病虫害图像;
对广佛手病虫害类别进行分类,并利用labelImg工具对所述广佛手病虫害图像一一进行标记,得到初始数据集。
7.根据权利要求1所述的一种基于改进yolov5网络的广佛手病虫害识别方法,其特征在于,所述初始数据集包括训练集、验证集和测试集;利用所述训练集用于对改进后的yolov5网络模型进行训练;利用所述验证集对训练好的yolov5网络模型进行验证;利用Attentive GAN算法对所述测试集进行处理,再利用处理后的所述测试集对验证后的yolov5网络模型的性能进行测试,得到最终的病虫害识别模型。
8.一种基于改进yolov5网络的广佛手病虫害识别系统,其特征在于,包括:
图像预处理模块,用于判断待检测图像为晴天拍摄或雨天拍摄,若为雨天拍摄,则利用Attentive GAN算法对待检测图像进行处理;若为晴天拍摄,则不进行处理;
病虫害识别模块,用于对yolov5网络模型的骨干网络和Neck模块进行改进,并利用改进的yolov5网络模型对待检测图像进行病虫害识别。
9.根据权利要求8所述的一种基于改进yolov5网络的广佛手病虫害识别系统,其特征在于,改进的所述骨干网络包括:Focus模块、Conv模块、4个Involution自卷积模块、C3模块、2个ESPMIXConv2d模块、SPP模块和ESPConv2d模块;
改进的Neck模块中的采样模块由通道注意力模块、空间注意力模块和Upsample上采样模块封装组成。
技术总结
本发明公开了一种基于改进yolov5网络的广佛手病虫害识别方法及系统,方法包括:获取广佛手病虫害图像,并进行标注,构建初始数据集;引入yolov5网络模型,并对yolov5网络模型的骨干网络和Neck模块进行改进;利用初始数据集对改进后的yolov5网络模型进行训练、验证和测试,得到最终的病虫害识别模型;对待检测图像进行预处理;判断待检测图像为晴天拍摄或雨天拍摄,若为雨天拍摄,则利用Attentive GAN算法对待检测图像进行处理;若为晴天拍摄,则不进行处理;基于病虫害识别模型对预处理后的待检测图像进行病虫害识别。本发明将改进的yolov5网络与Attentive GAN算法相结合,不仅能实现在雨天天气条件下对广佛手病虫害的识别,还能降低网络参数量和网络模型的大小,提高识别准确率。
技术研发人员:王卫星;骆润玫;胡凯;刘伟康;廖飞;曹亚芃;刘泽乾;
受保护的技术使用者:华南农业大学;
技术研发日:2021.10.20
技术公布日:2022.02.01
本文用于企业家、创业者技术爱好者查询,结果仅供参考。