4 个月前

端到端线框解析

端到端线框解析

摘要

我们提出了一种概念上简单但效果显著的算法,用于检测给定图像中的线框。与先前的方法(这些方法首先预测一个中间热图,然后通过启发式算法提取直线)相比,我们的方法可以端到端训练,并且能够直接输出包含语义上有意义且几何上显著的节点和线条的矢量化线框。为了更好地评估输出的质量,我们提出了一种新的线框评价指标,该指标对重叠的线段和错误的线条连接进行惩罚。我们进行了广泛的实验,并展示了我们的方法在性能上显著优于现有的最先进的线框和线条提取算法。我们希望这一简单的算法能够作为未来线框解析研究的一个基准。代码已公开发布在 https://github.com/zhou13/lcnn。

代码仓库

zhou13/lcnn
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
line-segment-detection-on-wireframe-datasetL-CNN
sAP10: 62.9
sAP15: 64.7
sAP5: 58.9
line-segment-detection-on-york-urban-datasetL-CNN
sAP10: 26.4
sAP5: 24.3

用 AI 构建 AI

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

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

Hyper Newsletters

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