Rice Diseases Identification Method Based on Improved YOLOv7-Tiny

Abstract

The accurate and rapid identification of rice diseases is crucial for enhancing rice yields. However, this task encounters several challenges: (1) Complex background problem: The rice background in a natural environment is complex, which interferes with rice disease recognition; (2) Disease region irregularity problem: Some rice diseases exhibit irregular shapes, and their target regions are small, making them difficult to detect; (3) Classification and localization problem: Rice disease recognition employs identical features for both classification and localization tasks, thereby affecting the training effect. To address the aforementioned problems, an enhanced rice disease recognition model leveraging the improved YOLOv7-Tiny is proposed. Specifically, in order to reduce the interference of complex background, the YOLOv7-Tiny model’s backbone network has been enhanced by incorporating the Convolutional Block Attention Module (CBAM); subsequently, to address the irregularity issue in the disease region, the RepGhost bottleneck module, which is based on structural reparameterization techniques, has been introduced; Finally, to resolve the classification and localization issue, a lightweight YOLOX decoupled head has been proposed. The experimental results have demonstrated that: (1) The enhanced YOLOv7-Tiny model demonstrated elevated F1 scores and mAP@.5, achieving 0.894 and 0.922, respectively, on the rice pest and disease dataset. These scores exceeded the original YOLOv7-Tiny model’s performance by margins of 3.1 and 2.2 percentage points, respectively. (2) In comparison to the YOLOv3-Tiny, YOLOv4-Tiny, YOLOv5-S, YOLOX-S, and YOLOv7-Tiny models, the enhanced YOLOv7-Tiny model achieved higher F1 scores and mAP@.5. The improved YOLOv7-Tiny model boasts a single image inference time of 26.4 ms, satisfying the requirement for real-time identification of rice diseases and facilitating deployment in embedded devices.