-
公开(公告)号:CN118037710A
公开(公告)日:2024-05-14
申请号:CN202410325056.8
申请日:2024-03-21
Applicant: 南京邮电大学
IPC: G06T7/00 , G06V30/148 , G06V10/762 , G06V10/774 , G06V10/80 , G06V10/82 , G06V10/25
Abstract: 本发明提出一种基于扩散模型的变电站光缆图智能解析方法、装置及存储介质,包括:对待识别的变电站光缆图进行预处理,获得处理后的图纸;标注处理后的图纸数据,构建训练集、验证集和测试集;构建图元检测模型,使用基于Swin transformer和EFPN网络的主干网络,并在解码器中加入CBAM注意力机制;训练集训练构建的图元检测模型,验证集进行最佳模型选择,得到训练后的图元检测模型;将测试集的图纸输入训练好的图元检测模型和字符识别模型;将图元检测结果和字符识别结果进行匹配,输出最终结果。采用在扩散模型基础上进一步修改DiffusionDet网络结构的目标检测方法,并提出融合拓扑匹配和聚类分析的配对算法,使得变电站光缆图的解析工作更加智能高效且具有较好的可迁移性。