林业机械与木工设备2024,Vol.52Issue(11) :22-28.

基于改进YOLOv5s模型的棉花地膜识别方法研究

Research on Cotton Mulch Recognition Method Based on Improved YOLOv5s Model

陈东胜 潘江如 董芙楠 董恒祥
林业机械与木工设备2024,Vol.52Issue(11) :22-28.

基于改进YOLOv5s模型的棉花地膜识别方法研究

Research on Cotton Mulch Recognition Method Based on Improved YOLOv5s Model

陈东胜 1潘江如 2董芙楠 2董恒祥3
扫码查看

作者信息

  • 1. 奎屯银力棉油机械有限公司,新疆奎屯 833200
  • 2. 新疆工程学院,新疆 乌鲁木齐 830000
  • 3. 新疆农业大学交通与物流工程学院,新疆乌鲁木齐 830052
  • 折叠

摘要

针对棉花杂质去除过程中地膜杂质难以去除的问题,研究采用深度学习的方法,提出一种基于改进YOLOv5s的地膜识别模型.首先,对YOLOv5s模型损失函数改进,在保证满足检测速度的基础上,提高对地膜的识别准确率;其次,添加CA(Coordinate Attention)注意力机制,增强算法的特征提取能力和检测精度,进一步提高对棉花地膜的识别准确率,以此得到地膜杂质的精确位置信息.试验结果表明:相比于传统方法,采用深度学习的方法可在原有去除基础上,进一步识别出89.9%的地膜杂质,有效提高棉花质量;且相比于原YOLOv5s模型,改进后模型识别的准确率和召回率分别为88.3%和86.3%,分别提高了 7.5%和7.9%,对地膜杂质的检测效果明显提高,有效解决了地膜杂质难以去除的问题.

Abstract

Aiming at the problem of difficult removal of plastic film impurities in the process of removing cotton impu-rities,a deep learning method was adopted for removal,and an improved YOLOv5s based plastic film recognition model was proposed.Firstly,the YOLOv5s model loss function was improved to improve the recognition accuracy of plastic film while ensuring the detection speed was met;Secondly,adding CA(Coordinate Attention)attention mechanism enhances the algorithm's feature extraction ability and detection accuracy,further improving the recogni-tion accuracy of cotton plastic film,and obtaining accurate location information of plastic film impurities.The experi-mental results show that compared to traditional methods,the use of deep learning can further identify 89.9%of plas-tic film impurities on the basis of original removal,effectively improving the quality of cotton;Compared with the original YOLOv5s model,the improved model achieved recognition accuracy and recall of 88.3%and 86.3%,re-spectively,with an increase of 7.5%and 7.9%.The detection effect of plastic film impurities was significantly im-proved,effectively solving the problem of difficult removal of plastic film impurities.

关键词

棉花/地膜/YOLOv5s/损失函数/CA注意力机制

Key words

cotton/geomembrane/YOLOv5s/loss function/CA attention mechanism

引用本文复制引用

出版年

2024
林业机械与木工设备
国家林业局哈尔滨林业机械研究所

林业机械与木工设备

影响因子:0.574
ISSN:2095-2953
段落导航相关论文