电网厂站接线图的图元检测方法、系统、设备及介质
摘要:
本发明公开了一种电网厂站接线图的图元检测方法、系统、设备及介质。该方法包括:获取训练图片,基于训练图片的全标注框生成标注信息,对训练图片和标注信息进行格式转化,得到框标注和点标注;利用随机梯度下降算法对主干特征提取网络进行训练,得到优化主干网络;基于优化主干网络提取训练图片中的特征图,根据特征图与点标注对PARPN网络进行训练,生成候选目标区域;根据优化主干网络、候选目标区域与框标注对识别网络进行训练,得到图元检测网络模型;获取识别数据,将识别数据输入至图元检测网络模型,生成图元检测结果。本申请设计一个专用于接线图图元的图元检测网络模型,提升了在接线图中对各个类别图元的检测和识别能力。
0/0