4 个月前

DT-LSD:基于可变形Transformer的线段检测

DT-LSD:基于可变形Transformer的线段检测

摘要

线段检测是计算机视觉中的一个基本低级任务,该任务的改进可以影响依赖于它的更高级方法。目前,大多数新开发的线段检测方法都是基于卷积神经网络(CNNs)的。我们的论文旨在解决阻碍基于变压器(Transformer)方法在更大范围内应用的挑战。具体而言,我们引入了一种新的模型——可变形变压器基线段检测器(Deformable Transformer-based Line Segment Detection, DT-LSD),该模型支持跨尺度交互并且能够快速训练。本研究提出了一种新颖的可变形变压器基线段检测器(DT-LSD),解决了LETR模型的不足之处。为了加快训练速度,我们引入了线对比去噪(Line Contrastive DeNoising, LCDN)技术,该技术稳定了一对一匹配过程,并将训练速度提高了34倍。实验结果表明,DT-LSD比其前代变压器基模型(LETR)更快且更准确,并且在精度方面超过了所有基于CNN的模型。在Wireframe数据集中,DT-LSD分别达到了$sAP^{10}$ 71.7和$sAP^{15}$ 73.9;而在YorkUrban数据集中,则分别达到了$sAP^{10}$ 33.2和$sAP^{15}$ 35.1。

代码仓库

SebastianJanampa/DT-LSD
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
line-segment-detection-on-york-urban-datasetDT-LSD
sAP10: 33.2
sAP15: 35.1
sAP5: 30.2

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DT-LSD:基于可变形Transformer的线段检测 | 论文 | HyperAI超神经