发明公开
- 专利标题: 基于YOLO目标检测框架的图元符号检测方法
-
申请号: CN202311366244.7申请日: 2023-10-20
-
公开(公告)号: CN117392697A公开(公告)日: 2024-01-12
- 发明人: 宋爱波 , 王天予 , 张勇 , 陈锐 , 苏猛猛 , 王清未
- 申请人: 东南大学 , 国家电网有限公司华北分部
- 申请人地址: 江苏省南京市江宁区东南大学路2号
- 专利权人: 东南大学,国家电网有限公司华北分部
- 当前专利权人: 东南大学,国家电网有限公司华北分部
- 当前专利权人地址: 江苏省南京市江宁区东南大学路2号
- 代理机构: 南京众联专利代理有限公司
- 代理商 许小莉
- 主分类号: G06V30/422
- IPC分类号: G06V30/422 ; G06V30/19 ; G06V10/82 ; G06N3/0464 ; G06N3/08
摘要:
本发明提供了一种基于YOLO目标检测框架的电气图元符号检测方法,该方法首先采集电气厂站图纸数据构造训练数据集;再设计规格自适应特征提取网络;最后网络输出得到图纸特征表示,并训练得到特征提取网络框架;使用通道注意力机制对YOLO目标检测框架进行改进,得到基于YOLO目标检测框架的图元符号检测模型,构造目标预测网络框架;初始化图元符号检测模型中的参数,并对特征提取网络框架和目标预测网络框架进行交替训练,对模型中的参数进行更新,得到训练完善的基于YOLO目标检测框架的图元符号检测模型。本发明可用于检测电气接线图纸中的各类电气图元符号,为电气接线图纸的矢量化工作乃至电网自动化调控提供基础。