一种用于风机图像故障标识的候选框调整方法技术

本发明专利技术涉及一种用于风机图像故障标识的候选框调整方法，所述风机图像包括可见光图像和红外热图像，所述调整方法包括：候选框坐标变换；调整参数a

【技术实现步骤摘要】
一种用于风机图像故障标识的候选框调整方法


[0001]本专利技术涉及图像识别
，具体涉及一种用于风机图像故障标识的候选框调整方法。

技术介绍

[0002]随着无人机及其相关技术的快速发展，无人机被广泛地应用于各种事故和故障的巡检中。使用无人机搭配摄像机、红外热像仪或激光雷达等设备对风机的各个部位拍摄图像，然后由图像识别算法对无人机拍摄的图像进行故障识别和标记，是当前无人机领域的研究重点，可以有效地提高风机的巡检效率和减少巡检人员的工作量。然而，传统图像识别算法特征提取能力较弱导致故障识别精度较低，基于深度学习的图像识别算法精度高速度快但却需要大量的训练样本。
[0003]本专利技术提出一种用于风机图像故障标识的候选框调整方法，旨在风机图像中的故障训练样本远远不足的情况下，设计一个启发式算法调整传统图像识别算法中的候选框，使其能够更好地选中目标即风机故障，从而提高传统图像识别算法的精度，为基于深度学习的图像识别算法积累训练样本。

技术实现思路

[0004]本专利技术针对现有的传统图像识别算法中候选框精度不足的问题，提出一种用于风机图像故障标识的候选框调整方法，通过设计启发式算法，对候选框的左上角和右下角坐标进行调整，并采用可见光图像和红外热图像的故障候选框互补的方式，使候选框能够更好地选中风机故障，从而提高故障识别的精度。
[0005]本专利技术公开了一种用于风机图像故障标识的候选框调整方法，所述风机图像包括可见光图像和红外热图像，包括以下步骤：
[0006]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
右下角的直角坐标；
[0007]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
右下角的直角坐标；
[0008]S3、候选框g
k
调整：候选框调整公式为调整：候选框调整公式为为两个四维向量的点乘，α为经验调节系数，A
4*4
为交叉矩阵；
[0009]S4、可见光图像和红外热图像候选框互补：对比两种图像的故障候选框，在可见光图像中，在没有被可见光图像候选框覆盖而被红外热图像候选框覆盖的地方添加候选框；在红外热图像中，在没有被红外热图像候选框覆盖而被可见光图像候选框覆盖的地方添加候选框。
[0010]可选的，所述的方法在步骤S1中所述的传统图像识别算法的步骤包括预处理、生成候选框、特征提取和分类器分类；所述的预处理包括去模糊、图像增强和去噪处理，生成候选框的方法包括选择性搜素、滑动窗口法和边缘方框法，特征提取的方法包括SIFI特征提取法、HOG特征提取法、Haar
‑
like特征提取法和RIFT特征提取法，分类器分类的方法包括SVM分类方法。
[0011]进一步的，所述的方法在步骤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的计算公式为
[0012]进一步的，所述的方法在步骤S3中所述的经验调节系数α与生成候选框时采取的尺寸和目标的形状有关，一般情况下可令α＝(1,1,1,1)；所述交叉矩阵
[0013]可选的，所述的一种用于风机图像故障标识的候选框调整方法，其中，所述故障标识类型包括塔筒的锈迹、掉漆、破损、变形，机舱的连接螺栓脱落、溢油、雷击点，主机体的零件脱落、掉漆、溢油，叶片的掉漆、破损、砂眼、裂纹、雷击点。
[0014]有益效果：本专利技术提供一种用于风机图像故障标识的候选框调整方法，使用传统图像识别算法在进行候选框极大值抑制时使用到的数据，设计启发式算法对候选框进行调整，可以在几乎不对原有的算法计算速度造成影响的情况下，提高算法识别的精度。
附图说明
[0015]图1是本专利技术的步骤流程图。
[0016]图2是本专利技术使用后的第一种前后风机图像候选框的变化。
[0017]图3是本专利技术使用后的第二种前后风机图像候选框的变化
具体实施方式
[0018]下面将结合本专利技术实施例中的附图，对本专利技术实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本专利技术一部分实施例，而不是全部的实施例。基于本专利技术中的实施例，本领域普通技术人员在没有做出创造性劳动的前提下所获得的其它实施例，都属于本专利技术保护的范围。<本文档来自技高网...

【技术保护点】

【技术特征摘要】
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
...

【专利技术属性】
技术研发人员：杨苹，张淼，王仁杰，刘泽健，范紫微，
申请(专利权)人：深圳华工能源技术有限公司，
类型：发明
国别省市：