用主动学习+轻量模型,让农业机器人在边缘设备上高效识别番茄和花。
Active Learning-Driven Lightweight YOLOv9: Enhancing Efficiency in Smart Agriculture
- 基于主动学习筛选高价值样本,减少标注负担。
- 轻量模块与注意力机制提升小目标和遮挡场景检测精度。
- 适合资源受限的智能农业场景,实测mAP达67.8%。
本研究针对温室环境中部署于边缘设备的农业机器人对番茄及番茄花的实时检测需求,解决实际成像条件下因相机距离变化导致的目标尺度差异大、植株结构严重遮挡以及类别分布高度不平衡等问题。传统依赖全量标注数据集的方法难以兼顾高精度与高效部署。为此,提出一种主动学习驱动的轻量级目标检测框架,融合数据分析、模型设计与训练策略。首先分析原始图像中目标尺寸分布,重新定义有效检测范围,提升学习稳定性;其次引入高效特征提取模块降低计算开销,并设计轻量注意力机制增强多尺度与遮挡场景下的特征表达能力;最后采用主动学习策略,在有限标注预算下迭代选择高信息量样本进行标注与训练,显著提升少数类与小目标的识别性能。实验表明,该方法在保持低参数量与推理成本(适用于边缘设备)的同时,有效提升番茄与花的检测效果,在有限标注条件下实现67.8% mAP的整体检测准确率,验证了其在智能农业应用中的实用性和可行性。
原文摘要 · Abstract (English)
This study addresses the demand for real-time detection of tomatoes and tomato flowers by agricultural robots deployed on edge devices in greenhouse environments. Under practical imaging conditions, object detection systems often face challenges such as large scale variations caused by varying camera distances, severe occlusion from plant structures, and highly imbalanced class distributions. These factors make conventional object detection approaches that rely on fully annotated datasets difficult to simultaneously achieve high detection accuracy and deployment efficiency. To overcome these limitations, this research proposes an active learning driven lightweight object detection framework, integrating data analysis, model design, and training strategy. First, the size distribution of objects in raw agricultural images is analyzed to redefine an operational target range, thereby improving learning stability under real-world conditions. Second, an efficient feature extraction module is incorporated to reduce computational cost, while a lightweight attention mechanism is introduced to enhance feature representation under multi-scale and occluded scenarios. Finally, an active learning strategy is employed to iteratively select high-information samples for annotation and training under a limited labeling budget, effectively improving the recognition performance of minority and small-object categories. Experimental results demonstrate that, while maintaining a low parameter count and inference cost suitable for edge-device deployment, the proposed method effectively improves the detection performance of tomatoes and tomato flowers in raw images. Under limited annotation conditions, the framework achieves an overall detection accuracy of 67.8% mAP, validating its practicality and feasibility for intelligent agricultural applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。