[发明专利]一种用于风机图像故障标识的候选框调整方法

权利要求书

1.一种用于风机图像故障标识的候选框调整方法，其特征在于，所述风机图像包括可见光图像和红外热图像，所述方法包括以下步骤：

S1、候选框坐标变换：基于传统图像识别算法得到候选框集合B，设计坐标变换矩阵和变换公式c_i＝b_i*a_4*4，将其变换为集合C；其中，B＝{b_i}，b_i＝(x_i,y_i,w_i,h_i)，(x_i,y_i)为候选框b_i中心点的直角坐标，(w_i,h_i)为b_i候选框的长和宽；C＝{c_i}，c_i＝(c_i1,c_i2,c_i3,c_i4)，(c_i1,c_i2)为候选框c_i左上角的直角坐标，(c_i3,c_i4)为候选框c_i右下角的直角坐标；

S2、调整参数a_ki计算：基于传统图像识别算法中的NMS，设计候选框的调整参数a_ki(g_k,c_i)＝relu(s_k+s_i-1-GIOU(g_k,c_i))*((c_i1-g_k1),(c_i2-g_k2),(c_i3-c_k3),(c_i4-g_k4))，a_ki为四维向量，s为候选框的置信分数，候选框g_k＝(g_k1,g_k2,g_k3,g_k4)，(g_k1,g_k2)为候选框g_k左上角的直角坐标，(g_k3,g_k4)为候选框g_k右下角的直角坐标。

S3、候选框g_k调整：候选框调整公式为为两个四维向量的点乘，α为经验调节系数，A_4*4为交叉矩阵；

S4、可见光图像和红外热图像候选框互补：对比两种图像的故障候选框，在可见光图像中，在没有被可见光图像候选框覆盖而被红外热图像候选框覆盖的地方添加候选框；在红外热图像中，在没有被红外热图像候选框覆盖而被可见光图像候选框覆盖的地方添加候选框。

2.根据权利要求1所述的一种用于风机图像故障标识的候选框调整方法，其特征在于，在步骤S1中所述的传统图像识别算法的步骤包括图像去模糊、增强和去噪，滑动窗口法选取候选框，HOG特征提取和SVM分类。

3.根据权利要求1所述的一种用于风机图像故障标识的候选框调整方法，其特征在于，在步骤S2所述的a_ki的计算过程为：将集合C中置信分数最高的候选框c_j放入集合G中并标记为g_k，删去集合C中的c_j且计算候选框g_k与集合C中剩余候选框c_i的GIOU(g_k,c_i)，若GIOU(g_k,c_i)≥0.5，则删去集合C中的c_i且计算调整参数a_ki(g_k,c_i)，重复此步骤直至集合其中，GIOU的计算公式为relu的计算公式为