CN115239642A - 一种输电线路中小金具缺陷的检测方法、检测装置和设备 - Google Patents

一种输电线路中小金具缺陷的检测方法、检测装置和设备 Download PDF

Info

Publication number
CN115239642A
CN115239642A CN202210779218.6A CN202210779218A CN115239642A CN 115239642 A CN115239642 A CN 115239642A CN 202210779218 A CN202210779218 A CN 202210779218A CN 115239642 A CN115239642 A CN 115239642A
Authority
CN
China
Prior art keywords
image
sub
feature
layer
inspection
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Pending
Application number
CN202210779218.6A
Other languages
English (en)
Inventor
李捷
冯宇
刘洪�
李光国
李杨
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Huayan Intelligent Technology Group Co ltd
Original Assignee
Huayan Intelligent Technology Group Co ltd
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by Huayan Intelligent Technology Group Co ltd filed Critical Huayan Intelligent Technology Group Co ltd
Priority to CN202210779218.6A priority Critical patent/CN115239642A/zh
Publication of CN115239642A publication Critical patent/CN115239642A/zh
Pending legal-status Critical Current

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/0002Inspection of images, e.g. flaw detection
    • G06T7/0004Industrial image inspection
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/10Segmentation; Edge detection
    • G06T7/11Region-based segmentation
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00Arrangements for image or video recognition or understanding
    • G06V10/70Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/764Arrangements for image or video recognition or understanding using pattern recognition or machine learning using classification, e.g. of video objects
    • G06V10/765Arrangements for image or video recognition or understanding using pattern recognition or machine learning using classification, e.g. of video objects using rules for classification or partitioning the feature space
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00Arrangements for image or video recognition or understanding
    • G06V10/70Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/77Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
    • G06V10/774Generating sets of training patterns; Bootstrap methods, e.g. bagging or boosting
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00Arrangements for image or video recognition or understanding
    • G06V10/70Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/77Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
    • G06V10/80Fusion, i.e. combining data from various sources at the sensor level, preprocessing level, feature extraction level or classification level
    • G06V10/806Fusion, i.e. combining data from various sources at the sensor level, preprocessing level, feature extraction level or classification level of extracted features
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00Arrangements for image or video recognition or understanding
    • G06V10/70Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/82Arrangements for image or video recognition or understanding using pattern recognition or machine learning using neural networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/10Image acquisition modality
    • G06T2207/10004Still image; Photographic image
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20081Training; Learning
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20084Artificial neural networks [ANN]
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20112Image segmentation details
    • G06T2207/20132Image cropping
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/30Subject of image; Context of image processing
    • G06T2207/30108Industrial image inspection
    • G06T2207/30164Workpiece; Machine component

