基于改进MobileNetV2的轻量化茶叶病害检测方法

    A lightweight disease identification method for tea leaves based on improved MobileNetV2

    • 摘要:
      目的 解决茶叶病虫害检测中现有深度学习模型难以兼顾精度与效率,尤其不适合在资源受限的嵌入式设备上部署的问题。
      方法 以MobileNetV2为基础架构引入2个关键改进,设计出轻量化且高精度的识别模型MobileNetV2-GCA-LS:一是设计了一种新颖的幽灵坐标注意力(Ghost coordinate attention, GCA)模块,该模块融合坐标注意力的位置敏感性与GhostNet的高效计算特性,增强对关键病害区域的特征表达;二是采用标签平滑(Label smoothing, LS)正则化策略优化训练过程,提升模型泛化能力。模型在公开的茶树病害数据集上进行了训练与验证。
      结果  MobileNetV2-GCA-LS模型在测试集上识别准确率达到了94.54%,F1为94.29%,性能显著优于MobileNetV2、MobileNetV3-Small、EfficientNet-B0、ResNet50和GhostNet。同时,该模型保持了较低的复杂度,参数量为2.6089×106,浮点运算次数(Floating point operations, FLOPs)为0.3347 ×1010,验证了其在资源受限设备上部署的可行性。
      结论 本研究提出的改进策略能够有效地提升模型识别茶叶病害的性能,在精度与效率间取得了良好的平衡,为智慧农业领域的病害智能监测与精准防控提供了实用的技术方案。

       

      Abstract:
      Objective To address the challenge in tea leaf disease detection, where existing deep learning models struggle to balance accuracy and efficiency, particularly unsuitable for deployment on resource-constrained embedded devices.
      Method A lightweight and high-accuracy recognition model of MobileNetV2-GCA-LS was proposed based on the MobileNetV2 architecture, incorporating two key innovations. First, a novel ghost coordinate attention (GCA) module was designed, which integrated the positional sensitivity of coordinate attention with the computational efficiency of GhostNet, thereby enhanced the feature representation of critical disease areas. Second, the label smoothing (LS) regularization strategy was employed to optimize the training process and improve the model generalization capability. The model was trained and validated on a publicly available tea leaf disease dataset.
      Result The proposed MobileNetV2-GCA-LS model achieved a recognition accuracy of 94.54% and F1 of 94.29% on the test set, significantly outperforming comparison models including MobileNetV2, MobileNetV3-Small, EfficientNet-B0, ResNet50 and GhostNet. Meanwhile, the model maintained low complexity with 2.608 9×106 parameters and 0.334 7×1010 floating point operations (FLOPs) of computational cost, demonstrating its feasibility for deployment on resource-constrained devices.
      Conclusion The proposed model effectively enhances the performance of tea leaf disease recognition, achieving a well-balanced trade-off between accuracy and efficiency. This provides a practical technical solution for intelligent monitoring and precise control of plant diseases in smart agriculture.

       

    /

    返回文章
    返回