基于无边缘检测 Transformer 的线段检测
计算机视觉与模式识别
2021-05-03 v2
摘要
在本文中，我们提出了一种使用 Transformers 的联合端到端线段检测算法，该算法无需后处理以及启发式引导的中间处理（边缘/连接点/区域检测）。我们的方法名为 LinE segment TRansformers (LETR)，利用 Transformers 中集成的令牌化查询、自注意力机制以及编码-解码策略，跳过了用于边缘元素检测和感知分组过程的标准启发式设计。我们为 Transformers 配备了多尺度编码器/解码器策略，以在直接端点距离损失下执行细粒度线段检测。该损失项特别适用于检测线段等不便用标准边界框表示来表征的几何结构。Transformers 通过学习逐层自注意力逐步细化线段。在我们的实验中，我们在 Wireframe 和 YorkUrban 基准上展示了最先进的结果。
引用
@article{arxiv.2101.01909,
title = {Line Segment Detection Using Transformers without Edges},
author = {Yifan Xu and Weijian Xu and David Cheung and Zhuowen Tu},
journal= {arXiv preprint arXiv:2101.01909},
year = {2021}
}
备注
Accepted to CVPR 2021 (Oral)