[发明专利]一种基于Phash算法结合深度学习的线路螺栓缺陷识别方法

权利要求书

1.一种基于Phash算法结合深度学习的线路螺栓缺陷识别方法，其特征在于，包括：

步骤S1：收集输电线路的螺栓图像，并在所述螺栓图像上框选出螺栓位置，将框选出的所述螺栓位置的图像分割提取，作为初始样本集；

步骤S2：通过Phash算法判断所述初始样本集中的螺栓缺失情况，选择相似度阈值；

步骤S3：将所述初始样本集中螺栓未缺失的图片作为基础样本集，对所述基础样本集进行样本量的扩充，得到扩充样本集；

步骤S4：通过深度学习构建用于识别螺栓缺陷的Faster R-CNN网络模型，将所述扩充样本集作为训练数据并对所述Faster R-CNN网络模型进行训练，将训练完成的所述FasterR-CNN网络模型放置在CPU上；

步骤S5：利用无人机拍摄所述输电线路的图片并作为待检测图像，通过Phash算法判断所述待检测图像中螺栓缺失情况，将所述待检测图像输入至训练完成后的所述Faster R-CNN网络模型，利用所述Faster R-CNN网络模型判断所述待检测图像中螺栓缺陷情况。

2.如权利要求1所述的基于Phash算法结合深度学习的线路螺栓缺陷识别方法，其特征在于，在所述步骤S2中，判断初始样本集中的螺栓缺失情况，若判断所述初始样本集中一图片存在螺栓缺失，则标记前述图片，定义为螺栓缺失；若判断所述初始样本集中另一图片不存在螺栓缺失，则执行所述步骤S3。

3.如权利要求1所述的基于Phash算法结合深度学习的线路螺栓缺陷识别方法，其特征在于，在所述步骤S2中，判断初始样本集中螺栓缺失情况的具体步骤包括：

步骤S201：将所述初始样本集中的任一待测图片进行灰度化处理，得到灰度图像，对所述灰度图像进行DCT变化，得到64*64的DCT系数矩阵，截取所述64*64的DCT系数矩阵中左上角16*16的DCT矩阵gary(i,j)，计算所述DCT矩阵gary(i,j)的平均值Av，其计算公式如下：

步骤S202：建立16*16的零矩阵zero(i,j)，将所述DCT矩阵gary(i,j)中各个数值与所述矩阵的平均值Av进行比较；

所述DCT矩阵gary(i,j)中一数值大于所述平均值Av，则将对应的所述零矩阵zero(i,j)中的数值改为1；所述DCT矩阵gary(i,j)中一数值不大于所述平均值Av，则将对应的所述零矩阵zero(i,j)中的数值改为0，

所述零矩阵zero(i,j)公式为：

所述零矩阵zero(i,j)输出16*16的所述待测图片的二值图像，其中，数值为1，则显示白色；数值为0，则显示黑色；

步骤S203：对已确认为螺栓未缺失的图片依次进行所述步骤S201和所述步骤S202中的处理后，得到所述螺栓未缺失图片的二值图像，将所述螺栓未缺失图片的二值图像与所述待测图片的二值图像进行比较，计算出数值不同的个数，得到汉明距离，通过所述汉明距离，间接获得图像的相似度，所述图像的相似度公式为：

式中：D表示汉明距离；P表示Phash算法的图像相似度。

4.如权利要求3所述的基于Phash算法结合深度学习的线路螺栓缺陷识别方法，其特征在于，在所述步骤S203中，通过对所述图像相似度P的值进行计算，得到其距离分布均值d和标准差ε，则所述相似度阈值在区间[d+ε,d-ε]内选择，根据所述待测图片的所述图像的相似度P的值和所述相似度阈值判断螺栓缺失情况。

5.如权利要求4所述的基于Phash算法结合深度学习的线路螺栓缺陷识别方法，其特征在于，在所述步骤S203中，所述待测图片的所述图像的相似度P的值大于所述相似度阈值，则所述待测图片为螺栓未缺失；所述待测图片的所述图像的相似度P的值不大于所述相似度阈值，则所述待测图片为螺栓缺失。

6.如权利要求1所述的基于Phash算法结合深度学习的线路螺栓缺陷识别方法，其特征在于，在所述步骤S3中，通过采用随机翻转、随机缩放、随机平移和随机图片增强中的一种或其任意组合，对所述基础样本集进行样本量的扩充，得到所述扩充样本集。