Fu Zilong, Yin Lifeng, Cui Can, Wang Yi
College of Rail Intelligent Engineering, Dalian Jiaotong University, Dalian, China.
Front Plant Sci. 2024 Dec 6;15:1499911. doi: 10.3389/fpls.2024.1499911. eCollection 2024.
Accurate diagnosis of grape leaf diseases is critical in agricultural production, yet existing detection techniques face challenges in achieving model lightweighting while ensuring high accuracy. In this study, a real-time, end-to-end, lightweight grape leaf disease detection model, MHDI-DETR, based on an improved RT-DETR architecture, is presented to address these challenges. The original residual backbone network was improved using the MobileNetv4 network, significantly reducing the model's computational requirements and complexity. Additionally, a lightSFPN feature fusion structure is presented, combining the Hierarchical Scale Feature Pyramid Network with the Dilated Reparam Block structure design from the UniRepLKNet network. This structure is designed to overcome the challenges of capturing complex high-level and subtle low-level features, and it uses Efficient Local Attention to focus more efficiently on regions of interest, thereby enhancing the model's ability to detect complex targets while improving accuracy and inference speed. Finally, the integration of GIou and Focaler-IoU into Focaler-GIoU enhances detection accuracy and convergence speed for small targets by focusing more effectively on both simple and difficult samples. The findings from the experiments suggest that The MHDI-DETR model results in a 56% decrease in parameters and a 49% reduction in floating-point operations, respectively, compared with the RT-DETR model, in terms of accuracy, the model achieved precision rates of 96.9%, 92.6%, and 72.5% for accuracy, mAP50, and mAP50:95, respectively. Compared with the RT-DETR model, these represent improvements of 1.9%, 1.2%, and 1.2%. Overall, the MHDI-DETR model surpasses the RT-DETR and other mainstream detection models in both detection accuracy and degree of lightness, achieving dual optimization in efficiency and accuracy, and providing an efficient technical solution for automated agricultural disease management.
准确诊断葡萄叶病害在农业生产中至关重要,但现有的检测技术在实现模型轻量化的同时确保高精度方面面临挑战。在本研究中,提出了一种基于改进的RT-DETR架构的实时、端到端、轻量化葡萄叶病害检测模型MHDI-DETR,以应对这些挑战。使用MobileNetv4网络对原始残差骨干网络进行了改进,显著降低了模型的计算需求和复杂度。此外,提出了一种轻量级SFPN特征融合结构,将分层尺度特征金字塔网络与UniRepLKNet网络的扩张重参数化块结构设计相结合。该结构旨在克服捕捉复杂高级和细微低级特征的挑战,并使用高效局部注意力更有效地聚焦于感兴趣区域,从而提高模型检测复杂目标的能力,同时提高准确性和推理速度。最后,将GIou和Focaler-IoU集成到Focaler-GIoU中,通过更有效地聚焦简单和困难样本,提高了小目标的检测精度和收敛速度。实验结果表明,与RT-DETR模型相比,MHDI-DETR模型的参数分别减少了56%,浮点运算减少了49%,在准确率方面,该模型在准确率、mAP50和mAP50:95上的精确率分别达到了96.9%、92.6%和72.5%。与RT-DETR模型相比,分别提高了1.9%、1.2%和1.2%。总体而言,MHDI-DETR模型在检测精度和轻量化程度上均超越了RT-DETR和其他主流检测模型,实现了效率和准确性的双重优化,为农业病害自动化管理提供了高效的技术解决方案。