arXiv:2505.22105cs.CV2025-05被引 1

改进SAM模型,精准分割高压线路走廊中的隐患目标

Adapting Segment Anything Model for Power Transmission Corridor Hazard Segmentation

  • 引入上下文感知提示适配器,融合全局与局部特征
  • 设计高保真掩码解码器,提升细结构目标分割精度
  • 构建首个大规模真实场景数据集,支持电力隐患检测

高压线路走廊隐患分割(PTCHS)旨在从复杂背景中分离输电设备与周边隐患,对保障电力传输安全具有重要意义。近期,通用分割模型SAM在分割任务中表现突出,但在复杂输电走廊场景中对细结构目标分割效果不佳。本文提出ELE-SAM,针对该任务进行适配:设计上下文感知提示适配器,通过融合全局-局部特征并聚焦关键区域生成更优提示;构建高保真掩码解码器,利用多粒度掩码特征并上采样至更高分辨率以增强细结构表达。为训练与推动该领域发展,构建首个大规模真实世界数据集ELE-40K,包含44,094张图像-掩码对。在该数据集上的实验表明,ELE-SAM相比基线模型平均提升16.8% mIoU和20.6% mBIoU;相较于HQSeg-44K上的先进方法,分别实现2.9% mIoU和3.8% mBIoU的绝对提升,验证了方法在高质量通用目标分割上的有效性。代码与数据集已开源。

原文摘要 · Abstract (English)

Power transmission corridor hazard segmentation (PTCHS) aims to separate transmission equipment and surrounding hazards from complex background, conveying great significance to maintaining electric power transmission safety. Recently, the Segment Anything Model (SAM) has emerged as a foundational vision model and pushed the boundaries of segmentation tasks. However, SAM struggles to deal with the target objects in complex transmission corridor scenario, especially those with fine structure. In this paper, we propose ELE-SAM, adapting SAM for the PTCHS task. Technically, we develop a Context-Aware Prompt Adapter to achieve better prompt tokens via incorporating global-local features and focusing more on key regions. Subsequently, to tackle the hazard objects with fine structure in complex background, we design a High-Fidelity Mask Decoder by leveraging multi-granularity mask features and then scaling them to a higher resolution. Moreover, to train ELE-SAM and advance this field, we construct the ELE-40K benchmark, the first large-scale and real-world dataset for PTCHS including 44,094 image-mask pairs. Experimental results for ELE-40K demonstrate the superior performance that ELE-SAM outperforms the baseline model with the average 16.8% mIoU and 20.6% mBIoU performance improvement. Moreover, compared with the state-of-the-art method on HQSeg-44K, the average 2.9% mIoU and 3.8% mBIoU absolute improvements further validate the effectiveness of our method on high-quality generic object segmentation. The source code and dataset are available at https://github.com/Hhaizee/ELE-SAM.

图像分割电力巡检SAM适配细结构分割

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