基于YOLO目标检测框架的图元符号检测方法
摘要:
本发明提供了一种基于YOLO目标检测框架的电气图元符号检测方法,该方法首先采集电气厂站图纸数据构造训练数据集;再设计规格自适应特征提取网络;最后网络输出得到图纸特征表示,并训练得到特征提取网络框架;使用通道注意力机制对YOLO目标检测框架进行改进,得到基于YOLO目标检测框架的图元符号检测模型,构造目标预测网络框架;初始化图元符号检测模型中的参数,并对特征提取网络框架和目标预测网络框架进行交替训练,对模型中的参数进行更新,得到训练完善的基于YOLO目标检测框架的图元符号检测模型。本发明可用于检测电气接线图纸中的各类电气图元符号,为电气接线图纸的矢量化工作乃至电网自动化调控提供基础。
0/0