4 个月前

整体吸引线框解析

整体吸引线框解析

摘要

本文提出了一种快速且简洁的解析方法,能够在单次前向传递中准确且稳健地检测输入图像中的矢量化线框。所提出的方法是端到端可训练的,由三个组件构成:(i) 线段和节点提议生成,(ii) 线段和节点匹配,(iii) 线段和节点验证。为了计算线段提议,提出了一种新颖的确切对偶表示方法,该方法利用了线段的简洁几何重参数化,并为输入图像形成了一个整体的四维吸引场图。节点可以被视为吸引场中的“盆地”。因此,所提出的方法被称为整体吸引线框解析器(HAWP)。在实验中,该方法在两个基准数据集上进行了测试:Wireframe 数据集和 YorkUrban 数据集。在这两个基准数据集上,它在准确性和效率方面均取得了最先进的性能。例如,在 Wireframe 数据集上,与之前的最先进方法 L-CNN 相比,它大幅提高了具有挑战性的平均结构精度(msAP)(绝对提高 2.8%),并在单个 GPU 上实现了 29.5 帧每秒(相对提高 89%)。此外,还进行了一项系统的消融研究以进一步证明所提出的方法的有效性。

代码仓库

cherubicXN/hawp
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
line-segment-detection-on-wireframe-datasetHAWP
FH: 83.1
sAP10: 66.5
sAP15: 68.2
sAP5: 62.5
line-segment-detection-on-york-urban-datasetHAWP
FH: 66.3
sAP10: 28.5
sAP15: 29.7
sAP5: 26.1

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
整体吸引线框解析 | 论文 | HyperAI超神经