Invention Publication
- Patent Title: 一种基于预训练的图纸元件识别方法、装置、介质及设备
-
Application No.: CN202311135594.2Application Date: 2023-09-05
-
Publication No.: CN117218674APublication Date: 2023-12-12
- Inventor: 王海楠 , 唐一铭 , 单华 , 宋煜 , 黄祥 , 陈玉权 , 张欣 , 王浩羽 , 顾臻烨
- Applicant: 江苏方天电力技术有限公司
- Applicant Address: 江苏省南京市江宁科学园天元中路19号
- Assignee: 江苏方天电力技术有限公司
- Current Assignee: 江苏方天电力技术有限公司
- Current Assignee Address: 江苏省南京市江宁科学园天元中路19号
- Agency: 南京纵横知识产权代理有限公司
- Agent 许婉静
- Main IPC: G06V30/422
- IPC: G06V30/422 ; G06V10/774

Abstract:
本发明公开了一种基于预训练的图纸元件识别方法、装置、介质及设备方法包括:获取多个类别的元件图像信息和待识别的工程图纸;将待识别的工程图纸输入预先训练的元件识别网络,进行元件识别,得到待识别元件的类别信息和坐标信息;其中,预先训练的元件识别网络,通过以下步骤得到:基于yolov5的网络结构生成初始元件识别网络,所述初始元件识别网络包括特征提取组件网络、特征混合组件网络和预测组件网络;基于随机生成的工程图纸对所述初始元件识别网络进行训练,得到预先训练的元件识别网络。本发明能够在只存在少量工程图纸样本的情况下,实现电气元件的目标识别。
Information query