基于图神经网络解析平面图图像的线段
计算机视觉与模式识别
2023-03-08 v1
摘要
本文提出一种基于 GNN 的线段解析器（GLSP），其利用交点热图预测线段端点，并使用图神经网络提取线段及其类别。不同于以往依赖语义分割的平面图识别方法，我们所提方法能够输出矢量化的线段，且需要更少的后处理步骤即可投入实际使用。我们的实验表明，该方法在多类线段检测任务（使用平面图图像）上优于最先进的线段检测模型。本文中，我们使用名为大规模住宅平面图数据（LRFP）的平面图数据集。该数据集共包含 271,035 张平面图图像。每张图片对应的标签包含比例信息、房间的类别与轮廓，以及门、窗、墙等线段的端点位置。我们的增强方法使该数据集尽可能适应众多国家和地区出图的绘制风格。
引用
@article{arxiv.2303.03851,
title = {Parsing Line Segments of Floor Plan Images Using Graph Neural Networks},
author = {Mingxiang Chen and Cihui Pan},
journal= {arXiv preprint arXiv:2303.03851},
year = {2023}
}