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