| 1,865 | 25 | 635 |
| 下载次数 | 被引频次 | 阅读次数 |
在YOLOv5深度神经网络基础上融合CA注意力机制,设计了田间杂草自动识别模型。对不同天气、不同背景下拍摄的包含农作物的杂草图片进行背景分割和数据增强等预处理后,利用深度卷积神经网络进行特征提取,采用随机梯度下降进行模型训练,并与YOLOv4、SSD、Faster R-CNN等方法进行对比实验,改进后的杂草模型具有更强的分类和检测能力,基本解决杂草与农作物相似度较高的问题。实验结果显示,融合CA注意力机制的YOLOv5的杂草识别模型大小为14.1 MB,识别精度达到0.911,召回率为0.950,F1为0.88,平均准确率为0.904,单张杂草图像检测平均耗时仅为20 ms,整体性能最好。模型能够准确识别出独立、与农作物交叉和贴近等不同空间位置的长茎尖叶杂草,为作物生长精准管理和田间精准喷药提供技术支持。
Abstract:An automatic field weed recognition model is designed based on YOLOv5 deep neural network incorporating CA attention mechanism. After pre-processing with the background segmentation and data enhancement for weed images containing crops taken under different weather and backgrounds, feature extraction is performed using deep convolutional neural networks, while the stochastic gradient descent is used for model training. Comparing with YOLOv4, SSD, Faster-RCNN and other methods, the improved weed model has stronger classification and detection ability, and basically solves the problem of high similarity between weeds and crops. Experimental results show that the weed recognition model of YOLOv5 fused with CA attention mechanism has a size of 14.1 MB, with a recognition accuracy of 0.911, a recall rate of 0.950, the F1 of 0.88, and an average accuracy of 0.904. The average time spent for single weed image detection is only 20 ms, which is the best overall performance. The model can accurately identify long-stemmed, sharp-leaved weeds in different spatial locations such as independent, intersecting with crops, and close to crops, providing technical support for accurate crop growth management and precise spraying in the field.
[1] 袁洪波,赵努东,程曼.基于图像处理的田间杂草识别研究进展与展望[J].农业机械学报,2020,51(增刊2):323-334.
[2] 耿亚玲,杜鹏程,宋姗姗,等.5种除草剂春季使用对麦田杂草的防治效果及对小麦的安全性[C]//绿色植保与乡村振兴—中国植物保护学会2018年学术年会论文集.西安:[s.n.],2018:242-243.
[3] 李香菊.近年我国农田杂草防控中的突出问题与治理对策[J].植物保护,2018,44(5):77-84.
[4] 彭明霞,夏俊芳,彭辉.融合FPN的Faster R-CNN复杂背景下棉田杂草高效识别方法[J].农业工程学报,2019,35(20):202-209.
[5] DAI X,XU Y L,ZHENG J Q,et al.Comparison of Image-based Methods for Determining the Inline Mixing Uniformity of Pesticides in Direct Nozzle Injection Systems[J].Biosystems Engineering,2020,190(C):157-175.
[6] 赵鹏,韦兴竹.基于多特征融合的田间杂草分类识别[J].农业机械学报,2014,45(3):275-281.
[7] 董亮,雷良育,李雪原,等.基于改进型人工神经网络的温室大棚蔬菜作物苗期杂草识别技术[J].北方园艺,2017(22):79-82.
[8] 周影,房建东,赵于东.基于主成分-贝叶斯分类模型的除草机器人杂草识别方法[J].机床与液压,2018,46(6):104-110.
[9] 姜红花,王鹏飞,张昭,等.基于卷积网络和哈希码的玉米田间杂草快速识别方法[J].农业机械学报,2018,49(11):30-38.
[10] 邓向武,齐龙,马旭,等.基于多特征融合和深度置信网络的稻田苗期杂草识别[J].农业工程学报,2018,34(14):165-172.
[11] WANG A C,ZHANG W,WEI X H.A Review on Weed Detection Using Ground-based Machine Vision and Image Processing Techniques[J].Computers and Electronics in Agriculture,2019,158:226-240.
[12] PUNITHAVATHI R,RANI D A C,SUGHASHINI K R,et al.Computer Vision and Deep Learning-enabled Weed Detection Model for Precision Agriculture[J].Computer Systems Science & Engineering,2023,44(3):2759-2774.
[13] 孙俊,何小飞,谭文军,等.空洞卷积结合全局池化的卷积神经网络识别作物幼苗与杂草[J].农业工程学报,2018,34(11):159-165.
[14] 彭明霞,夏俊芳,彭辉.融合FPN的Faster R-CNN复杂背景下棉田杂草高效识别方法[J].农业工程学报,2019,35(20):202-209.
[15] 孟庆宽,张漫,杨晓霞,等.基于轻量卷积结合特征信息融合的玉米幼苗与杂草识别[J].农业机械学报,2020,51(12):238-245.
[16] 刘闽,李喆,李曜丞,等.基于重参数化YOLOv5的输电线路缺陷边缘智能检测方法[J/OL].高电压技术:1-11[2023-03-10].https://doi.org/10.13336/j.1003-6520.hve.20220861.
[17] LIU W,ANGUELOV D,ERHAN D,et al.SSD:Single Shot MultiBox Detector[J/OL].(2016-12-29)[2023-03-10].https://arxiv.org/abs/1512.02325.
[18] REDMON J,DIVVALA S,GIRSHICK R,et al.You Only Look Once:Unified,Real-time Object Detection[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).Las Vegas:IEEE,2016:779-788.
[19] 闫彬,樊攀,王美茸,等.基于改进YOLOv5m的采摘机器人苹果采摘方式实时识别[J].农业机械学报,2022,53(9):29-38.
[20] LEI F,TANG F F,LI S H.Underwater Target Detection Algorithm Based on Improved YOLOv5[J].Journal of Marine Science and Engineering,2022,10(3):310.
[21] 宋怀波,王亚男,王云飞,等.基于YOLO v5s的自然场景油茶果识别方法[J].农业机械学报,2022,53(7):234-242.
[22] 赵志宏,李晴,杨绍普,等.基于BiLSTM与注意力机制的剩余使用寿命预测研究[J].振动与冲击,2022,41(6):44-50.
[23] 李向荣,孙立辉.融合注意力机制的多尺度红外目标检测[J/OL].红外技术:1-9[2023-03-10].http://kns.cnki.net/kcms/detail/53.1053.tn.20221013.1703.002.html.
[24] 王恒涛,张上,张朝阳,等.基于YOLOv5的轻量化PCB缺陷检测[J].无线电工程,2022,52(11):2094-2100.
基本信息:
中图分类号:S451;TP391.41;TP183
引用信息:
[1]郭柏璋,牟琦,冀汶莉.融合注意力机制的YOLOv5深度神经网络杂草识别方法[J].无线电工程,2023,53(12):2771-2782.
基金信息:
2019年西安市科技局项目(20193054YF042NS042)~~
2023-04-20
2023-04-20
2023-04-20