融合空间与通道特征的水稻叶片病害轻量化识别方法

Lightweight recognition method for rice leaf diseases integrating spatial and channel features

  • 摘要: 【目的】从网络结构与损失函数2个维度对YOLOv5n模型进行系统改进,全面提升模型轻量化程度及其对复杂形态水稻叶片病害的检测精度,为实现水稻叶片病害智能识别提供技术支持,进而满足现代精准农业智能化管理的需求。【方法】以YOLOv5n模型为基础,首先在骨干网络中引入CBRM模块、MLCA模块与ShuffleBlock模块,重构原有网络架构;然后在颈部网络的每个C3层前嵌入Shuffle Attention机制,实现通道与空间2个维度的特征协同学习;最后将原CIoU损失函数替换为EIoU损失函数,提升病斑定位精度及改善模型在训练过程中的收敛稳定性。采用YOLOv5n-SHCA模型和YOLOv5n模型在包含细菌性叶斑病、褐斑病、叶瘟病3种水稻叶片病害的图像数据集上进行检测,以评估YOLOv5n-SHCA模型的有效性。【结果】YOLOv5n-SHCA模型能稳定识别细菌性叶斑病、褐斑病、叶瘟病3种水稻叶片病害特征,且对小目标病斑识别敏感。与当前主流YOLO模型相比,YOLOv5n-SHCA模型在多项性能指标上均实现有效提升,同时参数规模大幅降低,体现了模型在检测精度与轻量化设计间的良好平衡,具备更优的实际应用潜力,具体表现:mAP@50为0.966,mAP@50-95为0.746,精确率为94.2%,召回率为92.0%,模型推理速度提升至1.4 ms/图,而参数规模降至1.53 M。YOLOv5n-SHCA模型对水稻叶片细菌性叶斑病、褐斑病、叶瘟病的检测精度分别为0.994、0.921和0.983,均优于YOLOv5n模型,能有效解决水稻叶片褐斑病因病斑密集、尺度微小、与叶瘟病形态相似等复杂因素对检测效果不利的问题,增强模型在复杂噪声背景下对病斑特征的感知与鉴别能力。【结论】由CBRM模块、MLCA模块与ShuffleBlock模块组成轻量化骨干网络,在颈部网络集成Shuffle Attention机制,并以EIoU损失函数替代CIoU损失函数而建立的YOLOv5n-SHCA模型,在有效提升对复杂形态水稻叶片病害检测精度的同时实现了模型轻量化,为水稻叶片病害的智能识别提供了技术支撑。

     

    Abstract: 【Objective】This study aimed to systematically improve the YOLOv5n model from two aspects (network structure and loss function) to comprehensively enhance the degree of model lightweighting and detection accuracy for complex rice leaf diseases, providing technical support for the intelligent recognition of rice leaf diseases, thus meeting the demands of intelligent management for modern precision agriculture.【Method】Based on the YOLOv5n model, the CBRM module, MLCA module, and ShuffleBlock module were first introduced into the backbone network to reconstruct the original network structure. Subsequently, the Shuffle Attention mechanism was embedded before each C3 layer of the neck network to facilitate collaborative feature learning in both channel and spatial dimensions. Finally, the original CIoU loss function was replaced with the EIoU loss function to improve disease spot localization accuracy and enhance the convergence stability of the model during training. To validate the effectiveness of the proposed method, experimental evaluations were conducted on an image dataset comprising three major types of rice leaf diseases: bacterial leaf spot, brown spot, and leaf blast.【Result】The YOLOv5n-SHCA model demonstrated stable recognition of characteristics of the three rice leaf diseases (bacterial leaf spet,brown spot, and leaf blast) and exhibited high sensitivity in detecting small-target disease spots. In comparative tests with the mainstream YOLO model, the YOLOv5n-SHCA model achieved effective improvements in multiple performance indicators, while substantially reducing the parameter scale, reflecting a favorable balance between detection accuracy and lightweight design, showing strong potential for practical application. Specifically, the model attained a mAP@50 of 0.966, a mAP@50-95 of 0.746, a precision of 94.2%, a recall of 92.0%, an inference speed of 1.4 ms/image, and a parameter size reduced to 1.53 M. The detection accuracies for bacterial leaf spot, brown spot, and leaf blast were 0.994, 0.921, and 0.983 respectively, all outperforming the YOLOv5n model. The proposed model effectively addressed the problems of complex factors such as dense disease spots, small disease spot scales, and morphological similarity of rice leaf blast on brown spot detection, thereby enhancing perception and discrimination of disease spot features of the model under complex background noise.【Conclusion】By constructing a lightweight backbone network comprising the CBRM module, MLCA module, and ShuffleBlock module, integrating the Shuffle Attention mechanism into the neck network, and adopting the EIoU loss function in place of CIoU, the proposed YOLOv5n-SHCA model effectively improves the detection accuracy of complex rice leaf diseases while achieving model lightweigh-ting, which provides technical support for the intelligent recognition of rice leaf diseases.

     

/

返回文章
返回