华南农业大学学报2024，Vol.45Issue(1)：108-115,8.DOI:10.7671/j.issn.1001-411X.202209029
基于改进YOLOv51的田间水稻稻穗识别
Rice panicle recognition in field based on improved YOLOv51 model
摘要
Abstract
[Objective]YOLOv51 algorithm model was introduced and improved to realize accurate,efficient and nondestructive detection of rice panicles in field environment.[Method]Taking rice in the field as the research object,rice image samples were collected by digital single-mirror reflex camera.The original image data were augmented and expanded offline after manual labeling,so as to construct an image data set for field rice.The YOLOv51 algorithm was improved adaptively,the effective channel attention(ECA)mechanism was put in front of the spatial pyramid pooling(SPP)layer and in the cross-stage-partial-connections(CSP)layer,and a comparative experiment was conducted.The optimal algorithm was selected as the benchmark model to carry out attention mechanism and data-enhanced ablation experiments,and the optimal performance model was obtained by testing.The improved YOLOv51 was compared with YOLOv51,YOLOv5x,SSD and Faster R-CNN.[Result]In the improved rice recognition framework of YOLOv51,placing ECA before the network SPP layer resulted in better performance.Using test set images to verify the model,the average accuracy of recognition results was 93.63％,the average recall rate was 90.94％,and the overall average accuracy reached 95.05％.Compared with the non-fused YOLOv51 algorithm,the average accuracy of the improved YOLOv51 algorithm was 3.03 percent higher and the detection rate was 8.20 frames per ms faster.Compared with the YOLOv5x algorithm,the average precision of the improved YOLOv51 algorithm was improved by 0.62 percent,the detection rate was faster by 5.41 frames per ms,and the memory occupation was reduced by 74.1 MB.The results showed that the comprehensive performance of the improved YOLOv51 algorithm was better than other algorithms in rice panicle detection in the field.[Conclusion]It is feasible to introduce the improved YOLOv51 algorithm into rice panicle detection in field environment.The algorithm has high accuracy,fast detection speed and small memory occupation,which can avoid the subjectivity of traditional manual detection and is of great significance for rice panicle detection and non-destructive yield estimation.关键词
水稻/估产/稻穗检测/YOLOv51/ECA/注意力机制Key words
Rice/Yield estimation/Rice panicle detection/YOLOv51/Efficient channel attention/Attention mechanism分类
农业科技引用本文复制引用
蔡竹轩,蔡雨霖,曾凡国,岳学军..基于改进YOLOv51的田间水稻稻穗识别[J].华南农业大学学报,2024,45(1):108-115,8.基金项目
广州市科技计划(202206010088) （202206010088）
省级大学生创新创业训练计划(S202210564011) （S202210564011）
