[发明专利]一种基于图像识别的电力物资铭牌检测方法

权利要求书

1.一种基于图像识别的电力物资铭牌检测方法，其特征在于，包括以下步骤：

步骤1：获取电力物资铭牌图像；

步骤2：基于获取的电力物资铭牌图像，基于SSD算法对信息缺失区域进行框定；

步骤3：基于矩阵分解对信息缺失区域进行信息补充；

步骤4：对信息补充后的铭牌图像进行文字的OCR。

2.根据权利要求1所述的一种基于图像识别的电力物资铭牌检测方法，其特征在于，步骤2中基于SSD算法对信息缺失区域进行框定具体包括以下步骤：

步骤2-1：搭建SSD网络；

步骤2-2：，按预设参数训练SSD网络，针对SSD网络中VGG-16部分进行训练，训练的预设参数包括：权重下降率，学习率，学习率下降因子以及训练批次大小；

步骤2-3：当训练网络收敛到预设值时，对网络进行微调，需要进行微调的参数包括：学习率和批次大小。

3.根据权利要求1所述的一种基于图像识别的电力物资铭牌检测方法，其特征在于，步骤3包括以下步骤：

步骤3-1：将图像灰度值转为矩阵信息，并标记步骤2中的框定区域，公式如下：

其中，V为步骤1中原始图像的矩阵信息，W为由步骤2确定的信息缺失矩阵，具体定义如下：

W＝ω_ij，i∈[1，m]，j∈[1，n]    (2)

其中，ω_ij为信息缺失矩阵的元素，m为矩阵W的行数，n为矩阵W的列数，若图像中位置ij的信息缺失，则ω_ij＝0，否则ω_ij＝1。

4.根据权利要求3所述的一种基于图像识别的电力物资铭牌检测方法，其特征在于，步骤3包括以下步骤：

步骤3-2:构建矩阵分解函数：

其中，矩阵E和矩阵F为待求的分解矩阵；为构建的补充信息矩阵。

5.根据权利要求4所述的一种基于图像识别的电力物资铭牌检测方法，其特征在于，步骤3包括以下步骤：

步骤3-3:构建矩阵分解的目标函数：

6.根据权利要求5所述的一种基于图像识别的电力物资铭牌检测方法，其特征在于，步骤3包括以下步骤：

步骤3-4：采用ADMM算法对公式(4)中的目标函数进行求解。

7.根据权利要求6所述的一种基于图像识别的电力物资铭牌检测方法，其特征在于，步骤3包括以下步骤：

步骤3-5：通过步骤3-4得到的矩阵E,F计算得到

8.根据权利要求7所述的一种基于图像识别的电力物资铭牌检测方法，其特征在于，步骤4包括以下步骤：

步骤4-1：基于步骤3得到的信息矩阵转为图像灰度值，同时对铭牌图像进行角度校正和去噪；

步骤4-2：将文字行进行列分割，得到字符；

步骤4-3：将字符输入OCR识别模型得到设备铭牌内容。

9.根据权利要求8所述的一种基于图像识别的电力物资铭牌检测方法，其特征在于，步骤3-4求解公式为：

为损失函数。

10.根据权利要求9所述的一种基于图像识别的电力物资铭牌检测方法，其特征在于，

步骤3-4具体包括如下步骤：

输入：正则化参数λ₀,λ₁，最大迭代个数max_iter，惩罚系数ρ，辅助变量z，k为迭代次数，第k次；

迭代过程中的拉格朗日乘子向量u^k，原始残差收敛精度ε^prim，对偶残差收敛精度ε^dual，初始化参数σ；

输出：矩阵E,F；