Landscapes

  • Engineering & Computer Science (AREA)
  • Theoretical Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • General Physics & Mathematics (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Evolutionary Computation (AREA)
  • Health & Medical Sciences (AREA)
  • General Health & Medical Sciences (AREA)
  • Software Systems (AREA)
  • Computing Systems (AREA)
  • Artificial Intelligence (AREA)
  • Medical Informatics (AREA)
  • Databases & Information Systems (AREA)
  • Multimedia (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Biomedical Technology (AREA)
  • Biophysics (AREA)
  • Computational Linguistics (AREA)
  • Data Mining & Analysis (AREA)
  • Molecular Biology (AREA)
  • General Engineering & Computer Science (AREA)
  • Mathematical Physics (AREA)
  • Quality & Reliability (AREA)
  • Image Analysis (AREA)

Abstract

本申请提供了一种输电线路中小金具缺陷的检测方法、检测装置和设备,所述方法包括:获取目标输电线路的巡检图像,并对所述巡检图像进行滑动窗口剪裁得到多个巡检子图像;将巡检图像和巡检子图像分别输入到预先训练好的小金具缺陷检测模型中,获得大图检测结果和子图检测结果;小金具缺陷检测模型是利用巡检图像和巡检子图像训练得到的;利用滑动窗口的坐标对每个巡检子图像输出的子图检测结果进行调整,获得子图检测结果在所述巡检图像上对应的子图调整结果;对大图检测结果和子图调整结果进行合并处理,得到对所述巡检图像中小金具缺陷的检测结果。通过所述方法,能够在保持较快的检测速度的情况下以较高的检测精度实现对小金具缺陷的检测。

Description

一种输电线路中小金具缺陷的检测方法、检测装置和设备
技术领域
本申请涉及输电线路缺陷检测技术领域,具体而言,涉及一种输电线路中小金具缺陷的检测方法、检测装置和设备。
背景技术
小金具为输电线路中广泛使用的铁制或铝制的金属附件,它关系着导线或杆塔的安全,即使一只损坏,也可能造成输电线路的故障。因此,能够及时精准地检测出有故障的小金具,能够维持输电线路的正常运行。
目前,基于深度学习和机器视觉的目标检测技术已经应用到了输电线路缺陷检测领域中。然而,由于小金具缺陷目标相对于原始巡检图像占比极小,导致图像中包含的小金具像素信息少,现有的一阶段的目标检测网络由于不需要候选区域,因此在检测速度较高的情况下降低了检测精度,在应用于输电线路中小金具缺陷的检测时,对于有缺陷的小金具发现率较低且检测结果误报率较高。此外,在检测精度较高的二阶段的目标检测网络中,由于需要候选区域,因此在提高了检测精度的同时也降低了检测速度。
综上所述,现有的目标检测的网络在检测速度较高的情况下降低了检测精度,在检测精度较高的情况下降低了检测速度,因此,现有的目标检测网络无法保持在较高的检测速度的情况下以较高的检测精度实现对小金具缺陷的检测。
发明内容
有鉴于此,本申请实施例的目的在于提供一种输电线路中小金具缺陷的检测方法、检测装置和设备,能够在保持较快的检测速度的情况下以较高的检测精度实现对小金具缺陷的检测。
第一方面,本申请实施例提供了一种输电线路中小金具缺陷的检测方法,所述检测方法包括:
获取目标输电线路的巡检图像,并对所述巡检图像进行滑动窗口剪裁得到多个巡检子图像;
针对每个所述巡检图像,将该巡检图像输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对该巡检图像输出的大图检测结果;
将该巡检图像对应的多个巡检子图像依次输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对每个巡检子图像输出的子图检测结果;
利用所述滑动窗口的坐标对每个巡检子图像输出的子图检测结果进行调整,获得所述子图检测结果在所述巡检图像上对应的子图调整结果;
对所述大图检测结果和所述子图调整结果进行合并处理,得到对所述巡检图像中小金具缺陷的检测结果。
可选地,所述利用所述滑动窗口的坐标对每个巡检子图像输出的子图检测结果进行调整,获得所述子图检测结果在所述巡检图像上对应的子图调整结果,包括:
针对每个巡检子图像,获取该巡检子图像对应的滑动窗口的左上角坐标;
获取该巡检子图像中每个小金具的检测框的坐标;
利用所述滑动窗口的左上角坐标调整该巡检子图像中每个小金具的检测框的坐标,得到调整后的小金具的检测框的坐标;
在该巡检图像上按照所述调整后的小金具的检测框的坐标显示调整后的小金具的检测框,以获得所述子图检测结果在该巡检图像上对应的子图调整结果。
可选地,所述小金具缺陷检测模型通过以下方式训练而被得到:
获取训练集样本图像;
从所述训练集样本图像中抽取多个样本图像,得到原始训练子集样本图像;
针对所述原始训练子集样本图像中的每个样本图像,进行滑动窗口剪裁,获得多个样本子图像;
将所述多个样本子图像添加到所述原始训练子集样本图像中,获得训练子集样本图像;
利用所述训练集样本图像对初始小金具缺陷检测模型进行训练,获得预训练小金具缺陷检测模型;
利用所述训练子集样本图像对所述预训练小金具缺陷检测模型进行训练,以获得小金具缺陷检测模型。
可选地,所述小金具缺陷检测模型包括:特征提取层、特征融合层和检测层;所述将所述巡检图像输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果,包括:
将所述巡检图像输入到所述特征提取层对所述巡检图像进行特征提取,获得第一特征图像;所述第一特征图像包括第一尺度特征图像、第二尺度特征图像和第三尺度特征图像;
将所述第一特征图像输入到所述特征融合层对所述第一特征图像进行特征融合,得到第二特征图像;所述第二特征图像包括第四尺度特征图像、第五尺度特征图像和第六尺度特征图像;
将所述第二特征图像输入到所述检测层对所述第二特征图像进行检测,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果。
可选地,所述特征提取层包括:第一卷积层和第二卷积层组;所述第二卷积层组包括第三卷积层组和一个第四卷积层;所述第三卷积层组包括N个第五卷积层;所述将所述巡检图像输入到所述特征提取层对所述巡检图像进行特征提取,获得第一特征图像,包括:
将所述巡检图像输入到所述第一卷积层对所述目标巡检图像进行卷积操作,获得第一子特征图像;
将所述第一子特征图像输入到第三卷积层组,得到第二子特征图像组;所述第二子特征图像组包括在第N-1个第五卷积层和第N个第五卷积层输出的第二子特征图像;
将所述第N个第五卷积层输出的第二子特征图像输入到第四卷积层,得到第三子特征图像;
将第二子特征图像组包括的第二子特征图像和第三子特征图像确定为第一特征图像。
可选地,每个所述第五卷积层包括:下采样层和CSPNET网络层;所述CSPNET网络层包括多个级联的ConvNeXt Block残差结构;
所述将所述第一子特征图像输入到第三卷积层组,得到第二子特征图像组,包括:
将所述第一子特征图像输入到第三卷积层组,通过第N-1个第五卷积层中的下采样层得到第一下采样图像,将所述第一下采样图像输入到该第五卷积层中的CSPNET网络层进行升通道操作,得到第N-1个第五卷积层输出的第二子特征图像;
将所述第N-1个第五卷积层输出的第二子特征图像输入到第N个第五卷积层,通过该第五卷积层中的下采样层得到第二下采样图像,将所述第二下采样图像输入到该第五卷积层中的CSPNET网络层进行升通道操作,得到第N个第五卷积层输出的第二子特征图像;
将所述第N-1个第五卷积层输出的第二子特征图像和所述第N个第五卷积层输出的第二子特征图像,作为第二子特征图像组;
可选地,所述第四卷积层包括:下采样层、池化层和CSPNET网络层;所述将所述第N个第五卷积层输出的第二子特征图像输入到第四卷积层,得到第三子特征图像,包括:
将所述第N个第五卷积层输出的第二子特征图像输入到下采样层进行下采样处理,得到第三下采样特征图像;
将所述第三下采样特征图像输入到所述池化层进行最大池化处理,得到池化特征图像;
将所述池化特征图像输入到所述CSPNET网络层进行升通道操作,得到第三子特征图像。
可选地,所述特征融合层包括:自顶向下的特征金字塔网络、多个自处理层和自底向上的路径增强金字塔网络;所述将所述第一特征图像输入到所述特征融合层对所述第一特征图像进行特征融合,得到第二特征图像,包括:
将所述第一特征图像输入到所述自顶向下的特征金字塔网络进行下融合处理,获得所述自顶向下的特征金字塔网络中的每一层金字塔网络输出的浅层融合特征图像;
针对所述每一层金字塔网络输出的浅层融合特征图像,分别将该层金字塔网络输出的浅层融合特征图像输入到对应的自处理层进行卷积处理,获得每一层金字塔网络对应的自处理层输出的自处理特征图像;
将所述自处理特征图像输入到所述自底向上的路径增强金字塔网络进行上融合处理,获得所述自底向上的路径增强金字塔网络中的每一层金字塔网络输出的深层融合特征图像;
将所述自底向上的路径增强金字塔网络中的每一层金字塔网络输出的深层融合特征图像作为第二特征图像。
可选地,所述检测层包括三个检测头;每个检测头包括两层注意力机制结构组;每层注意力机制结构组包括多个注意力机制结构;
所述将所述第二特征图像输入到所述检测层对所述第二特征图像进行检测,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果,包括:
将所述第二特征图像输入到所述检测头中进行图像感知,以得到三种尺度感知图像;每种尺度感知图像中包括回归特征图像和分类特征图像;
针对每种尺度感知图像,将该种尺度感知图像输入到对应的回归与分类检测网络中进行回归与分类处理,得到该种尺度感知图像的回归结果和分类结果;
根据所述回归结果和分类结果,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果。
第二方面,本申请实施例提供了一种输电线路中小金具缺陷的检测装置,所述检测装置包括:
获取模块,用于获取目标输电线路的巡检图像,并对所述巡检图像进行滑动窗口剪裁得到多个巡检子图像;
大图检测模块,用于针对每个所述巡检图像,将该巡检图像输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对该巡检图像输出的大图检测结果;
小图检测模块,用于将该巡检图像对应的多个巡检子图像依次输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对每个巡检子图像输出的子图检测结果;
调整模块,用于利用所述滑动窗口的坐标对每个巡检子图像输出的子图检测结果进行调整,获得所述子图检测结果在所述巡检图像上对应的子图调整结果;
合并检测模块,用于对所述大图检测结果和所述子图调整结果进行合并处理,得到对所述巡检图像中小金具缺陷的检测结果。
第三方面,本申请实施例提供了一种电子设备,包括:处理器、存储器和总线,所述存储器存储有所述处理器可执行的机器可读指令,当电子设备运行时,所述处理器与所述存储器之间通过总线通信,所述处理器执行所述机器可读指令,以执行第一方面任一项所述的输电线路中小金具缺陷的检测方法的步骤。
第四方面,本申请实施例提供了一种计算机可读存储介质,所述计算机可读存储介质上存储有计算机程序,该计算机程序被处理器运行时执行第一方面任一项所述的输电线路中小金具缺陷的检测方法的步骤。
本申请实施例提供的输电线路中小金具缺陷的检测方法、检测装置和设备,能够基于改进的YOLOX目标检测模型对输电线路中小金具进行检测,在保持较快的检测速度的情况下以较高的检测精度实现对小金具缺陷的检测。
附图说明
为了更清楚地说明本申请实施例的技术方案,下面将对实施例中所需要使用的附图作简单地介绍,应当理解,以下附图仅示出了本申请的某些实施例,因此不应被看作是对范围的限定,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他相关的附图。
图1示出了本申请实施例提供的一种输电线路中小金具缺陷的检测方法的流程图;
图2示出了本申请实施例提供的基于YOLOX改进的小金具缺陷检测模型的网络结构示意图;
图3示出了本申请实施例提供的通过小金具缺陷检测模型获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果的流程图;
图4示出了本申请实施例提供的特征提取层的结构示意图;
图5示出了本申请实施例提供的通过所述特征提取层对所述巡检图像进行特征提取,获得第一特征图像的步骤的流程图;
图6示出了本申请实施例提供的第一卷积层的结构示意图;
图7示出了本申请实施例提供的CSPNET网络层的结构示意图;
图8示出了本申请实施例提供的ConvNeXt Block残差结构的结构示意图;
图9示出了本申请实施例提供的池化层的结构示意图;
图10示出了本申请实施例提供的特征融合层的网络结构示意图;
图11示出了本申请实施例提供的通过所述特征融合层对所述第一特征图像进行特征融合,得到第二特征图像的步骤的流程图;
图12示出了本申请实施例提供的检测层的检测头的结构示意图;
图13示出了本申请实施例提供的一种通过检测层获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果的步骤的流程图;
图14示出了本申请实施例提供的输电线路中小金具缺陷的检测方法的示意图;
图15示出了本申请实施例提供的一种输电线路中小金具缺陷的检测装置的结构示意图;
图16示出了本申请实施例提供的一种电子设备的结构示意图。
具体实施方式
为使本申请实施例的目的、技术方案和优点更加清楚,下面将结合本申请实施例中附图,对本申请实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本申请一部分实施例,而不是全部的实施例。通常在此处附图中描述和示出的本申请实施例的组件可以以各种不同的配置来布置和设计。因此,以下对在附图中提供的本申请的实施例的详细描述并非旨在限制要求保护的本申请的范围,而是仅仅表示本申请的选定实施例。基于本申请的实施例,本领域技术人员在没有做出创造性劳动的前提下所获得的所有其他实施例,都属于本申请保护的范围。
目前,常用于输电线路缺陷检测领域中缺陷识别的深度学习算法包括two-stage(两阶段)检测器,如:Fast-RCNN、Libra R-CNN、Faster-RCNN、Cascade-RCNN等,和one-stage(单阶段)检测器,如RetinaNet、SSD、YOLO及其改进系列等。两类算法各有优劣,通常两阶段算法精度较高,但检测速度慢;而单阶段算法检测速度快,但精度也略有下降。在工程应用中,一方面需要目标检测网络有较好的检测效果,另一方面网络的检测速度和实时性还应满足实际需求。
基于此,本申请实施例提供了一种输电线路中小金具缺陷的检测方法、检测装置和设备,下面通过实施例进行描述。
为便于对本实施例进行理解,首先对本申请实施例所公开的一种输电线路中小金具缺陷的检测方法进行详细介绍。参见图1所示,图1示出了本申请实施例提供的一种输电线路中小金具缺陷的检测方法的流程图。如图1所示,本申请实施例提供的一种小金具缺陷的检测方法包括以下步骤:
S1、获取目标输电线路的巡检图像,并对所述巡检图像进行滑动窗口剪裁得到多个巡检子图像。
这里,目标输电线路为预先选定的巡检区域中的输电线路,所述巡检图像可以为针对目标输电线路拍摄的多张巡检图像。
在该步骤中,具体地,针对每张巡检图像,进行滑动窗口剪裁,得到该巡检图像对应的多张巡检子图像。
作为示例,可以根据从上到下,从左到右的顺序进行滑动窗口剪裁。作为示例,滑动窗口的大小可以为巡检图像的短边长度的二分之一、四分之一和八分之一等。作为示例,滑动窗口的步长可以为滑动窗口大小的二分之一、四分之一、八分之一等。
S2、针对每个所述巡检图像,将该巡检图像输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对该巡检图像输出的大图检测结果。
作为示例,所述小金具缺陷检测模型通过以下方式训练而被得到:
首先获取训练集样本图像,然后从所述训练集样本图像中抽取多个样本图像,得到原始训练子集样本图像,然后针对所述原始训练子集样本图像中的每个样本图像,进行滑动窗口剪裁,获得多个样本子图像,然后将所述多个样本子图像添加到所述原始训练子集样本图像中,获得训练子集样本图像,然后利用所述训练集样本图像对初始小金具缺陷检测模型进行训练,获得预训练小金具缺陷检测模型,最后利用所述训练子集样本图像对所述预训练小金具缺陷检测模型进行训练,以获得小金具缺陷检测模型。
这里,作为示例,可以对目标输电线路拍摄多张巡检图像,然后将一部分巡检图像作为检测图像,另一部分巡检图像作为训练集样本图像。
这里,在得到训练集样本图像后,将所述训练集样本图像输入到初始小金具缺陷检测模型中进行参数的迭代优化,当迭代次数达到预先设置的第一迭代次数时停止训练,得到预训练小金具缺陷检测模型。在得到预训练小金具缺陷检测模型之后,将所述训练子集样本图像输入到所述预训练小金具缺陷检测模型中进行参数的迭代优化,当迭代次数达到预先设置的第二迭代次数时停止训练,获得小金具缺陷检测模型。
这里,初始小金具缺陷检测模型为基于现有的深度学习目标检测网络YOLOX改进的一种改进型YOLOX目标检测网络。基于此,本申请实施例中的小金具缺陷检测模型为基于上述改进型YOLOX目标检测网络进行训练得到的。
具体地,请参阅图2,图2示出了本申请实施例提供的基于YOLOX改进的小金具缺陷检测模型的网络结构示意图。
如图2所示,本申请实施例中的小金具缺陷检测模型包括:特征提取层、特征融合层和检测层。
原始的YOLOX目标检测网络是由CSPDarknet特征提取层、PAFPN特征融合层和检测头层组成。本申请实施例中基于CSPDarknet特征提取层进行改进,得到本申请实施例中的特征提取层,基于PAFPN特征融合层进行改进,得到本申请实施例中的特征融合层,基于检测头层进行改进得到本申请实施例中的检测层。
下面,将结合图3介绍在步骤S2中,如何通过小金具缺陷检测模型,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果。
具体地,请参阅图3,图3示出了本申请实施例提供的通过小金具缺陷检测模型获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果的流程图。
如图3所示,在步骤S21、将所述巡检图像输入到所述特征提取层对所述巡检图像进行特征提取,获得第一特征图像;所述第一特征图像包括第一尺度特征图像、第二尺度特征图像和第三尺度特征图像。
请参阅图4,图4示出了本申请实施例提供的特征提取层的结构示意图。如图4所示,所述特征提取层包括:第一卷积层和第二卷积层组;所述第二卷积层组包括第三卷积层组和一个第四卷积层;所述第三卷积层组包括N个第五卷积层;每个所述第五卷积层包括:下采样层和CSPNET网络层,所述第四卷积层包括:下采样层、池化层和CSPNET网络层,这里,所述CSPNET网络层包括多个级联的ConvNeXt Block残差结构。
请参阅图5,图5示出了本申请实施例提供的通过所述特征提取层对所述巡检图像进行特征提取,获得第一特征图像的步骤的流程图。
如图5所示,在步骤S211、将所述巡检图像输入到所述第一卷积层对所述目标巡检图像进行卷积操作,获得第一子特征图像。
具体地,请参阅图6,图6示出了本申请实施例提供的第一卷积层的结构示意图;如图6所示,第一卷积层由一个简单的卷积网络替换了原始YOLOX网络中的Focus stem层,作为示例,该卷积网络可以由一个卷积核(d)大小为6×6,步长为s=2的卷积层组成,需要说明的是,本领域技术人员应当知晓,该卷积网络的卷积核大小和步长可以为其他大小和步长,本申请对此不作限制。
在该步骤中,首先通过所述卷积核对输入为H×W×3的图像变换为H/2×W/2×64的特征层,然后使用Layer Norm和GELU激活函数对该特征层做归一化和非线性映射,得到第一子特征图像;
通过本申请实施例中的第一卷积层,可以避免原来的Focus stem层带来的space-to-depth operation(空间到深度方向的转换操作),减小了网络的FLOPs和复杂度。
在步骤S212、将所述第一子特征图像输入到第三卷积层组,得到第二子特征图像组;所述第二子特征图像组包括在第N-1个第五卷积层和第N个第五卷积层输出的第二子特征图像。
作为示例,该步骤可以包括:步骤a、将所述第一子特征图像输入到第三卷积层组,通过第N-1个第五卷积层中的下采样层得到第一下采样图像,将所述第一下采样图像输入到该第五卷积层中的CSPNET网络层进行升通道操作,得到第N-1个第五卷积层输出的第二子特征图像。
步骤b、将所述第N-1个第五卷积层输出的第二子特征图像输入到第N个第五卷积层,通过该第五卷积层中的下采样层得到第二下采样图像,将所述第二下采样图像输入到该第五卷积层中的第二CSPNET网络层进行升通道操作,得到第N个第五卷积层输出的第二子特征图像;
步骤c、将所述第N-1个第五卷积层输出的第二子特征图像和所述第N个第五卷积层输出的第二子特征图像,作为第二子特征图像组。
具体地,请参阅图7,图7示出了本申请实施例提供的CSPNET网络层的结构示意图,如图7所示,所述CSPNET网络层包括多个级联的ConvNeXt Block残差结构。
请参阅图8,图8示出了本申请实施例提供的ConvNeXt Block残差结构的结构示意图。
ConvNeXt Block残差结构为由一个卷积核大小为7×7和两个卷积核大小为1×1的卷积层组成的反瓶颈残差结构。对输入为H×W×C的特征层,先使用1×1的卷积把通道数升为原先的4倍即H×W×4C,最后使用1×1的卷积降为同输入特征层相同的通道数C,这样能让信息在不同通道特征空间转换时避免压缩通道维度带来的信息损失。
CSPNET网络层的核心结构即是由N个ConvNeXt Block残差结构块堆叠而成的,在左右两部分网络中:一部分即是由N个ConvNeXt Block残差结构块堆叠构成;另一部分可以看作一个大的残差边,大残差边的输出特征直接和另一部分的输出特征在通道维作拼接,最后经过1×1的卷积层调整通道和归一化输出。
在步骤S213、将所述第N个第五卷积层输出的第二子特征图像输入到第四卷积层,得到第三子特征图像;
具体地,该步骤包括:
步骤(1),将所述第N个第五卷积层输出的第二子特征图像输入到下采样层进行下采样处理,得到第三下采样特征图像;
步骤(2),将所述第三下采样特征图像输入到所述池化层进行最大池化处理,得到池化特征图像;
步骤(3),将所述池化特征图像输入到所述CSPNET网络层进行升通道操作,得到第三子特征图像。
具体地,请参阅图9,图9示出了本申请实施例提供的池化层的结构示意图,如图9所示,池化层采用具有较小FLOPs的快速级联空间池化结构SPPF。
SPPF主要是由3个相同大小池化核的最大池化层Maxpool级联组成,通过最大池化进行特征提取,提高网络的感受野。相比原YOLOx中使用的空间池化结构SPP,实验表明使用SPPF在保持检测精度不变的同时具有更快的推理速度。
在步骤S214、将第二子特征图像组包括的第二子特征图像和第三子特征图像确定为第一特征图像。
这里,第N-1个第五卷积层输出的第二子特征图像、第N个第五卷积层输出的第二子特征图像以及第三子特征图像即分别为第一尺度特征图像、第二尺度特征图像和第三尺度特征图像。
请继续参阅图3,在步骤S22、将所述第一特征图像输入到所述特征融合层对所述第一特征图像进行特征融合,得到第二特征图像;所述第二特征图像包括第四尺度特征图像、第五尺度特征图像和第六尺度特征图像。
具体地,请参阅图10,图10示出了本申请实施例提供的特征融合层的网络结构示意图。
如图10所示,特征融合层包括:自顶向下的特征金字塔网络、多个自处理层和自底向上的路径增强金字塔网络。
这里,自处理层由N个级联的ConvNeXt Block残差结构块构成。
在原始YOLOX网络中,PAFPN特征融合层是在自顶向下的特征金字塔网络FPN的输出后连接了一个自底向上的路径增强金字塔网络。但是PAFPN特征融合层的这种融合方式是在跨层的不同尺度特征层间的融合,比如P5特征层和经过上采样的N4的特征层的输出直接进行融合、N3特征层经过上采样直接和P4特征层融合,跨层之间特征层的直接融合并没有充分利用不同尺度特征的信息,基于此,本申请实施例基于PAFPN特征融合层进行了改进,在原始YOLOX网络中自顶向下的特征金字塔网络的每一层金字塔网络后面都增加了自处理层,通过自处理进行信息的深度融合,相比于PAFPN特征融合层能够充分利用不同尺度特征的信息。
具体地,请参阅图11,图11示出了本申请实施例提供的通过所述特征融合层对所述第一特征图像进行特征融合,得到第二特征图像的步骤的流程图。
如图11所示,步骤S22包括:
步骤S221、将所述第一特征图像输入到所述自顶向下的特征金字塔网络进行下融合处理,获得所述自顶向下的特征金字塔网络中的每一层金字塔网络输出的浅层融合特征图像。
步骤S222、针对所述每一层金字塔网络输出的浅层融合特征图像,分别将该层金字塔网络输出的浅层融合特征图像输入到对应的自处理层进行卷积处理,获得每一层金字塔网络对应的自处理层输出的自处理特征图像。
步骤S223、将所述自处理特征图像输入到所述自底向上的路径增强金字塔网络进行上融合处理,获得所述自底向上的路径增强金字塔网络中的每一层金字塔网络输出的深层融合特征图像。
步骤S224、将所述自底向上的路径增强金字塔网络中的每一层金字塔网络输出的深层融合特征图像作为第二特征图像。
请继续参阅图3,在步骤S23、将所述第二特征图像输入到所述检测层对所述第二特征图像进行检测,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果。
请参阅图12,图12示出了本申请实施例提供的检测层中的检测头的结构示意图。
这里,所述检测层包括三个检测头;每个检测头包括两层注意力机制结构组;每层注意力机制结构组包括多个注意力机制结构,其中,每个所述注意力机制结构包括尺度感知模块,空间感知模块和任务感知模块。
这里,在原始YOLOX目标检测网络的检测层中,是直接在自底向上的特征融合金字塔网络的每一层金字塔网络的输出后堆叠多个普通的卷积层,然后在卷积层后分别连接一个回归卷积子网络和分类卷积子网络。本申请实施例对原始YOLOX检测头层进行了改进,用包括了多个注意力机制结构组的的检测头替换了原始的堆叠的多个普通卷积层,本申请实施例中的这种检测层,通过在特征层间、特征层空间位置上和特征通道内引入注意力机制,分别用以目标检测任务的于尺度感知、空间位置感知和任务感知,在不增加计算量的情况下显著的提升了检测头的表征能力。
请参阅图13,图13示出了本申请实施例提供的一种通过检测层获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果的步骤的流程图。
如图13所示,在步骤S231,将所述第二特征图像输入到所述检测头中进行图像感知,以得到三种尺度感知图像;每种尺度感知图像中包括回归特征图像和分类特征图像。
在步骤S232、针对每种尺度感知图像,将该种尺度感知图像输入到对应的回归与分类检测网络中进行回归与分类处理,得到该种尺度感知图像的回归结果和分类结果;
在步骤S233、根据所述回归结果和分类结果,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果。
请继续参阅图1,在步骤S3,将该巡检图像对应的多个巡检子图像依次输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对每个巡检子图像输出的子图检测结果。
这里,通过预先训练好的小金具缺陷检测模型获得所述小金具缺陷检测模型针对每个巡检子图像输出的子图检测结果的步骤和上述获得大图检测结果的步骤同理,因此不再赘述。
在步骤S4,利用所述滑动窗口的坐标对每个巡检子图像输出的子图检测结果进行调整,获得所述子图检测结果在所述巡检图像上对应的子图调整结果。
作为示例,在该步骤中,首先可以针对每个巡检子图像,获取该巡检子图像对应的滑动窗口的左上角坐标;
例如,第n个滑动窗口的左上角坐标可以为[shift_xnl,shift_ynl];
然后获取该巡检子图像中每个小金具的检测框的坐标;
例如,每个小金具的检测框的坐标可以为[xnl i,ynl i,xnr i,ynr i];
这里,xnl i为第n个滑动窗口对应的巡检子图像中第i个小金具的检测框的左上角横坐标;ynl i为第n个滑动窗口对应的巡检子图像中第i个小金具的检测框的左上角纵坐标;xnr i为第n个滑动窗口对应的巡检子图像中第i个小金具的检测框的右下角横坐标;ynr i为第n个滑动窗口对应的巡检子图像中第i个小金具的检测框的右下角纵坐标。
然后,利用所述滑动窗口的左上角坐标调整该巡检子图像中每个小金具的检测框的坐标,得到调整后的小金具的检测框的坐标;
例如,可以利用以下公式得到调整后的小金具的检测框的坐标box:
box=[xnl i+shift_xnl,ynl i+shift_ynl,xnr i+shift_xnl,ynr i+shift_ynl];
最后,在该巡检图像上按照所述调整后的小金具的检测框的坐标显示调整后的小金具的检测框,以获得所述子图检测结果在该巡检图像上对应的子图调整结果。
例如,若巡检图像中的小金具的检测框有1个,巡检子图像有4个,每个巡检子图像中的小金具的检测框有1个,那么调整后的小金具的检测框有4个。子图调整结果为在巡检图像上显示的4个调整后的小金具检测框和原有的1个小金具的检测框。
在步骤S5,对所述大图检测结果和所述子图调整结果进行合并处理,得到对所述巡检图像中小金具缺陷的检测结果。
这里,合并处理包括非极大值抑制、大尺度非极大值抑制、非极大值融合等,以过滤或者融合合并后的重叠检测框,得到最终的检测结果。
这里,小金具缺陷的检测结果包括检测框,检测框坐标和标签,标签包括:缺陷和非缺陷。
下面,将结合图14介绍在一个具体的示例中,如何对小金具缺陷进行检测。
请参阅图14,图14示出了本申请实施例提供的一种输电线路中小金具缺陷的检测方法的示意图。
如图14所示,在基于滑窗剪裁的检测模块中,首先,获取目标输电线路的巡检图像M1、M2……MN,并对每个巡检图像均进行滑动窗口剪裁得到多个巡检子图像,这里,多个巡检子图像中包括每个巡检图像对应的巡检子图像。这里,例如针对巡检图像M1进行滑动窗口剪裁,获得该巡检图像对应的巡检子图像m1、m2……mn
然后,针对每个所述巡检图像,例如针对巡检图像M1,将该巡检图像M1输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对该巡检图像M1输出的大图检测结果;
然后,将该巡检图像M1对应的多个巡检子图像,即巡检子图像m1、m2……mn,依次输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对每个巡检子图像输出的子图检测结果;
然后针对每个巡检子图像,例如针对巡检子图像m1,获取该巡检子图像m1对应的滑动窗口的左上角坐标,并获取该巡检子图像m1中每个小金具的检测框的坐标,然后利用所述滑动窗口的左上角坐标调整该巡检子图像m1中每个小金具的检测框的坐标,得到调整后的小金具的检测框的坐标,然后在该巡检图像M1上按照所述调整后的小金具的检测框的坐标显示调整后的小金具的检测框,以获得所述子图检测结果在该巡检图像M1上对应的子图调整结果。
最后,对所述大图检测结果和所述子图调整结果进行合并处理(例如非极大值抑制),得到对所述巡检图像中小金具缺陷的检测结果。这里,基于滑窗剪裁的训练微调模型模块用于训练小金具缺陷检测模型。如图14所示,在基于滑窗剪裁的训练微调模型模块中,通过以下方式训练小金具缺陷检测模型:
首先获取训练集样本图像P1、P2、P3……Pn,然后从所述训练集样本图像P1、P2、P3……Pn中抽取多个样本图像,例如P1、P2,得到原始训练子集样本图像,然后针对所述原始训练子集样本图像中的每个样本图像,例如针对样本图像P1进行滑动窗口剪裁,获得多个样本子图像p1、p2,然后将所述多个样本子图像p1、p2添加到所述原始训练子集样本图像中,获得训练子集样本图像P1、P2、p1、p2,然后利用所述训练集样本图像P1、P2、P3……Pn对初始小金具缺陷检测模型进行训练,获得预训练小金具缺陷检测模型,然后利用所述训练子集样本图像P1、P2、p1、p2对所述预训练小金具缺陷检测模型进行训练,获得小金具缺陷检测模型。
本申请实施例中,能够在保持较快的检测速度的情况下以较高的检测精度实现对小金具缺陷的检测。
基于同一发明构思,本申请实施例中还提供了与上述输电线路中小金具缺陷的检测方法对应的输电线路中小金具缺陷的检测装置。
参见图15所示,图15为本申请实施例提供的一种输电线路中小金具缺陷的检测装置的结构示意图,该输电线路中小金具缺陷的检测装置100包括:
获取模块101,用于获取目标输电线路的巡检图像,并对所述巡检图像进行滑动窗口剪裁得到多个巡检子图像;
大图检测模块102,用于针对每个所述巡检图像,将该巡检图像输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对该巡检图像输出的大图检测结果;
小图检测模块103,用于将该巡检图像对应的多个巡检子图像依次输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对每个巡检子图像输出的子图检测结果;
调整模块104,用于利用所述滑动窗口的坐标对每个巡检子图像输出的子图检测结果进行调整,获得所述子图检测结果在所述巡检图像上对应的子图调整结果;
合并检测模块105,用于对所述大图检测结果和所述子图调整结果进行合并处理,得到对所述巡检图像中小金具缺陷的检测结果。
在一种可能的实施方式中,调整模块104,具体用于:
针对每个巡检子图像,获取该巡检子图像对应的滑动窗口的左上角坐标;
获取该巡检子图像中每个小金具的检测框的坐标;
利用所述滑动窗口的左上角坐标调整该巡检子图像中每个小金具的检测框的坐标,得到调整后的小金具的检测框的坐标;
在该巡检图像上按照所述调整后的小金具的检测框的坐标显示调整后的小金具的检测框,以获得所述子图检测结果在该巡检图像上对应的子图调整结果。
在一种可能的实施方式中,上述输电线路中小金具缺陷的检测装置100还包括:训练模块106(未在图中示出)。
所述训练模块106具体用于:获取训练集样本图像;
从所述训练集样本图像中抽取多个样本图像,得到原始训练子集样本图像;
针对所述原始训练子集样本图像中的每个样本图像,进行滑动窗口剪裁,获得多个样本子图像;
将所述多个样本子图像添加到所述原始训练子集样本图像中,获得训练子集样本图像;
利用所述训练集样本图像对初始小金具缺陷检测模型进行训练,获得预训练小金具缺陷检测模型;
利用所述训练子集样本图像对所述预训练小金具缺陷检测模型进行训练,以获得小金具缺陷检测模型。
在一种可能的实施方式中,上述大图检测模块102,具体用于:将所述巡检图像输入到所述特征提取层对所述巡检图像进行特征提取,获得第一特征图像;所述第一特征图像包括第一尺度特征图像、第二尺度特征图像和第三尺度特征图像;
将所述第一特征图像输入到所述特征融合层对所述第一特征图像进行特征融合,得到第二特征图像;所述第二特征图像包括第四尺度特征图像、第五尺度特征图像和第六尺度特征图像;
将所述第二特征图像输入到所述检测层对所述第二特征图像进行检测,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果。
在一种可能的实施方式中,上述大图检测模块102,还用于:
将所述巡检图像输入到所述第一卷积层对所述目标巡检图像进行卷积操作,获得第一子特征图像;
将所述第一子特征图像输入到第三卷积层组,得到第二子特征图像组;所述第二子特征图像组包括在第N-1个第五卷积层和第N个第五卷积层输出的第二子特征图像;
将所述第N个第五卷积层输出的第二子特征图像输入到第四卷积层,得到第三子特征图像;
将第二子特征图像组包括的第二子特征图像和第三子特征图像确定为第一特征图像。
在一种可能的实施方式中,上述大图检测模块102,还用于:
所述将所述第一子特征图像输入到第三卷积层组,得到第二子特征图像组,包括:
将所述第一子特征图像输入到第三卷积层组,通过第N-1个第五卷积层中的下采样层得到第一下采样图像,将所述第一下采样图像输入到该第五卷积层中的CSPNET网络层进行升通道操作,得到第N-1个第五卷积层输出的第二子特征图像;
将所述第N-1个第五卷积层输出的第二子特征图像输入到第N个第五卷积层,通过该第五卷积层中的下采样层得到第二下采样图像,将所述第二下采样图像输入到该第五卷积层中的CSPNET网络层进行升通道操作,得到第N个第五卷积层输出的第二子特征图像;
将所述第N-1个第五卷积层输出的第二子特征图像和所述第N个第五卷积层输出的第二子特征图像,作为第二子特征图像组。
在一种可能的实施方式中,上述大图检测模块102,还用于:
将所述第N个第五卷积层输出的第二子特征图像输入到下采样层进行下采样处理,得到第三下采样特征图像;
将所述第三下采样特征图像输入到所述池化层进行最大池化处理,得到池化特征图像;
将所述池化特征图像输入到所述CSPNET网络层进行升通道操作,得到第三子特征图像。
在一种可能的实施方式中,上述大图检测模块102,还用于:
将所述第一特征图像输入到所述自顶向下的特征金字塔网络进行下融合处理,获得所述自顶向下的特征金字塔网络中的每一层金字塔网络输出的浅层融合特征图像;
针对所述每一层金字塔网络输出的浅层融合特征图像,分别将该层金字塔网络输出的浅层融合特征图像输入到对应的自处理层进行卷积处理,获得每一层金字塔网络对应的自处理层输出的自处理特征图像;
将所述自处理特征图像输入到所述自底向上的路径增强金字塔网络进行上融合处理,获得所述自底向上的路径增强金字塔网络中的每一层金字塔网络输出的深层融合特征图像;
将所述自底向上的路径增强金字塔网络中的每一层金字塔网络输出的深层融合特征图像作为第二特征图像。
在一种可能的实施方式中,上述大图检测模块102,还用于:
所述将所述第二特征图像输入到所述检测层对所述第二特征图像进行检测,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果,包括:
将所述第二特征图像输入到所述检测头中进行图像感知,以得到三种尺度感知图像;每种尺度感知图像中包括回归特征图像和分类特征图像;
针对每种尺度感知图像,将该种尺度感知图像输入到对应的回归与分类检测网络中进行回归与分类处理,得到该种尺度感知图像的回归结果和分类结果;
根据所述回归结果和分类结果,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果。
本申请实施例提供的输电线路中小金具缺陷的检测装置,能够基于改进的YOLOX目标检测模型对输电线路中小金具进行检测,在保持较快的检测速度的情况下以较高的检测精度实现对小金具缺陷的检测。
参见图16所示,图16为本申请实施例提供的一种电子设备的示意图,该电子设备200包括:处理器201、存储器202和总线203,所述存储器202存储有所述处理器201可执行的机器可读指令,当电子设备运行时,所述处理器201与所述存储器202之间通过总线203通信,所述处理器201执行所述机器可读指令,以执行如上述输电线路中小金具缺陷的检测方法的步骤。
具体地,上述存储器202和处理器201能够为通用的存储器和处理器,这里不做具体限定,当处理器201运行存储器202存储的计算机程序时,能够执行上述输电线路中小金具缺陷的检测方法。
对应于上述输电线路中小金具缺陷的检测方法,本申请实施例还提供了一种计算机可读存储介质,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被处理器运行时执行上述输电线路中小金具缺陷的检测方法的步骤。
所属领域的技术人员可以清楚地了解到,为描述的方便和简洁,上述描述的系统和装置的具体工作过程,可以参考前述方法实施例中的对应过程,在此不再赘述。在本申请所提供的几个实施例中,应该理解到,所揭露的系统、装置和方法,可以通过其它的方式实现。以上所描述的装置实施例仅仅是示意性的,例如,所述模块的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,又例如,多个模块或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些通信接口,装置或模块的间接耦合或通信连接,可以是电性,机械或其它的形式。
所述作为分离部件说明的模块可以是或者也可以不是物理上分开的,作为模块显示的部件可以是或者也可以不是物理模块,即可以位于一个地方,或者也可以分布到多个网络模块上。可以根据实际的需要选择其中的部分或者全部模块来实现本实施例方案的目的。
另外,在本申请各个实施例中的各功能模块可以集成在一个处理模块中,也可以是各个模块单独物理存在,也可以两个或两个以上模块集成在一个模块中。
所述功能如果以软件功能模块的形式实现并作为独立的产品销售或使用时,可以存储在一个处理器可执行的非易失的计算机可读取存储介质中。基于这样的理解,本申请的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质中,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行本申请各个实施例所述方法的全部或部分步骤。而前述的存储介质包括:U盘、移动硬盘、只读存储器(Read-OnlyMemory,ROM)、随机存取存储器(Random Access Memory,RAM)、磁碟或者光盘等各种可以存储程序代码的介质。
最后应说明的是:以上所述实施例,仅为本申请的具体实施方式,用以说明本申请的技术方案,而非对其限制,本申请的保护范围并不局限于此,尽管参照前述实施例对本申请进行了详细的说明,本领域的普通技术人员应当理解:任何熟悉本技术领域的技术人员在本申请揭露的技术范围内,其依然可以对前述实施例所记载的技术方案进行修改或可轻易想到变化,或者对其中部分技术特征进行等同替换;而这些修改、变化或者替换,并不使相应技术方案的本质脱离本申请实施例技术方案的精神和范围,都应涵盖在本申请的保护范围之内。因此,本申请的保护范围应以权利要求的保护范围为准。

Claims (12)

1.一种输电线路中小金具缺陷的检测方法,其特征在于,所述检测方法包括:
获取目标输电线路的巡检图像,并对所述巡检图像进行滑动窗口剪裁得到多个巡检子图像;
针对每个所述巡检图像,将该巡检图像输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对该巡检图像输出的大图检测结果;
将该巡检图像对应的多个巡检子图像依次输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对每个巡检子图像输出的子图检测结果;
利用所述滑动窗口的坐标对每个巡检子图像输出的子图检测结果进行调整,获得所述子图检测结果在所述巡检图像上对应的子图调整结果;
对所述大图检测结果和所述子图调整结果进行合并处理,得到对所述巡检图像中小金具缺陷的检测结果。
2.根据权利要求1所述的检测方法,其特征在于,所述利用所述滑动窗口的坐标对每个巡检子图像输出的子图检测结果进行调整,获得所述子图检测结果在所述巡检图像上对应的子图调整结果,包括:
针对每个巡检子图像,获取该巡检子图像对应的滑动窗口的左上角坐标;
获取该巡检子图像中每个小金具的检测框的坐标;
利用所述滑动窗口的左上角坐标调整该巡检子图像中每个小金具的检测框的坐标,得到调整后的小金具的检测框的坐标;
在该巡检图像上按照所述调整后的小金具的检测框的坐标显示调整后的小金具的检测框,以获得所述子图检测结果在该巡检图像上对应的子图调整结果。
3.根据权利要求1所述的检测方法,其特征在于,所述小金具缺陷检测模型通过以下方式训练而被得到:
获取训练集样本图像;
从所述训练集样本图像中抽取多个样本图像,得到原始训练子集样本图像;
针对所述原始训练子集样本图像中的每个样本图像,进行滑动窗口剪裁,获得多个样本子图像;
将所述多个样本子图像添加到所述原始训练子集样本图像中,获得训练子集样本图像;
利用所述训练集样本图像对初始小金具缺陷检测模型进行训练,获得预训练小金具缺陷检测模型;
利用所述训练子集样本图像对所述预训练小金具缺陷检测模型进行训练,以获得小金具缺陷检测模型。
4.根据权利要求1所述的检测方法,其特征在于,所述小金具缺陷检测模型包括:特征提取层、特征融合层和检测层;所述将所述巡检图像输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果,包括:
将所述巡检图像输入到所述特征提取层对所述巡检图像进行特征提取,获得第一特征图像;所述第一特征图像包括第一尺度特征图像、第二尺度特征图像和第三尺度特征图像;
将所述第一特征图像输入到所述特征融合层对所述第一特征图像进行特征融合,得到第二特征图像;所述第二特征图像包括第四尺度特征图像、第五尺度特征图像和第六尺度特征图像;
将所述第二特征图像输入到所述检测层对所述第二特征图像进行检测,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果。
5.根据权利要求3所述的检测方法,其特征在于,所述特征提取层包括:第一卷积层和第二卷积层组;所述第二卷积层组包括第三卷积层组和一个第四卷积层;所述第三卷积层组包括N个第五卷积层;所述将所述巡检图像输入到所述特征提取层对所述巡检图像进行特征提取,获得第一特征图像,包括:
将所述巡检图像输入到所述第一卷积层对所述目标巡检图像进行卷积操作,获得第一子特征图像;
将所述第一子特征图像输入到第三卷积层组,得到第二子特征图像组;所述第二子特征图像组包括在第N-1个第五卷积层和第N个第五卷积层输出的第二子特征图像;
将所述第N个第五卷积层输出的第二子特征图像输入到第四卷积层,得到第三子特征图像;
将第二子特征图像组包括的第二子特征图像和第三子特征图像确定为第一特征图像。
6.根据权利要求5所述的检测方法,其特征在于,每个所述第五卷积层包括:下采样层和CSPNET网络层;所述CSPNET网络层包括多个级联的ConvNeXt Block残差结构;
所述将所述第一子特征图像输入到第三卷积层组,得到第二子特征图像组,包括:
将所述第一子特征图像输入到第三卷积层组,通过第N-1个第五卷积层中的下采样层得到第一下采样图像,将所述第一下采样图像输入到该第五卷积层中的CSPNET网络层进行升通道操作,得到第N-1个第五卷积层输出的第二子特征图像;
将所述第N-1个第五卷积层输出的第二子特征图像输入到第N个第五卷积层,通过该第五卷积层中的下采样层得到第二下采样图像,将所述第二下采样图像输入到该第五卷积层中的CSPNET网络层进行升通道操作,得到第N个第五卷积层输出的第二子特征图像;
将所述第N-1个第五卷积层输出的第二子特征图像和所述第N个第五卷积层输出的第二子特征图像,作为第二子特征图像组。
7.根据权利要求5所述的检测方法,所述第四卷积层包括:下采样层、池化层和CSPNET网络层;所述将所述第N个第五卷积层输出的第二子特征图像输入到第四卷积层,得到第三子特征图像,包括:
将所述第N个第五卷积层输出的第二子特征图像输入到下采样层进行下采样处理,得到第三下采样特征图像;
将所述第三下采样特征图像输入到所述池化层进行最大池化处理,得到池化特征图像;
将所述池化特征图像输入到所述CSPNET网络层进行升通道操作,得到第三子特征图像。
8.根据权利要求4所述的检测方法,其特征在于,所述特征融合层包括:自顶向下的特征金字塔网络、多个自处理层和自底向上的路径增强金字塔网络;所述将所述第一特征图像输入到所述特征融合层对所述第一特征图像进行特征融合,得到第二特征图像,包括:
将所述第一特征图像输入到所述自顶向下的特征金字塔网络进行下融合处理,获得所述自顶向下的特征金字塔网络中的每一层金字塔网络输出的浅层融合特征图像;
针对所述每一层金字塔网络输出的浅层融合特征图像,分别将该层金字塔网络输出的浅层融合特征图像输入到对应的自处理层进行卷积处理,获得每一层金字塔网络对应的自处理层输出的自处理特征图像;
将所述自处理特征图像输入到所述自底向上的路径增强金字塔网络进行上融合处理,获得所述自底向上的路径增强金字塔网络中的每一层金字塔网络输出的深层融合特征图像;
将所述自底向上的路径增强金字塔网络中的每一层金字塔网络输出的深层融合特征图像作为第二特征图像。
9.根据权利要求4所述的检测方法,其特征在于,所述检测层包括三个检测头;每个检测头包括两层注意力机制结构组;每层注意力机制结构组包括多个注意力机制结构;
所述将所述第二特征图像输入到所述检测层对所述第二特征图像进行检测,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果,包括:
将所述第二特征图像输入到所述检测头中进行图像感知,以得到三种尺度感知图像;每种尺度感知图像中包括回归特征图像和分类特征图像;
针对每种尺度感知图像,将该种尺度感知图像输入到对应的回归与分类检测网络中进行回归与分类处理,得到该种尺度感知图像的回归结果和分类结果;
根据所述回归结果和分类结果,获得所述小金具缺陷检测模型针对所述巡检图像输出的大图检测结果。
10.一种输电线路中小金具缺陷的检测装置,其特征在于,所述装置包括:
获取模块,用于获取目标输电线路的巡检图像,并对所述巡检图像进行滑动窗口剪裁得到多个巡检子图像;
大图检测模块,用于针对每个所述巡检图像,将该巡检图像输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对该巡检图像输出的大图检测结果;
小图检测模块,用于将该巡检图像对应的多个巡检子图像依次输入到预先训练好的小金具缺陷检测模型中,获得所述小金具缺陷检测模型针对每个巡检子图像输出的子图检测结果;
调整模块,用于利用所述滑动窗口的坐标对每个巡检子图像输出的子图检测结果进行调整,获得所述子图检测结果在所述巡检图像上对应的子图调整结果;
合并检测模块,用于对所述大图检测结果和所述子图调整结果进行合并处理,得到对所述巡检图像中小金具缺陷的检测结果。
11.一种电子设备,其特征在于,包括:处理器、存储器和总线,所述存储器存储有所述处理器可执行的机器可读指令,当电子设备运行时,所述处理器与所述存储器之间通过所述总线进行通信,所述机器可读指令被所述处理器运行时执行如权利要求1至9任一所述的输电线路中小金具缺陷的检测方法的步骤。
12.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被处理器运行时执行如权利要求1至9任一所述的输电线路中小金具缺陷的检测方法的步骤。
CN202210779218.6A 2022-07-01 2022-07-01 一种输电线路中小金具缺陷的检测方法、检测装置和设备 Pending CN115239642A (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202210779218.6A CN115239642A (zh) 2022-07-01 2022-07-01 一种输电线路中小金具缺陷的检测方法、检测装置和设备

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202210779218.6A CN115239642A (zh) 2022-07-01 2022-07-01 一种输电线路中小金具缺陷的检测方法、检测装置和设备

Publications (1)

Publication Number Publication Date
CN115239642A true CN115239642A (zh) 2022-10-25

Family

ID=83671658

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202210779218.6A Pending CN115239642A (zh) 2022-07-01 2022-07-01 一种输电线路中小金具缺陷的检测方法、检测装置和设备

Country Status (1)

Country Link
CN (1) CN115239642A (zh)

Cited By (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116385952A (zh) * 2023-06-01 2023-07-04 华雁智能科技(集团)股份有限公司 配网线路小目标缺陷检测方法、装置、设备及存储介质
CN116596895A (zh) * 2023-05-22 2023-08-15 华雁智科(杭州)信息技术有限公司 一种变电设备图像缺陷识别方法及系统
CN117291922A (zh) * 2023-11-27 2023-12-26 浙江日井泵业股份有限公司 一种不锈钢多级泵叶轮缺陷视觉检测方法
CN117333491A (zh) * 2023-12-01 2024-01-02 北京航空航天大学杭州创新研究院 一种钢材表面缺陷检测方法及系统

Cited By (8)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116596895A (zh) * 2023-05-22 2023-08-15 华雁智科(杭州)信息技术有限公司 一种变电设备图像缺陷识别方法及系统
CN116596895B (zh) * 2023-05-22 2024-10-11 华雁智科(杭州)信息技术有限公司 一种变电设备图像缺陷识别方法及系统
CN116385952A (zh) * 2023-06-01 2023-07-04 华雁智能科技(集团)股份有限公司 配网线路小目标缺陷检测方法、装置、设备及存储介质
CN116385952B (zh) * 2023-06-01 2023-09-01 华雁智能科技(集团)股份有限公司 配网线路小目标缺陷检测方法、装置、设备及存储介质
CN117291922A (zh) * 2023-11-27 2023-12-26 浙江日井泵业股份有限公司 一种不锈钢多级泵叶轮缺陷视觉检测方法
CN117291922B (zh) * 2023-11-27 2024-01-30 浙江日井泵业股份有限公司 一种不锈钢多级泵叶轮缺陷视觉检测方法
CN117333491A (zh) * 2023-12-01 2024-01-02 北京航空航天大学杭州创新研究院 一种钢材表面缺陷检测方法及系统
CN117333491B (zh) * 2023-12-01 2024-03-15 北京航空航天大学杭州创新研究院 一种钢材表面缺陷检测方法及系统

Similar Documents

Publication Publication Date Title
CN115239642A (zh) 一种输电线路中小金具缺陷的检测方法、检测装置和设备
CN113392960B (zh) 一种基于混合空洞卷积金字塔的目标检测网络及方法
US11836967B2 (en) Method and device for small sample defect classification and computing equipment
CN111524135A (zh) 基于图像增强的输电线路细小金具缺陷检测方法及系统
CN113139543B (zh) 目标对象检测模型的训练方法、目标对象检测方法和设备
CN111667476B (zh) 布料瑕疵检测方法、装置、电子设备及可读存储介质
CN116883801A (zh) 基于注意力机制与多尺度特征融合的YOLOv8目标检测方法
CN112950579B (zh) 图像质量评价方法、装置和电子设备
CN114463759A (zh) 一种基于无锚框算法的轻量化文字检测方法及装置
CN114580515A (zh) 用于半导体脱焊智能检测的神经网络训练方法
CN116503318A (zh) 一种融合CAT-BiFPN与注意力机制的航拍绝缘子多缺陷检测方法、系统及设备
US20240071111A1 (en) Method, apparatus and system for inspecting cell crush
CN117197624A (zh) 一种基于注意力机制的红外-可见光图像融合方法
CN117522871B (zh) 基于视觉图像交互的半导体晶圆检测方法及系统
CN118115835A (zh) 导光板缺陷小样本数据扩充方法、系统、设备及存储介质
CN117853778A (zh) 一种基于改进的htc铸件dr图像缺陷识别方法
CN114463300A (zh) 钢材表面缺陷检测方法、电子设备、存储介质
CN114693963A (zh) 基于电力数据特征提取的识别模型训练、识别方法及装置
CN117690161A (zh) 一种基于图像融合的行人检测方法、设备和介质
CN117197530A (zh) 一种基于改进YOLOv8模型及余弦退火学习率衰减法的绝缘子缺陷识别方法
CN115331081A (zh) 图像目标检测方法与装置
CN112529881B (zh) 一种电力控制柜线缆异常识别方法及装置
CN117011216A (zh) 一种缺陷检测方法、装置、电子设备及存储介质
Liu et al. Spark plug defects detection based on improved Faster-RCNN algorithm
Zhao et al. Mobile phone screen surface scratch detection based on optimized YOLOv5 model (OYm)

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication
SE01 Entry into force of request for substantive examination
SE01 Entry into force of request for substantive examination