一种电气图元器件识别模型训练方法

本发明公开了一种电气图元器件识别模型训练方法，属于图像识别技术领域。该方法包括：获取多个电气图中所有电器元件样本图像，作为训练集；针对所有训练集图像进行分类，得到各训练子集；分别选取每个子集中1/m数量的图像组成第一次迭代的训练集；使用识别模型对第一次迭代的训练集做聚类，统计每一个训练子集下的准确率；针对准确率最低的训练子集，下一次选取其图像数量的2/m，其余训练子集选取图像数量的1/m组成下一次迭代的训练集，不断重复上述操作，直至训练好的识别模型。本发明解决了训练中的样本不平衡对模型收敛速度的影响，从而使得模型收敛速度更快，更好的进行电气图元器件的识别。