arXiv:2409.14976cs.CV2024-09被引 6

用简单架构实现顶尖边缘检测,不靠复杂设计也能超越现有方法。

A new baseline for edge detection: Make Encoder-Decoder great again

  • 双分支编码器分离位置与语义特征,提升特征压缩效率。
  • 级联融合解码器逐步优化位置特征,边缘图仅依赖精炼后结果。
  • 无需复杂训练或高算力,性能达新基准,适合轻量化部署。

基于深度学习的边缘检测器性能已超越人类,但高昂的计算成本和复杂的训练策略限制了其发展与应用。本文提出一种朴素的编码器-解码器结构,通过设计双边编码器,将位置特征与语义特征的提取过程解耦。由于位置分支不再为语义分支提供线索,特征丰富度可进一步压缩,从而实现模型更紧凑。我们提出级联特征融合解码器,让语义特征逐步优化位置特征,最终生成的边缘图仅依赖精炼后的位置特征,避免原始位置与语义特征直接参与输出。这有效抑制了位置特征中的噪声和语义特征中的定位误差。所提新基准(NBED)在多个边缘检测基准上表现优异,甚至超越部分高计算量、复杂训练策略的方法。在BSDS500上的ODS达到0.838,达到当前最优水平。研究表明,高质量特征才是关键,无需复杂设计,编码器-解码器架构仍可重振雄风。代码已开源。

原文摘要 · Abstract (English)

The performance of deep learning based edge detector has far exceeded that of humans, but the huge computational cost and complex training strategy hinder its further development and application. In this paper, we eliminate these complexities with a vanilla encoder-decoder based detector. Firstly, we design a bilateral encoder to decouple the extraction process of location features and semantic features. Since the location branch no longer provides cues for the semantic branch, the richness of features can be further compressed, which is the key to make our model more compact. We propose a cascaded feature fusion decoder, where the location features are progressively refined by semantic features. The refined location features are the only basis for generating the edge map. The coarse original location features and semantic features are avoided from direct contact with the final result. So the noise in the location features and the location error in the semantic features can be suppressed in the generated edge map. The proposed New Baseline for Edge Detection (NBED) achieves superior performance consistently across multiple edge detection benchmarks, even compared with those methods with huge computational cost and complex training strategy. The ODS of NBED on BSDS500 is 0.838, achieving state-of-the-art performance. Our study shows that what really matters in the current edge detection is high-quality features, and we can make the encoder-decoder based detector great again even without complex training strategies and huge computational cost. The code is available at https://github.com/Li-yachuan/NBED.

边缘检测轻量模型编码器-解码器

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