顶刊TPAMI 2021！南开大学提出深度霍夫变换：语义线检测新方法

Deep Hough Transform for Semantic Line Detection

主页：https://mmcheng.net/dhtline/

论文：http://mftp.mmcheng.net/Papers/21PAMI-DHT-line.pdf

代码：https://github.com/Hanqer/deep-hough-transform

resnet50起步，模型104m。

 

南开大学计算媒体实验室在机器学习顶刊 IEEE TPAMI 发表论文提出”深度霍夫变换“ (deep hough transform, 简称 DHT)，通过在深度特征上进行霍夫变换，从而兼顾CNN的特征学习能力和霍夫变换的高效性。该论文同时收集了一个包含室内/室外场景的大型语义线检测数据集 NKL （NanKai Lines）。NKL数据集有6,500张图像以及超过1万根语义线标注，为该领域目前最大的公开数据集。

 

该论文第一作者为原南开大学博士生，现腾讯优图实验室赵凯研究员；通讯作者为南开大学程明明教授。南开大学硕士研究生韩琦为文章共同第一作者。

 

语义线

 

检测图像中的直线（线段）是一个历史悠久的计算机视觉问题，图像中的线结构是一种有用的中间层表示（mid-level representation），在人类视觉中从底层到高层语义信息转换中有着重要作用 [1]。

 

自然图像中的“语义线”指的是图像中能够勾勒出图像内容结构的直线，例如不同区域的分割线，建筑物的中轴线等等。语义线检测在摄影构图[2, 3]，图像处理[4]等下游任务中有着广泛应用。如图1所示，将图像中的语义线置于照片的黄金分割比位置能拍出视觉效果更好的图片 [2]。       

 

图1：左图为原图，右图将语义线置于画面的黄金分割比位置，能得到视觉效果更好的照片[2]。

 

深度霍夫变换算法

 

霍夫变换[6]是一种经典的直线检测方法，它通过将任意直线转换到参数空间，从而将图像空间的线检测问题简化为参数空间的点检测问题。例如，平面上的任意直线都可以用  y = k x + b 来表示，因此任意一条直线都可以表示为参数空间中的一个点(k, b)。

 

传统的霍夫变换需要先用边缘检测算法（例如canny算子）对图像进行边缘检测，然后在边缘检测的结果之上进行霍夫变换，最后在变换空间通过寻找局部最大值来检测直线。

 

直线参数化

 

为了进行霍夫变换，首先要定义直线参数化的规则。 

图2：直线的参数化

 

如上图所示，对任意一条直线，用其角度 theta 和直线到原点的距离 r 作为直线的参数，这样做的好处是可以参数化任意角度/位置的直线（用y = kx + b 无法参数化与 y 轴平行的直线）。

 

在深度特征上进行霍夫变换

 

如图3所示，对于一个输入图片，先通过CNN 特征提取器得到空间特征X，“深度霍夫变换”遍历图像中所有可能的直线并沿着直线将特征聚合到参数空间中对应的点上。在实现中，使用简单的求和操作实现特征聚合。

         图3  左：深度霍夫变换沿着图像空间中的直线将特征”聚合“到参数空间中的一个点。

 

经过深度霍夫变换，原特征空间 X 上的一根直线对应参数空间 Y 上的一个点。在很多视觉任务中，特征的上下文（context）对于检测结果十分重要，深度霍夫变换将直线转化为点，因此提取相邻直线的上下文信息对应于在参数空间中提取相邻点的特征，大大简化和上下文信息提取的效率。

       图4：图像空间中的线对应参数空间的一个点，一个简单的 3x3 卷积即可提取中心橙色线的上下文信息。

 

如图4所示，在原图像中提取橙色线的上下文信息需要复杂的索引操作；经过霍夫变换之后，只需要一个简单的 3x3 卷积即可提取中心线周围直线的特征。

 

总的来说，DHT 算法有如下几个优点：

1. 高度可并行化，相比基于目标检测的SLNet方法加速了接近20 倍；

2. 无需非极大值抑制；

3. 高效的上下文直线信息提取；

 

