2 个月前
TP-LSD:基于三顶点的线段检测器
Huang, Siyu ; Qin, Fangbo ; Xiong, Pengfei ; Ding, Ning ; He, Yijia ; Liu, Xiao

摘要
本文提出了一种新颖的深度卷积模型——基于三点的线段检测器(TP-LSD),能够在实时速度下检测图像中的线段。以往的相关方法通常采用两步策略,依赖于启发式后处理或额外的分类器。为了实现更快、更紧凑的一步检测,我们引入了三点表示法,将线段检测转化为对每个线段的根点和两个端点进行端到端预测。TP-LSD 模型包含两个分支:三点提取分支和线段分割分支。前者预测根点的热图以及两个端点的位移图;后者从背景中分割出直线上的像素。此外,线段分割图在第一个分支中被重新利用作为结构先验。我们提出了一种新的评估指标,并在 Wireframe 和 YorkUrban 数据集上评估了该方法,不仅展示了与最新方法相当的精度,还实现了高达 78 帧每秒(FPS)的实时运行速度,输入分辨率为 $320 \times 320$。