用AI识别16种杂草11个生长期,实现精准农业中的自动除草。
WeedVision: Multi-Stage Growth and Classification of Weeds using DETR and RetinaNet for Precision Agriculture
- 结合DETR与RetinaNet模型,按生长阶段分类杂草。
- RetinaNet在测试集上达0.904 mAP,比DETR高6.4%。
- 适合需要实时检测的智能农业系统开发者参考。
杂草管理是农业中的关键挑战,因杂草与作物争夺资源导致严重减产。准确识别不同生长阶段的杂草对有效管理至关重要,但对农户而言仍具挑战性,需识别多种物种在多个生长期的表现。本研究采用先进的目标检测模型——基于ResNet50的检测变压器(DETR)和基于ResNeXt101的RetinaNet,对16种经济重要杂草共174类生长阶段(从幼苗到成熟期共11周)进行识别与分类。构建了一个包含203,567张图像的高质量数据集,由专家逐帧标注物种与生长阶段。模型经严格训练与评估,结果显示:RetinaNet性能更优,训练集mAP达0.907,测试集为0.904,显著高于DETR的0.854和0.840;同时召回率更高,推理速度达7.28 FPS,更适合实时应用。两种模型均随植株成熟而精度提升。该研究为发展精准、可持续、自动化的杂草管理策略提供关键支持,推动基于AI的农业系统向实时物种特异性检测迈进。
原文摘要 · Abstract (English)
Weed management remains a critical challenge in agriculture, where weeds compete with crops for essential resources, leading to significant yield losses. Accurate detection of weeds at various growth stages is crucial for effective management yet challenging for farmers, as it requires identifying different species at multiple growth phases. This research addresses these challenges by utilizing advanced object detection models, specifically, the Detection Transformer (DETR) with a ResNet50 backbone and RetinaNet with a ResNeXt101 backbone, to identify and classify 16 weed species of economic concern across 174 classes, spanning their 11 weeks growth stages from seedling to maturity. A robust dataset comprising 203,567 images was developed, meticulously labeled by species and growth stage. The models were rigorously trained and evaluated, with RetinaNet demonstrating superior performance, achieving a mean Average Precision (mAP) of 0.907 on the training set and 0.904 on the test set, compared to DETR's mAP of 0.854 and 0.840, respectively. RetinaNet also outperformed DETR in recall and inference speed of 7.28 FPS, making it more suitable for real time applications. Both models showed improved accuracy as plants matured. This research provides crucial insights for developing precise, sustainable, and automated weed management strategies, paving the way for real time species specific detection systems and advancing AI-assisted agriculture through continued innovation in model development and early detection accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。