A lightweight MHDI-DETR model for detecting grape leaf diseases
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...
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Frontiers Media S.A.
2024-12-01
|
| Series: | Frontiers in Plant Science |
| Subjects: | |
| Online Access: | https://www.frontiersin.org/articles/10.3389/fpls.2024.1499911/full |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1846139946667081728 |
|---|---|
| author | Zilong Fu Lifeng Yin Can Cui Yi Wang |
| author_facet | Zilong Fu Lifeng Yin Can Cui Yi Wang |
| author_sort | Zilong Fu |
| collection | DOAJ |
| description | 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. |
| format | Article |
| id | doaj-art-b6a19f3d94dc49dbb2ee64c6c424fc08 |
| institution | Kabale University |
| issn | 1664-462X |
| language | English |
| publishDate | 2024-12-01 |
| publisher | Frontiers Media S.A. |
| record_format | Article |
| series | Frontiers in Plant Science |
| spelling | doaj-art-b6a19f3d94dc49dbb2ee64c6c424fc082024-12-06T04:32:37ZengFrontiers Media S.A.Frontiers in Plant Science1664-462X2024-12-011510.3389/fpls.2024.14999111499911A lightweight MHDI-DETR model for detecting grape leaf diseasesZilong FuLifeng YinCan CuiYi WangAccurate 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.https://www.frontiersin.org/articles/10.3389/fpls.2024.1499911/fullRT-DETRtarget detectiongrapevine leaf diseaselightweighting modeldeep learning |
| spellingShingle | Zilong Fu Lifeng Yin Can Cui Yi Wang A lightweight MHDI-DETR model for detecting grape leaf diseases Frontiers in Plant Science RT-DETR target detection grapevine leaf disease lightweighting model deep learning |
| title | A lightweight MHDI-DETR model for detecting grape leaf diseases |
| title_full | A lightweight MHDI-DETR model for detecting grape leaf diseases |
| title_fullStr | A lightweight MHDI-DETR model for detecting grape leaf diseases |
| title_full_unstemmed | A lightweight MHDI-DETR model for detecting grape leaf diseases |
| title_short | A lightweight MHDI-DETR model for detecting grape leaf diseases |
| title_sort | lightweight mhdi detr model for detecting grape leaf diseases |
| topic | RT-DETR target detection grapevine leaf disease lightweighting model deep learning |
| url | https://www.frontiersin.org/articles/10.3389/fpls.2024.1499911/full |
| work_keys_str_mv | AT zilongfu alightweightmhdidetrmodelfordetectinggrapeleafdiseases AT lifengyin alightweightmhdidetrmodelfordetectinggrapeleafdiseases AT cancui alightweightmhdidetrmodelfordetectinggrapeleafdiseases AT yiwang alightweightmhdidetrmodelfordetectinggrapeleafdiseases AT zilongfu lightweightmhdidetrmodelfordetectinggrapeleafdiseases AT lifengyin lightweightmhdidetrmodelfordetectinggrapeleafdiseases AT cancui lightweightmhdidetrmodelfordetectinggrapeleafdiseases AT yiwang lightweightmhdidetrmodelfordetectinggrapeleafdiseases |