提升分割边界精度,解决小目标模糊问题
A Deep Learning Framework for Boundary-Aware Semantic Segmentation
- 基于Mask2Former构建边界感知特征图,引入跨尺度特征桥接机制
- 在Cityscapes上mIOU、mDICE、mRecall均显著提升,边界保留更优
- 适合自动驾驶、医疗图像等高精度分割场景
语义分割是计算机视觉中的基础任务,广泛应用于自动驾驶、遥感图像分析和医学图像处理。近年来,基于Transformer的分割方法在全局特征建模上表现优异,但仍存在目标边界模糊、小目标识别不足的问题。为此,本文提出一种基于Mask2Former的语义分割算法,引入边界增强特征桥接模块(BEFBM),旨在提升目标边界精度与分割一致性。该方法在Mask2Former框架基础上构建边界感知特征图,并引入特征桥接机制,实现有效跨尺度特征融合,增强模型对目标边界的关注能力。在Cityscapes数据集上的实验表明,相比主流分割方法,该方法在mIOU、mDICE和mRecall等指标上均有显著提升,且在复杂场景中表现出更优的边界保持能力。视觉分析进一步验证了模型在细粒度区域的优势。未来研究将聚焦计算效率优化,并探索其在其他高精度分割任务中的潜力。
原文摘要 · Abstract (English)
As a fundamental task in computer vision, semantic segmentation is widely applied in fields such as autonomous driving, remote sensing image analysis, and medical image processing. In recent years, Transformer-based segmentation methods have demonstrated strong performance in global feature modeling. However, they still struggle with blurred target boundaries and insufficient recognition of small targets. To address these issues, this study proposes a Mask2Former-based semantic segmentation algorithm incorporating a boundary enhancement feature bridging module (BEFBM). The goal is to improve target boundary accuracy and segmentation consistency. Built upon the Mask2Former framework, this method constructs a boundary-aware feature map and introduces a feature bridging mechanism. This enables effective cross-scale feature fusion, enhancing the model's ability to focus on target boundaries. Experiments on the Cityscapes dataset demonstrate that, compared to mainstream segmentation methods, the proposed approach achieves significant improvements in metrics such as mIOU, mDICE, and mRecall. It also exhibits superior boundary retention in complex scenes. Visual analysis further confirms the model's advantages in fine-grained regions. Future research will focus on optimizing computational efficiency and exploring its potential in other high-precision segmentation tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。