arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2409.00909cs.CVcs.AI
收藏

ViRED:工程图纸中视觉关系的预测

ViRED: Prediction of Visual Relations in Engineering Drawings

University of Science and Technology of China(中国科学技术大学);School of Software, Tsinghua University(清华大学软件学院);Nanyang Technological University(南洋理工大学)

Chao Gu, Ke Lin, Yiyang Luo, Jiahui Hou, Xiang-Yang Li

AI总结:

本文提出基于视觉的关系检测模型ViRED,通过视觉编码器、对象编码器和关系解码器识别电气工程图纸中表格与电路的关联,在关系预测任务上达到96%准确率并实现快速推理。

摘要:

为了准确理解工程图纸,必须建立图纸中图像与其描述表之间的对应关系。现有的文档理解方法主要将文本作为主要模态,这不适用于包含大量图像信息的文档。在视觉关系检测领域,任务的结构本质上限制了其评估图纸中所有实体对之间关系的能力。为解决这一问题,我们提出了一种基于视觉的关系检测模型,名为ViRED,用于识别电气工程图纸中表格与电路之间的关联。我们的模型主要由三部分组成:视觉编码器、对象编码器和关系解码器。我们使用PyTorch实现ViRED以评估其性能。为验证ViRED的有效性,我们进行了一系列实验。实验结果表明,在工程图纸数据集上,我们的方法在关系预测任务中达到了96%的准确率,相较于现有方法有显著提升。结果还表明,即使单张工程图纸中包含大量对象,ViRED也能以较快的速度进行推理。

英文摘要:

To accurately understand engineering drawings, it is essential to establish the correspondence between images and their description tables within the drawings. Existing document understanding methods predominantly focus on text as the main modality, which is not suitable for documents containing substantial image information. In the field of visual relation detection, the structure of the task inherently limits its capacity to assess relationships among all entity pairs in the drawings. To address this issue, we propose a vision-based relation detection model, named ViRED, to identify the associations between tables and circuits in electrical engineering drawings. Our model mainly consists of three parts: a vision encoder, an object encoder, and a relation decoder. We implement ViRED using PyTorch to evaluate its performance. To validate the efficacy of ViRED, we conduct a series of experiments. The experimental results indicate that, within the engineering drawing dataset, our approach attained an accuracy of 96\% in the task of relation prediction, marking a substantial improvement over existing methodologies. The results also show that ViRED can inference at a fast speed even when there are numerous objects in a single engineering drawing.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