实验结果

 

定量对比

 

论文主要在 SEL 和 NKL 数据集上进行试验，对比了 HED边缘检测 + 霍夫变换，SLNet 两种语义线检测方法。实验结果表明深度霍夫变换算法检测精度显著高于其他两种算法，同时大幅提高检测速度。

       表1：不同方法在 SEL 和 NKL 数据集上的定量对比。

 

表1记录了不同方法在 SEL 和 NKL 数据集上的定量对比。CD，EMD 和 EA 分别指的是 Chamfer distance， earth-mover's distance 和 EA-score 三种评价直线之间相似度的度量。Avg P, R, F分别表示平均 precision，recall 和 F-measure。

 

可视化结果展示

 

 

上图是SEL数据集上不同方法检测结果的可视化对比，最后一行为 ground-truth。

 

推理耗时对比

 

DHT算法不仅能取得更高的检测精度，同时也能极大地降低检测耗时。这主要归功于 DHT 算法的高度可并行化。

 

下表统计了不同方法在 Titan XP 显卡上的平均推理耗时，SLNet 最快需要0.447秒，HED+霍夫变换的方法需要0.155秒，而DHT 算法仅需0.020秒。

                

NKL：新的语义线检测数据集

 

在提出深度霍夫变换算法的同时，改论文还同时构建了新的语义线检测数据集 NKL （NanKai Lines）。NKL数据集包含6,500张样本，共有13,148根人工标注的语义线。相比当前最大的语义线检测数据集 SEL，NKL不论是在样本数，标注数目还是场景多样性上都要更丰富。读者可以访问 http://kaizhao.net/nkl 下载和了解 NKL 数据集相关的信息。

       表2：SEL 数据集和 NKL 数据集对比。

 

为了验证 NKL 数据集在图像场景中的多样性，该论文使用在 Places365 数据集上预训练的模型分别对 NKL 和 SEL 数据集中的样本进行推理，预测的分类结果如图5所示：

       图5：SEL和论文新构建的 NKL 数据集的场景分布

 

从图5可以看出，SEL 的大部分样本集中在天空、场地等少数场景，而NKL数据集中样本的分布更为多样化，同时场景类别也更多。

 

使用“计图”框架加速推理

 

该论文同时也提供了基于清华大学自主研发的“计图”（jittor）框架的推理代码，得益于Jittor框架“元算子融合”和“统一计算图”两项新技术，DHT 算发无需任何优化情况下推理速度相比PyTorch 就有显著提升。

 

在 Titan XP 显卡上使用 batch size=4进行前向预测，PyTorch 框架下的 FPS 为 48，而同条件下计图可达到每秒 55 帧。

 

 

[1] J. B. Burns, A. R. Hanson, and E. M. Riseman, “Extracting straight lines,” IEEE Trans. Pattern Anal. Mach. Intell., vol. 8, no. 4, pp. 425–455, 1986.

[2] L. Liu, R. Chen, L. Wolf, and D. Cohen-Or, “Optimizing photo composition,” Comput. Graph. Forum, vol. 29, no. 2, pp. 469–478, 2010.

[3] M. Freeman, The photographer’s eye: composition and design for better digital photos. CRC Press, 2007

[4] M.-M. Cheng, X.-C. Liu, J. Wang, S.-P. Lu, Y.-K. Lai, and P. L. Rosin, “Structure-preserving neural style transfer,” IEEE Trans. Image Process., vol. 29, pp. 909–920, 2020.

[5] J.-T. Lee, H.-U. Kim, C. Lee, and C.-S. Kim, “Semantic line detection and its applications,” in Int. Conf. Comput. Vis., 2017, pp. 3229–3237.

[6] R. O. Duda and P. E. Hart, “Use of the hough transformation to detect lines and curves in pictures,” Sri International Menlo Park Ca Artificial Intelligence Center, Tech. Rep., 1971.

[7] Hu, Shi-Min, et al. "Jittor: a novel deep learning framework with meta-operators and unified graph execution." Science China Information Sciences 63.12 (2020): 1-21.