arXiv:2510.24885cs.CV2025-10被引 6

把水果成熟度从分类转为连续概率估计,提升采摘机器人决策精度

FruitProm: Probabilistic Maturity Estimation and Detection of Fruits and Vegetables

  • 将成熟度建模为连续概率分布,用不确定性量化判断置信度
  • 在大规模数据集上实现85.6%的检测精度,同时输出成熟度概率
  • 适合需要精准判断成熟度的智能农业系统,如自动采摘机器人

水果蔬菜成熟度评估是农业自动化中的关键任务,直接影响产量预测和机器人采摘。当前深度学习方法大多将成熟度视为离散分类问题(如未熟、成熟、过熟),这种刚性设定与生物成熟过程的连续性相悖,导致信息丢失和类别边界模糊。本文挑战这一范式,将成熟度估计重构为连续的概率学习任务。我们对最先进的实时目标检测器RT-DETRv2进行创新性架构改造,引入专用的概率头,使模型能为每个检测到的物体预测一个连续的成熟度分布,同时学习其均值及对应不确定性。该不确定性对机器人下游决策至关重要,可提供选择性采摘的置信度。实验表明,该方法不仅提供更丰富且符合生物学特性的成熟度表征,还保持优异检测性能,在一个具有挑战性的大规模水果数据集上达到85.6%的mAP。相比传统分类方法,本方案实现更细致、更准确的成熟度评估,为现代农业中智能、不确定性感知的自动化系统铺平道路。

原文摘要 · Abstract (English)

Maturity estimation of fruits and vegetables is a critical task for agricultural automation, directly impacting yield prediction and robotic harvesting. Current deep learning approaches predominantly treat maturity as a discrete classification problem (e.g., unripe, ripe, overripe). This rigid formulation, however, fundamentally conflicts with the continuous nature of the biological ripening process, leading to information loss and ambiguous class boundaries. In this paper, we challenge this paradigm by reframing maturity estimation as a continuous, probabilistic learning task. We propose a novel architectural modification to the state-of-the-art, real-time object detector, RT-DETRv2, by introducing a dedicated probabilistic head. This head enables the model to predict a continuous distribution over the maturity spectrum for each detected object, simultaneously learning the mean maturity state and its associated uncertainty. This uncertainty measure is crucial for downstream decision-making in robotics, providing a confidence score for tasks like selective harvesting. Our model not only provides a far richer and more biologically plausible representation of plant maturity but also maintains exceptional detection performance, achieving a mean Average Precision (mAP) of 85.6\% on a challenging, large-scale fruit dataset. We demonstrate through extensive experiments that our probabilistic approach offers more granular and accurate maturity assessments than its classification-based counterparts, paving the way for more intelligent, uncertainty-aware automated systems in modern agriculture

成熟度估计概率模型农业机器人目标检测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。