Improved YOLOv5-Based Method for Date Palm Leaf Disease Recognition
作者:Yichen Zhao, Siyu Heng, Ying Wang, Tianmeng He · 发表于:2024 3rd International Conference on Artificial Intelligence and Software Engineering (ICAISE) · 年份:2024 · DOI:10.1109/icaise65384.2024.00013 · 被引用次数:1
The Significance of Deep Learning-Based Date Palm Leaf Disease Recognition for the Local Economy of Xinjiang, China” Cultivation of date palms is intricately tied to the daily needs of people, and it plays a vital role in the local economy of Xinjiang, China. Establishing a deep learning model capable of recognizing diseases in date palm leaves is of paramount importance. Such a model can enhance the quality of dates and reduce economic losses for farmers. Addressing challenges like disease spot occlusion, small disease targets that often lead to missed detections, and low recognition accuracy in complex background images, this study introduces an enhanced YOLOv5s-based model referred to as YOLOv5s-CBS. By incorporating the CBAM attention mechanism into the network's prediction head, it strengthens the feature extraction capability for small disease targets. This enhancement allows the model to better extract the features of the disease targets on date palm leaves, addressing the challenges of occluded and small targets and improving detection accuracy. The model employs the SIoU function to accelerate training and enhance inference accuracy. It also improves the feature fusion portion of the backbone network by using a bidirectional weighted feature pyramid network (BiFPN) to strengthen multi-scale feature fusion, ultimately improving recognition across different scales. To validate the effectiveness of YOLOv5s-CBS, it is compared with six other models: Faster R-CNN, YOLOv5m...