基于深度学习的流程图线段检测方法OA
Deep learning-based method for detecting line segments in flowcharts
流程图线段检测受高质量数据集匮乏以及端点标签抗干扰弱等影响,导致该领域尚未得到充分探索.为此该文结合 YOLOv5 模型提出了基于标签重塑与双度量联合的流程图线段检测方法:首先,构建包含19 647 条线段的流程图线段数据集 FlowchartLine,提供丰富的训练样本;其次,提出端点引导的标签重塑模块,将流程图线段检测从传统端点检测问题转化为目标检测任务,有效缓解传统端点检测方法缺乏整体线段表征导致的多端点邻近误检问题;最后,提出一种几何感知增强联合损失,通过将归一化瓦斯坦距离(NWD)与原始完整交并比(CIoU)损失进行耦合,增强模型对线段检测的精细矫正能力.实验结果表明,该文方法在 FlowchartLine 流程图线段数据集上的 sAP 5、sAP 10、sAP 15结构平均精度指标分别为 98.2%、98.3%与 98.4%,优于当前线段检测方法1.6%、1.0%与0.9%,能够更准确地检测流程图线段.
Flowchart line segment detection has been underexplored due to the scarcity of high-quality datasets and the weak interference resistance of endpoint labels.To address these challenges,this article proposes a flowchart line segment detection method based on label reshaping and dual-metric combined with the YOLOv5 model.Firstly,the FlowchartLine dataset containing 19 647 line segments is constructed to provide abundant training samples.Secondly,an endpoint-guided label reshaping module is introduced to transform the task from traditional endpoint detection to object detection,effectively mitigating the false detection issues caused by adjacent endpoints in traditional endpoint detection methods that lack holistic line segment representation.Finally,a geometry-aware enhanced joint loss is developed by coupling normalized Wasserstein distance(NWD)with original complete intersection over union(CIoU)loss to improve precise correction capability for line segment detection.Experimental results demonstrate that the proposed method achieves structural average precision scores of 98.2%,98.3%and 98.4%at sAP5,sAP10 and sAP15 thresholds respectively,outperforming current line segment detection method by 1.6%,1.0%and 0.9%,demonstrating superior accuracy in flowchart line segment detection.
姚瑶;陈涛;孙泽人
南京理工大学 计算机科学与工程学院,江苏 南京 210094南京理工大学 计算机科学与工程学院,江苏 南京 210094南京理工大学 计算机科学与工程学院,江苏 南京 210094
信息技术与安全科学
流程图线段检测标签重塑目标检测YOLOv5归一化瓦斯坦距离
flowchartline segment detectionlabel reshapingobject detectionYOLOv5normalized Wasserstein distance
《南京理工大学学报(自然科学版)》 2026 (2)
161-171,11
国家自然科学基金(62506169)中央高校基本科研业务费专项资金(30923010303)
评论