AHN-YOLO: A Lightweight Tomato Detection Method for Dense Small-Sized Features Based on YOLO Architecture
2025
Wenhui Zhang | Feng Jiang
Convolutional neural networks (CNNs) are increasingly applied in crop disease identification, yet most existing techniques are optimized solely for laboratory environments. When confronted with real-world challenges such as diverse disease morphologies, complex backgrounds, and subtle feature variations, these models often exhibit insufficient robustness. To effectively identify fine-grained disease features in complex scenarios while reducing deployment and training costs, this paper proposes a novel network architecture named AHN-YOLO, based on an improved YOLOv11-n framework that demonstrates balanced performance in multi-scale feature processing. The key innovations of AHN-YOLO include (1) the introduction of an ADown module to reduce model parameters; (2) the adoption of a Normalized Wasserstein Distance (NWD) loss function to stabilize small-feature detection; and (3) the proposal of a lightweight hybrid attention mechanism, Light-ES, to enhance focus on disease regions. Compared to the original architecture, AHN-YOLO achieves a 17.1 % reduction in model size. Comparative experiments on a tomato disease detection dataset under real-world complex conditions demonstrate that AHN-YOLO improves accuracy, recall, and mAP-50 by 9.5%, 7.5%, and 9.2%, respectively, indicating a significant enhancement in detection precision. When benchmarked against other lightweight models in the field, AHN-YOLO exhibits superior training efficiency and detection accuracy in complex, dense scenarios, demonstrating clear advantages.
显示更多 [+] 显示较少 [-]