YOLOWeeds: A novel benchmark of YOLO object detectors for multi-class weed detection in cotton production systems

被引:153
作者
Dang, Fengying [1 ]
Chen, Dong [2 ]
Lu, Yuzhen [3 ,4 ,5 ]
Li, Zhaojian [1 ]
机构
[1] Michigan State Univ, Dept Mech Engn, E Lansing, MI 48824 USA
[2] Michigan State Univ, Dept Elect & Comp Engn, E Lansing, MI 48824 USA
[3] Michigan State Univ, Dept Biosyst & Agr Engn, E Lansing, MI 48824 USA
[4] Mississippi State Univ, Dept Agr & Biol Engn, Mississippi State, MS 39762 USA
[5] Michigan State Univ, Biosyst & Agr Engn, 524 S Shaw Ln, E Lansing, MI 48824 USA
关键词
Cotton; Dataset; Deep learning; Machine vision; Precision agriculture; Weed detection; AMARANTH AMARANTHUS-PALMERI; CLASSIFICATION; MANAGEMENT; GLYPHOSATE; EVOLUTION; INDEXES; CROPS;
D O I
10.1016/j.compag.2023.107655
中图分类号
S [农业科学];
学科分类号
09 ;
摘要
Weeds are among the major threats to cotton production. Overreliance on herbicides for weed control has accelerated the evolution of herbicide-resistance in weeds and caused increasing concerns about environments, food safety and human health. Machine vision systems for automated/robotic weeding have received growing interest towards the realization of integrated, sustainable weed management. However, in the presence of unstructured field environments and significant biological variability of weeds, it remains a serious challenge to develop reliable weed identification and detection systems. A promising solution to address this challenge are the development of arge-scale, annotated image datasets of weeds specific to cropping systems and data-driven AI (artificial intelligence) models for weed detection. Among various deep learning architectures, a diversity of YOLO (You Only Look Once) detectors is well-suited for real-time application and has enjoyed great popularity for generic object detection. This study presents a new dataset (CottoWeedDet12) of weeds important to cotton production in the southern United States (U.S.); it consists of 5648 images of 12 weed classes with a total of 9370 bounding box annotations, collected under natural light conditions and at varied weed growth stages in cotton fields. A novel, comprehensive benchmark of 25 state-of-the-art YOLO object detectors of seven versions including YOLOv3, YOLOv4, Scaled-YOLOv4, YOLOR and YOLOv5, YOLOv6 and YOLOv7, has been established for weed detection on the dataset. Evaluated through the Monte-Caro cross validation with 5 replications, the detection accuracy in terms of mAP@0.5 ranged from 88.14 % by YOLOv3-tiny to 95.22 % by YOLOv4, and the accuracy in terms of mAP@[0.5:0.95] ranged from 68.18 % by YOLOv3-tiny to 89.72 % by Scaled-YOLOv4. All the YOLO models especially YOLOv5n and YOLOv5s have shown great potential for real-time weed detection, and data augmentation could increase weed detection accuracy. Both the weed detection dataset2 and software program codes for model benchmarking in this study are publicly available3, which will be to be valuable resources for promoting future research on big data and AI-empowered weed detection and control for cotton and potentially other crops.
引用
收藏
页数:13
相关论文
共 86 条
[1]   Performance of deep learning models for classifying and detecting common weeds in corn and soybean production systems [J].
Ahmad, Aanis ;
Saraswat, Dharmendra ;
Aggarwal, Varun ;
Etienne, Aaron ;
Hancock, Benjamin .
COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2021, 184 (184)
[2]  
[Anonymous], 2015, Advances in Neural Information Processing Systems
[3]   Opportunities for Robotic Systems and Automation in Cotton Production [J].
Barnes, Edward ;
Morgan, Gaylon ;
Hake, Kater ;
Devine, Jon ;
Kurtz, Ryan ;
Ibendahl, Gregory ;
Sharda, Ajay ;
Rains, Glen ;
Snider, John ;
Maja, Joe Mari ;
Thomasson, J. Alex ;
Lu, Yuzhen ;
Gharakhani, Hussein ;
Griffin, James ;
Kimura, Emi ;
Hardin, Robert ;
Raper, Tyson ;
Young, Sierra ;
Fue, Kadeghe ;
Pelletier, Mathew ;
Wanjura, John ;
Holt, Greg .
AGRIENGINEERING, 2021, 3 (02) :339-362
[4]   Thermal weed control technologies for conservation agriculture-a review [J].
Bauer, Martin V. ;
Marx, Christian ;
Bauer, Fabienne V. ;
Flury, Dominique M. ;
Ripken, Tammo ;
Streit, Bernhard .
WEED RESEARCH, 2020, 60 (04) :241-250
[5]   Robot for weed species plant-specific management [J].
Bawden, Owen ;
Kulk, Jason ;
Russell, Ray ;
McCool, Chris ;
English, Andrew ;
Dayoub, Feras ;
Lehnert, Chris ;
Perez, Tristan .
JOURNAL OF FIELD ROBOTICS, 2017, 34 (06) :1179-1199
[6]  
Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, 10.48550/arXiv.2004.10934, DOI 10.48550/ARXIV.2004.10934]
[7]   Albumentations: Fast and Flexible Image Augmentations [J].
Buslaev, Alexander ;
Iglovikov, Vladimir I. ;
Khvedchenya, Eugene ;
Parinov, Alex ;
Druzhinin, Mikhail ;
Kalinin, Alexandr A. .
INFORMATION, 2020, 11 (02)
[8]   Performance evaluation of deep transfer learning on multi-class identification of common weed species in cotton production systems [J].
Chen, Dong ;
Lu, Yuzhen ;
Li, Zhaojian ;
Young, Sierra .
COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2022, 198
[9]  
Chen Dong, 2022, arXiv, DOI DOI 10.48550/ARXIV.2210.09509
[10]   You Only Look One-level Feature [J].
Chen, Qiang ;
Wang, Yingming ;
Yang, Tong ;
Zhang, Xiangyu ;
Cheng, Jian ;
Sun, Jian .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :13034-13043