Citation: | JIANG Fei, YE Wei, LI Zhaoxing, et al. A maize disease detection method based on feature interaction under imbalanced sample condition[J]. Journal of South China Agricultural University, 2025, 46(3): 399-406. DOI: 10.7671/j.issn.1001-411X.202410019 |
To address the issues of imbalanced data samples and low detection accuracy in maize leaf disease detection under complex environments.
An improved object detection network SF_YOLOv5 was proposed. First, based on the multi-scale pyramid structure of YOLOv5, a novel spatial-feature pyramid structure (SPD-FPN) was designed to enhance the network’s ability to recognize small-target disease features at high-resolution levels while retaining large-target information at low-resolution levels, thereby improving overall detection accuracy and robustness. Second, the Focal Loss function was introduced to increase the weight of hard-to-classify samples and reduce the influence of easily classified samples, ensuring that the model focused more on the minority samples often overlooked in imbalanced datasets. Additionally, transfer learning was applied to the design of SF_YOLOv5, where pre-trained YOLOv5 model parameters were transferred to the improved SF_YOLOv5 network for training. This leveraged knowledge from large-scale datasets to enhance the model’s generalization capability for maize disease detection.
Experimental validation on the constructed maize disease dataset showed that SF_YOLOv5 achieved a mean average precision (mAP) of 93.3% and a recall of 89.6%, significantly outperforming the original YOLOv5 model. And the model was small in size, and had been deployed in mobile devices.
The results demonstrate that the improved network performs better than the original model in detecting maize leaf diseases under imbalanced sample conditions. This approach can be applied to intelligent diagnosis of maize diseases in farmland scenarios with imbalanced data, providing a theoretical foundation for real-time maize disease detection in the agricultural sector.
[1] |
许景辉, 邵明烨, 王一琛, 等. 基于迁移学习的卷积神经网络玉米病害图像识别[J]. 农业机械学报, 2020, 51(2): 230-236. doi: 10.6041/j.issn.1000-1298.2020.02.025
|
[2] |
刘合兵, 鲁笛, 席磊. 基于MobileNetV2和迁移学习的玉米病害识别研究[J]. 河南农业大学学报, 2022, 56(6): 1041-1051. doi: 10.3969/j.issn.1000-2340.2022.6.hennannydxxb202206016
|
[3] |
LI E L, WANG L W, XIE Q J, et al. A novel deep learning method for maize disease identification based on small sample-size and complex background datasets[J]. Ecological Informatics, 2023, 75: 102011. doi: 10.1016/j.ecoinf.2023.102011.
|
[4] |
PAN S Q, QIAO J F, WANG R, et al. Intelligent diagnosis of northern corn leaf blight with deep learning model[J]. Journal of Integrative Agriculture, 2022, 21(4): 1094-1105. doi: 10.1016/S2095-3119(21)63707-3
|
[5] |
LI Y W, CHEN Y X, WANG Y. Disease recognition of maize leaf based on KNN and feature extraction[J]. International Journal of Pattern Recognition and Artificial Intelligence, 2022, 36(11): 2257004. doi: 10.1142/s021800142257004x.
|
[6] |
JOCHER G. YOLOv5 by Ultralytics (Version 7.0) [CP/OL]. 2020. https://doi.org/10.5281/zenodo.3908559
|
[7] |
JJOCHER G, CHAURASIA A, QIU J. Ultralytics YOLO (Version 8.0. 0) [CP/OL]. 2023. https://github.com/ultralytics/ultralytics
|
[8] |
LIU W, ANGUELOV D, ERHAN D, et al. SSD: Single shot MultiBox detector[C]//Computer Vision – ECCV 2016. Cham: Springer International Publishing, 2016: 21-37.
|
[9] |
GIRSHICK R, DONAHUE J, DARRELL T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C]//2014 IEEE Conference on Computer Vision and Pattern Recognition. Columbus, OH, USA: IEEE, 2014: 580-587.
|
[10] |
SINDHWANI N, VERMA S, BAJAJ T, et al. Comparative analysis of intelligent driving and safety assistance systems using YOLO and SSD model of deep learning[J]. International Journal of Information System Modeling and Design, 2021, 12(1): 131-146. doi: 10.4018/IJISMD.2021010107
|
[11] |
KIM J A, SUNG J Y, PARK S H. Comparison of faster-RCNN, YOLO, and SSD for real-time vehicle type recognition[C]//2020 IEEE International Conference on Consumer Electronics - Asia (ICCE-Asia). Seoul, Korea: IEEE, 2020: 1-4.
|
[12] |
PHADTARE M , CHOUDHARI V, PEDRAM R, et al. Comparison between YOLO and SSD mobile net for object detection in a surveillance drone[J]. International Journal of Engineering Science, 2021, 5: 1-5.
|
[13] |
WANG C Y, YEH I H, LIAO H M. YOLOv9: Learning what you want to learn using programmable gradient information[EB/OL]. arXiv: 2402.13616v2(2024-02-29) [2024-10-10]. https://arxiv.org/pdf/2402.13616.pdf.
|
[14] |
WANG A, CHEN H, LIU L H, et al. YOLOv10: Real-time end-to-end object detection[EB/OL]. arXiv: 2405.14458v2(2024-10-30) [2025-03-03]. https://arxiv.org/pdf/2405.14458v2.pdf.
|
[15] |
SUNKARA R, LUO T. No more strided convolutions or pooling: A new CNN building block for low-resolution images and small objects[C]//European conference on machine learning and knowledge discovery in databases. Cham: Springer Nature Switzerland. 2022: 443-459.
|
[16] |
WANG J F, CHEN Y, DONG Z K, et al. Improved YOLOv5 network for real-time multi-scale traffic sign detection[J]. Neural Computing and Applications, 2023, 35(10): 7853-7865. doi: 10.1007/s00521-022-08077-5
|
[17] |
LIN T Y, DOLLÁR P, GIRSHICK R, et al. Feature pyramid networks for object detection[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). Honolulu, HI, USA: IEEE, 2017: 936-944.
|
[18] |
LIN T Y, GOYAL P, GIRSHICK R, et al. Focal loss for dense object detection[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2020, 42(2): 318-327. doi: 10.1109/TPAMI.2018.2858826
|