arXiv:2602.13681cs.CVcs.AI2026-02被引 1

融合U-Net与FPN模型提升垃圾分割精度,助力智能回收。

An Ensemble Learning Approach towards Waste Segmentation in Cluttered Environment

  • 用加权平均法集成U-Net和FPN,兼顾细节与尺度适应性。
  • 在真实垃圾场景数据集上实现0.8306的IoU,优于单模型。
  • 适合工业级垃圾分类系统,可减少人工干预。

环境污染是全球性重大问题,回收利用是可行解决方案之一。本文聚焦垃圾分拣这一回收关键环节,旨在通过精准分割实现机器人对传送带上的垃圾准确定位与抓取。真实垃圾环境复杂,存在形变、无规律且重叠的物品,增加了分割难度。本研究提出一种集成学习方法(EL-4),结合擅长捕捉细粒度边界特征的U-Net与能有效处理尺度变化和上下文信息的FPN,采用加权平均融合其分割掩码。所用数据集贴近真实垃圾场景,经预处理增强深度学习模型的特征学习能力。实验结果表明,该集成模型达到0.8306的交并比(IoU),高于U-Net的0.8065;Dice损失降至0.09019,优于FPN的0.1183。该方法可提升物料回收设施的分拣效率,实现低人工干预下的高效原材料获取,提高整体处理通量。

原文摘要 · Abstract (English)

Environmental pollution is a critical global issue, with recycling emerging as one of the most viable solutions. This study focuses on waste segregation, a crucial step in recycling processes to obtain raw material. Recent advancements in computer vision have significantly contributed to waste classification and recognition. In waste segregation, segmentation masks are essential for robots to accurately localize and pick objects from conveyor belts. The complexity of real-world waste environments, characterized by deformed items without specific patterns and overlapping objects, further complicates waste segmentation tasks. This paper proposes an Ensemble Learning approach to improve segmentation accuracy by combining high performing segmentation models, U-Net and FPN, using a weighted average method. U-Net excels in capturing fine details and boundaries in segmentation tasks, while FPN effectively handles scale variation and context in complex environments, and their combined masks result in more precise predictions. The dataset used closely mimics real-life waste scenarios, and preprocessing techniques were applied to enhance feature learning for deep learning segmentation models. The ensemble model, referred to as EL-4, achieved an IoU value of 0.8306, an improvement over U-Net's 0.8065, and reduced Dice loss to 0.09019 from FPN's 0.1183. This study could contribute to the efficiency of waste sorting at Material Recovery Facility, facilitating better raw material acquisition for recycling with minimal human intervention and enhancing the overall throughput.

垃圾分割集成学习计算机视觉智能回收

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