CN113344852A - 一种电力场景通用物品的目标检测方法、装置及存储介质 - Google Patents
一种电力场景通用物品的目标检测方法、装置及存储介质 Download PDFInfo
- Publication number
- CN113344852A CN113344852A CN202110487848.1A CN202110487848A CN113344852A CN 113344852 A CN113344852 A CN 113344852A CN 202110487848 A CN202110487848 A CN 202110487848A CN 113344852 A CN113344852 A CN 113344852A
- Authority
- CN
- China
- Prior art keywords
- prediction
- neural network
- target frame
- image
- network model
- 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.)
- Withdrawn
Links
- 238000001514 detection method Methods 0.000 title claims abstract description 55
- 238000013527 convolutional neural network Methods 0.000 claims abstract description 55
- 238000012549 training Methods 0.000 claims abstract description 50
- 238000000034 method Methods 0.000 claims abstract description 31
- 238000002372 labelling Methods 0.000 claims abstract description 25
- 230000006870 function Effects 0.000 claims abstract description 16
- 238000012545 processing Methods 0.000 claims description 17
- 238000004590 computer program Methods 0.000 claims description 10
- 238000011176 pooling Methods 0.000 claims description 6
- 230000000903 blocking effect Effects 0.000 claims description 4
- 230000001629 suppression Effects 0.000 claims description 3
- 238000003062 neural network model Methods 0.000 claims description 2
- 238000000638 solvent extraction Methods 0.000 claims description 2
- 230000003190 augmentative effect Effects 0.000 claims 1
- 230000005540 biological transmission Effects 0.000 description 4
- 238000010586 diagram Methods 0.000 description 4
- 238000000605 extraction Methods 0.000 description 4
- 230000009286 beneficial effect Effects 0.000 description 3
- 230000000694 effects Effects 0.000 description 2
- 238000005516 engineering process Methods 0.000 description 2
- 230000007613 environmental effect Effects 0.000 description 2
- 238000002474 experimental method Methods 0.000 description 2
- 238000005259 measurement Methods 0.000 description 2
- 238000012360 testing method Methods 0.000 description 2
- 230000000007 visual effect Effects 0.000 description 2
- 240000007651 Rubus glaucus Species 0.000 description 1
- 235000011034 Rubus glaucus Nutrition 0.000 description 1
- 235000009122 Rubus idaeus Nutrition 0.000 description 1
- 230000001133 acceleration Effects 0.000 description 1
- 238000004458 analytical method Methods 0.000 description 1
- 238000013135 deep learning Methods 0.000 description 1
- 230000005611 electricity Effects 0.000 description 1
- 230000002708 enhancing effect Effects 0.000 description 1
- 238000011156 evaluation Methods 0.000 description 1
- 238000005286 illumination Methods 0.000 description 1
- 238000012544 monitoring process Methods 0.000 description 1
- 230000001537 neural effect Effects 0.000 description 1
- 238000007781 pre-processing Methods 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
- 230000009466 transformation Effects 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/0002—Inspection of images, e.g. flaw detection
- G06T7/0004—Industrial image inspection
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/24—Classification techniques
- G06F18/241—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T5/00—Image enhancement or restoration
- G06T5/90—Dynamic range modification of images or parts thereof
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20081—Training; Learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/30—Subject of image; Context of image processing
- G06T2207/30108—Industrial image inspection
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Engineering & Computer Science (AREA)
- Evolutionary Computation (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Computational Linguistics (AREA)
- Software Systems (AREA)
- Mathematical Physics (AREA)
- Health & Medical Sciences (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Computing Systems (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Evolutionary Biology (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Bioinformatics & Computational Biology (AREA)
- Quality & Reliability (AREA)
- Image Analysis (AREA)
Abstract
本发明涉及一种电力场景通用物品的目标检测方法、装置及存储介质,该方法包括:获取电力场景下的多幅样本图像,并对多幅样本图像进行数据增强,建立扩充数据集;根据带有标注信息的扩充数据集,确定训练样本集;将训练样本集输入至深度卷积神经网络模型,确定预测类别和预测目标框;根据实际类别、实际目标框、预测类别和预测目标框,确定损失函数,根据损失函数的值调整深度卷积神经网络模型的参数直至满足收敛条件,并将训练完备的深度卷积神经网络模型进行存储;将待测图像输入至训练完备的深度卷积神经网络模型中,识别预测通用物品。本发明基于小样本学习进行目标检测,可有效提高速度和精度。
Description
技术领域
本发明涉及目标检测技术领域,尤其涉及一种电力场景通用物品的目标检测方法、装置及存储介质。
背景技术
传统目标检测采用人工特征提取的方式,当背景环境和目标形态发生变化时,人工特征提取的方式鲁棒性较差,目标监测方法的精确度会降低。基于卷积神经网络的目标检测方法是将深度学习技术与计算机视觉技术相结合,具有局部感受野(视觉感受区域的大小)、结构层次、特征抽取与分类结合的全局训练特征。然而,现阶段视觉目标检测的研究方法和技术手段需要大量训练数据及复杂模型,存在数据标注成本太高、实时处理能力不强等问题。综上,如何提供实时性强、训练成本低的电力场景通用物品检测方法是亟待解决的问题。
发明内容
有鉴于此,有必要提供一种电力场景通用物品的目标检测方法、装置及存储介质,用以解决现有技术中电力场景目标检测实时性不高的问题。
本发明提供一种电力场景通用物品的目标检测方法,包括:
获取电力场景下的多幅样本图像,并对所述多幅样本图像进行数据增强,建立扩充数据集;
根据带有标注信息的所述扩充数据集,确定训练样本集,其中,标注信息经过数据增强后的多幅样本图像中存在的通用物品的实际类别和实际目标框;
将所述训练样本集输入至深度卷积神经网络模型,确定预测的通用物品的预测类别和预测目标框;
根据实际类别、实际目标框、预测类别和预测目标框,确定损失函数,根据所述损失函数的值调整所述深度卷积神经网络模型的参数直至满足收敛条件,完成对所述深度卷积神经网络模型的训练,并将通过使用基于数据增强的小样本学习方法训练完备的深度卷积神经网络模型进行存储;
将待测图像输入至训练完备的深度卷积神经网络模型中,识别预测通用物品,并输出对应的预测类别和预测目标框。
进一步地,所述获取多幅样本图像,并对所述多幅样本图像进行数据增强,建立扩充数据集包括:
获取所述多幅样本图像;
将所述多幅样本图像进行图像分块,确定对应的多张图像子块;
根据所述图像子块和对应的所述多张扩充子块,建立对每一张图像子块进行多种图像处理操作,得到对应的多张扩充子块;
所述多张扩充子块构成所述扩充数据集。
进一步地,所述多种图像处理操作包括调节图像亮度、调节图像色度、调节图像对比度、调节图像锐度以及人工加噪声中的至少一种。
进一步地,所述深度卷积神经网络模型的网络结构依次包括多个卷积层,每个卷积层具有对应的最大值池化层。
进一步地,所述根据实际类别、实际目标框、预测类别和预测目标框,确定损失函数包括:
根据所述预测目标框和所述实际目标框的重叠度,确定第一置信度损失;
根据所述预测目标框和所述实际目标框的误差,确定预测框误差损失;
计算将所述预测目标框和所述实际目标框的中心置于同一位置的重叠度,确定第二置信度损失;
根据所述预测目标框和所述实际目标框的坐标误差,确定坐标损失;
根据所述实际类别和所述预测类别,确定分类损失;
将所述第一置信度损失、所述预测框误差损失、所述第二置信度损失、所述坐标损失和所述分类损失和对应的权重相乘,确定所述损失函数。
进一步地,所述将待测图像输入至所述训练完备的深度卷积神经网络模型中,识别预测通用物品,并输出对应的预测类别和预测目标框包括:
采用特征工程,确定所述待测图像的目标候选区域;
将所述目标候选区域进行图像分块,确定对应的多张图像子块;
将所述多张图像子块输入至所述训练完备的深度卷积神经网络模型,识别预测通用物品,并采用非极大值抑制进行处理,输出对应的预测类别和预测目标框。
进一步地,所述待测图像通过电力作业机器人在电力场景下进行采集。
本发明还提供了一种电力场景通用物品的目标检测装置,包括:
获取单元,用于获取多幅样本图像,并对所述多幅样本图像进行数据增强,建立扩充数据集;
训练单元,用于根据带有标注信息的所述扩充数据集,确定训练样本集,其中,标注信息经过数据增强后的多幅样本图像中存在的通用物品的实际类别和实际目标框;还用于将所述训练样本集输入至深度卷积神经网络模型,确定预测的通用物品的预测类别和预测目标框;还用于根据实际类别、实际目标框、预测类别和预测目标框,确定损失函数,根据所述损失函数的值调整所述深度卷积神经网络模型的参数直至满足收敛条件,完成对所述深度卷积神经网络模型的训练,并将通过使用基于数据增强的小样本学习方法训练完备的深度卷积神经网络模型进行存储;
检测单元,用于将待测图像输入至所述训练完备的深度卷积神经网络模型中,识别预测通用物品,并输出对应的预测类别和预测目标框。
本发明还提供了一种电力场景通用物品的目标检测装置,包括处理器以及存储器,存储器上存储有计算机程序,计算机程序被处理器执行时,实现如上所述的电力场景通用物品的目标检测方法。
本发明还提供了一种计算机可读存储介质,所述计算机该程序被处理器执行时,实现如上所述的电力场景通用物品的目标检测方法。
与现有技术相比,本发明的有益效果包括:首先,对电力场景下的多幅样本图像进行有效的获取,并对样本图像进行数据增强,扩充训练样本,避免重复繁琐的标注工作;然后,针对扩充数据集进行标注,由于扩充数据集是基于多幅样本图像的,因而标注效率可大量提高,快速得到充分的训练样本集;进而,基于实际类别、实际目标框、预测类别和预测目标框进行深度卷积神经网络模型的训练,并将训练完备的深度卷积神经网络模型进行存储;最后,将待测图像作为输入,利用所述训练完备的深度卷积神经网络模型即可识别预测通用物品。综上,本发明基于数据增强的小样本学习方法,有效避免环境变化的干扰,能准确地识别目标,同时,基于小样本学习的目标检测模型,避免了繁琐的数据标注过程,节约了训练成本,且检测速度快、检测准确率高,可用于物品实时自动识别检测,在保证速度的前提下,可提升相应的准确率,能对不同环境下的物品进行有效识别。
附图说明
图1为本发明提供的电力场景通用物品的目标检测方法一实施例的流程示意图;
图2为本发明提供的图1中步骤S1一实施例的流程示意图;
图3为本发明提供的深度卷积神经网络模型一实施例的网络结构示意图;
图4为本发明提供的图1中步骤S4一实施例的流程示意图;
图5为本发明提供的图1中步骤S5一实施例的流程示意图;
图6为本发明提供的电力场景通用物品的目标检测装置一实施例的结构示意图。
具体实施方式
下面结合附图来具体描述本发明的优选实施例,其中,附图构成本申请一部分,并与本发明的实施例一起用于阐释本发明的原理,并非用于限定本发明的范围。
本发明实施例提供了一种电力场景通用物品的目标检测方法,结合图1来看,图1为本发明提供的电力场景通用物品的目标检测方法一实施例的流程示意图,包括步骤S1至步骤S5,其中:
在步骤S1中,获取电力场景下的多幅样本图像,并对所述多幅样本图像进行数据增强,建立扩充数据集;其中,数据增强包括多种图像处理的方式,使样本图像得到扩充;
在步骤S2中,根据带有标注信息的所述扩充数据集,确定训练样本集,其中,标注信息经过数据增强后的多幅样本图像中存在的通用物品的实际类别和实际目标框;其中,通用物品包括电力场景物品,具体包括,输电物品:输电铁塔、输电电线、输电电缆;变电物品:变压器、断路器、隔离开关;配电物品:配电柜、熔断器、按钮、指示灯、仪表;目标框包括目标的位置坐标,目标框的宽、高;
在步骤S3中,将所述训练样本集输入至深度卷积神经网络模型,确定预测的通用物品的预测类别和预测目标框;
在步骤S4中,根据实际类别、实际目标框、预测类别和预测目标框,确定损失函数,根据所述损失函数的值调整所述深度卷积神经网络模型的参数直至满足收敛条件,完成对所述深度卷积神经网络模型的训练,并将通过使用基于数据增强的小样本学习方法训练完备的深度卷积神经网络模型进行存储;
在步骤S5中,将待测图像输入至所述训练完备的深度卷积神经网络模型中,识别预测通用物品,并输出对应的预测类别和预测目标框。
在本发明实施例中,首先,对电力场景下的多幅样本图像进行有效的获取,并对样本图像进行数据增强,扩充训练样本,避免重复繁琐的标注工作;然后,针对扩充数据集进行标注,由于扩充数据集是基于多幅样本图像的,因而标注效率可大量提高,快速得到充分的训练样本集;进而,基于实际类别、实际目标框、预测类别和预测目标框进行深度卷积神经网络模型的训练,并将训练完备的深度卷积神经网络模型进行存储;最后,将待测图像作为输入,利用所述训练完备的深度卷积神经网络模型即可识别预测通用物品。
作为优选的实施例,结合图2来看,图2为本发明提供的图1中步骤S1一实施例的流程示意图,上述步骤S1包括步骤S11至步骤S14,其中:
在步骤S11中,获取所述多幅样本图像;
在步骤S12中,将所述多幅样本图像进行图像分块,确定对应的多张图像子块;
在步骤S13中,根据所述图像子块和对应的所述多张扩充子块,建立对每一张图像子块进行多种图像处理操作,得到对应的多张扩充子块;
在步骤S14中,所述多张扩充子块构成所述扩充数据集。
作为具体实施例,本发明实施例对数据进行有效的扩充,同时将样本数据减少,保证模型简单快速地进行检测,且避免了繁琐的标注工作。
作为优选的实施例,所述多种图像处理操作包括调节图像亮度、调节图像色度、调节图像对比度、调节图像锐度以及人工加噪声中的至少一种。作为具体实施例,本发明实施例采用多种图像处理操作手段,对样本图像进行有效的扩充。此处理在实际效果上有助于低照度可见光下、复杂背景下的电力场景物品识别,样本图像的数据增强有助于缓解过拟合问题,同时训练出的最终模型会具有更好的鲁棒性和泛化能力。
在本发明一个具体的实施例中,使用网络爬虫获取电力场景下的样本图像3000幅,并采集各种电力场景下图像12000幅,组成电力场景下通用物品检测数据库。
在本发明一个具体的实施例中,对样本图像进行预处理,将图像分成13*13块,对样本图像进行数据增强,具体包括图像亮度、色度、对比度、锐度以及人工加噪声等多种处理。通过数据增强后,图像数据集扩充到4万张。并扩充数据集,对数据集进行标注,得到训练样本集。
作为优选的实施例,结合图3来看,图3为本发明提供的深度卷积神经网络模型一实施例的网络结构示意图,所述深度卷积神经网络模型的网络结构依次包括多个卷积层,每个卷积层具有对应的最大值池化层。作为具体实施例,本发明实施例建立网络结构层次,保证模型检测的精度和速度。
在本发明一个具体的实施例中,构建16层深度卷积神经网络模型M。使用数据库对16层卷积神经网络模型M进行预训练,得到预检测模型M1;使用训练样本集对预检测模型M1进行二次训练,得最终目标检测模型M2。其中,对样本图像进行缩放,将图像分辨率缩小为512×512像素,然后对图像中目标进行人工标注,得到子图像。使用最终目标检测模型M2对子图像进行特征提取,得到目标框。
采用基于区域的深度卷积神经网络,在卷积层后接入感兴趣区域池化层,然后接入全连接层。本文采用的深度卷积神经网络结构参数如表1所示。算法模型如下式所示:
y=M2(x)
其中,x为输入图像,y为物品分类以及其位置坐标,M2为卷积神经网络模型。网络由卷积网络层和最大值池化层组成。卷积网络层用于提取图像的特征,最大值池化层用于选择重要特征,使用最终目标检测模型M2对子图像进行特征提取。相关卷积网络结构参数如表1所示。
表1
作为优选的实施例,结合图4来看,图4为本发明提供的图1中步骤S4一实施例的流程示意图一,上述步骤S4包括步骤S41至步骤S46,其中:
在步骤S41中,根据所述预测目标框和所述实际目标框的重叠度,确定第一置信度损失;
在步骤S42中,根据所述预测目标框和所述实际目标框的误差,确定预测框误差损失;
在步骤S43中,计算将所述预测目标框和所述实际目标框的中心置于同一位置的重叠度,确定第二置信度损失;
在步骤S44中,根据所述预测目标框和所述实际目标框的坐标误差,确定坐标损失;
在步骤S45中,根据所述实际类别和所述预测类别,确定分类损失;
在步骤S46中,将所述第一置信度损失、所述预测框误差损失、所述第二置信度损失、所述坐标损失和所述分类损失和对应的权重相乘,确定所述损失函数。
作为具体实施例,本发明实施例结合实际类别、实际目标框、预测类别和预测目标框对训练函数进行有效的确定,保证网络的训练高效性,进而提高整个网络的识别准确度。
在本发明一个具体的实施例中,模型训练过程采用的损失函数如下式所示:
其中,λnoobjλpriorλcoordλobjλclass为不同类型的损失函数的权重。noobj是指矩形框里没有目标,obj是指矩形框里有目标,coord是指矩形框坐标,class目标种类,这里为12种目标,IOU是指叠加区域。i、j为13*13图像分块的行号和列号,k为预测框号。b代表目标框,r(x,y,w,h)是目标框的位置坐标(x,y)和矩形框宽和高(w,h)。
作为优选的实施例,结合图5来看,图5为本发明提供的图1中步骤S5一实施例的流程示意图,上述步骤S5包括步骤S51至步骤S53,其中:
在步骤S51中,采用特征工程,确定所述待测图像的目标候选区域;
在步骤S52中,将所述目标候选区域进行图像分块,确定对应的多张图像子块;
在步骤S53中,将所述多张图像子块输入至所述训练完备的深度卷积神经网络模型,识别预测通用物品,并输出对应的预测类别和预测目标框。
作为具体实施例,本发明实施例根据电力场景下作业机器人采集的图像进行预处理;采用特征工程确定目标候选区域,基于小样本学习的卷积神经网路模型进行目标检测,可有效提高速度和精度。
作为优选的实施例,所述待测图像通过电力作业机器人在电力场景下进行采集。作为具体实施例,本发明实施例利用电力作业机器人进行有效的图像采集。
需要说明的是,采用TensorFlow2.0等平台以及标准开源库对本文算法性能进行分析实验,并在树莓派3B平台上加装Intel神经计算加速棒进行算法实测比较;根据试验结果,调整参数重复实验。在不同条件下检验算法的实际效果,并根据实测结果对算法进行分析以及优化。为了兼顾跟踪性能和系统开销,采用手动调参实现信息的有效处理,并进行评估,当达到精度要求时终止训练过程,其中,系统测试平台参数如下表2所示:
表2
在本发明一个具体的实施例中,本发明提供的方法应用于电力作业机器人。具体实施场景如下:电力作业机器人的摄像头拍摄真实图像——对图像进行预处理(缩小)——用特征工程得到图像中的候选区域——事先训练好的检测模型(通过基于数据增强的小样本学习训练)——候选目标(产生多目标框)——最终目标(非极大值抑制)。利用小样本训练得到最终检测模型,并应用于实际场景。
本发明实施例还提供了一种电力场景通用物品的目标检测装置,结合图6来看,图6为本发明提供的电力场景通用物品的目标检测装置一实施例的结构示意图,电力场景通用物品的目标检测装置600包括:
获取单元601,用于获取多幅样本图像,并对所述多幅样本图像进行数据增强,建立扩充数据集;
训练单元602,用于根据带有标注信息的所述扩充数据集,确定训练样本集,其中,标注信息经过数据增强后的多幅样本图像中存在的通用物品的实际类别和实际目标框;还用于将所述训练样本集输入至深度卷积神经网络模型,确定预测的通用物品的预测类别和预测目标框;还用于根据实际类别、实际目标框、预测类别和预测目标框,确定损失函数,根据所述损失函数的值调整所述深度卷积神经网络模型的参数直至满足收敛条件,完成对所述深度卷积神经网络模型的训练,并将通过使用基于数据增强的小样本学习方法训练完备的深度卷积神经网络模型进行存储;
检测单元603,用于将待测图像输入至所述训练完备的深度卷积神经网络模型中,识别预测通用物品,并输出对应的预测类别和预测目标框。
本发明实施例还提供了一种电力场景通用物品的目标检测装置,包括处理器以及存储器,存储器上存储有计算机程序,计算机程序被处理器执行时,实现如上所述的电力场景通用物品的目标检测方法。
本发明实施例还提供了一种计算机可读存储介质,其上存储有计算机程序,计算机程序被处理器执行时,实现如上所述的电力场景通用物品的目标检测方法。
本发明公开了一种电力场景通用物品的目标检测方法、装置及存储介质,首先,对电力场景下的多幅样本图像进行有效的获取,并对样本图像进行数据增强,扩充训练样本,避免重复繁琐的标注工作;然后,针对扩充数据集进行标注,由于扩充数据集是基于多幅样本图像的,因而标注效率可大量提高,快速得到充分的训练样本集;进而,基于实际类别、实际目标框、预测类别和预测目标框进行深度卷积神经网络模型的训练,并将训练完备的深度卷积神经网络模型进行存储;最后,将待测图像作为输入,利用所述训练完备的深度卷积神经网络模型即可识别预测通用物品。
本发明技术方案,基于数据增强的小样本学习方法,有效避免环境变化的干扰,能准确地识别目标,同时,基于小样本学习的目标检测模型,避免了繁琐的数据标注过程,节约了训练成本,且检测速度快、检测准确率高,可用于物品实时自动识别检测,在保证速度的前提下,可提升相应的准确率,能对不同环境下的物品进行有效识别。
以上所述,仅为本发明较佳的具体实施方式,但本发明的保护范围并不局限于此,任何熟悉本技术领域的技术人员在本发明揭露的技术范围内,可轻易想到的变化或替换,都应涵盖在本发明的保护范围之内。
Claims (10)
1.一种电力场景通用物品的目标检测方法,其特征在于,包括:
获取电力场景下的多幅样本图像,并对所述多幅样本图像进行数据增强,建立扩充数据集;
根据带有标注信息的所述扩充数据集,确定训练样本集,其中,标注信息经过数据增强后的多幅样本图像中存在的通用物品的实际类别和实际目标框;
将所述训练样本集输入至深度卷积神经网络模型,确定预测的通用物品的预测类别和预测目标框;
根据实际类别、实际目标框、预测类别和预测目标框,确定损失函数,根据所述损失函数的值调整所述深度卷积神经网络模型的参数直至满足收敛条件,完成对所述深度卷积神经网络模型的训练,并将通过使用基于数据增强的小样本学习方法训练完备的深度卷积神经网络模型进行存储;
将待测图像输入至训练完备的深度卷积神经网络模型中,识别预测通用物品,并输出对应的预测类别和预测目标框。
2.根据权利要求1所述的电力场景通用物品的目标检测方法,其特征在于,所述获取多幅样本图像,并对所述多幅样本图像进行数据增强,建立扩充数据集包括:
获取所述多幅样本图像;
将所述多幅样本图像进行图像分块,确定对应的多张图像子块;
根据所述图像子块和对应的所述多张扩充子块,建立对每一张图像子块进行多种图像处理操作,得到对应的多张扩充子块;
所述多张扩充子块构成所述扩充数据集。
3.根据权利要求2所述的电力场景通用物品的目标检测方法,其特征在于,所述多种图像处理操作包括调节图像亮度、调节图像色度、调节图像对比度、调节图像锐度以及人工加噪声中的至少一种。
4.根据权利要求1所述的电力场景通用物品的目标检测方法,其特征在于,所述深度卷积神经网络模型的网络结构依次包括多个卷积层,每个卷积层具有对应的最大值池化层。
5.根据权利要求1所述的电力场景通用物品的目标检测方法,其特征在于,所述根据实际类别、实际目标框、预测类别和预测目标框,确定损失函数包括:
根据所述预测目标框和所述实际目标框的重叠度,确定第一置信度损失;
根据所述预测目标框和所述实际目标框的误差,确定预测框误差损失;
计算将所述预测目标框和所述实际目标框的中心置于同一位置的重叠度,确定第二置信度损失;
根据所述预测目标框和所述实际目标框的坐标误差,确定坐标损失;
根据所述实际类别和所述预测类别,确定分类损失;
将所述第一置信度损失、所述预测框误差损失、所述第二置信度损失、所述坐标损失和所述分类损失和对应的权重相乘,确定所述损失函数。
6.根据权利要求1所述的电力场景通用物品的目标检测方法,其特征在于,所述将待测图像输入至所述训练完备的深度卷积神经网络模型中,识别预测通用物品,并输出对应的预测类别和预测目标框包括:
采用特征工程,确定所述待测图像的目标候选区域;
将所述目标候选区域进行图像分块,确定对应的多张图像子块;
将所述多张图像子块输入至所述训练完备的深度卷积神经网络模型,识别预测通用物品,并采用非极大值抑制进行处理,输出对应的预测类别和预测目标框。
7.根据权利要求1所述的电力场景通用物品的目标检测方法,其特征在于,所述待测图像通过电力作业机器人在电力场景下进行采集。
8.一种电力场景通用物品的目标检测装置,其特征在于,包括:
获取单元,用于获取多幅样本图像,并对所述多幅样本图像进行数据增强,建立扩充数据集;
训练单元,用于根据带有标注信息的所述扩充数据集,确定训练样本集,其中,标注信息经过数据增强后的多幅样本图像中存在的通用物品的实际类别和实际目标框;还用于将所述训练样本集输入至深度卷积神经网络模型,确定预测的通用物品的预测类别和预测目标框;还用于根据实际类别、实际目标框、预测类别和预测目标框,确定损失函数,根据所述损失函数的值调整所述深度卷积神经网络模型的参数直至满足收敛条件,完成对所述深度卷积神经网络模型的训练,并将通过使用基于数据增强的小样本学习方法训练完备的深度卷积神经网络模型进行存储;
检测单元,用于将待测图像输入至所述训练完备的深度卷积神经网络模型中,识别预测通用物品,并输出对应的预测类别和预测目标框。
9.一种电力场景通用物品的目标检测装置,其特征在于,包括处理器以及存储器,存储器上存储有计算机程序,计算机程序被处理器执行时,实现根据权利要求1-7任一项所述的电力场景通用物品的目标检测方法。
10.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机该程序被处理器执行时,实现如权利要求1-7任一项所述的电力场景通用物品的目标检测方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110487848.1A CN113344852A (zh) | 2021-04-30 | 2021-04-30 | 一种电力场景通用物品的目标检测方法、装置及存储介质 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110487848.1A CN113344852A (zh) | 2021-04-30 | 2021-04-30 | 一种电力场景通用物品的目标检测方法、装置及存储介质 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN113344852A true CN113344852A (zh) | 2021-09-03 |
Family
ID=77469577
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110487848.1A Withdrawn CN113344852A (zh) | 2021-04-30 | 2021-04-30 | 一种电力场景通用物品的目标检测方法、装置及存储介质 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113344852A (zh) |
Cited By (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113762397A (zh) * | 2021-09-10 | 2021-12-07 | 北京百度网讯科技有限公司 | 检测模型训练、高精度地图更新方法、设备、介质及产品 |
CN113807424A (zh) * | 2021-09-10 | 2021-12-17 | 华侨大学 | 一种基于离线数据增强的固废数据集生成方法 |
CN113837086A (zh) * | 2021-09-24 | 2021-12-24 | 南通大学 | 一种基于深度卷积神经网络的水库钓鱼人检测方法 |
CN114529751A (zh) * | 2021-12-28 | 2022-05-24 | 国网四川省电力公司眉山供电公司 | 一种电力场景智能识别样本数据的自动筛选方法 |
CN114612702A (zh) * | 2022-01-24 | 2022-06-10 | 珠高智能科技(深圳)有限公司 | 基于深度学习的图像数据标注系统及其方法 |
CN115035406A (zh) * | 2022-06-08 | 2022-09-09 | 中国科学院空间应用工程与技术中心 | 遥感场景数据集的标注方法、系统、存储介质及电子设备 |
-
2021
- 2021-04-30 CN CN202110487848.1A patent/CN113344852A/zh not_active Withdrawn
Cited By (9)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113762397A (zh) * | 2021-09-10 | 2021-12-07 | 北京百度网讯科技有限公司 | 检测模型训练、高精度地图更新方法、设备、介质及产品 |
CN113807424A (zh) * | 2021-09-10 | 2021-12-17 | 华侨大学 | 一种基于离线数据增强的固废数据集生成方法 |
CN113762397B (zh) * | 2021-09-10 | 2024-04-05 | 北京百度网讯科技有限公司 | 检测模型训练、高精度地图更新方法、设备、介质及产品 |
CN113807424B (zh) * | 2021-09-10 | 2024-04-16 | 华侨大学 | 一种基于离线数据增强的固废数据集生成方法 |
CN113837086A (zh) * | 2021-09-24 | 2021-12-24 | 南通大学 | 一种基于深度卷积神经网络的水库钓鱼人检测方法 |
CN114529751A (zh) * | 2021-12-28 | 2022-05-24 | 国网四川省电力公司眉山供电公司 | 一种电力场景智能识别样本数据的自动筛选方法 |
CN114612702A (zh) * | 2022-01-24 | 2022-06-10 | 珠高智能科技(深圳)有限公司 | 基于深度学习的图像数据标注系统及其方法 |
CN115035406A (zh) * | 2022-06-08 | 2022-09-09 | 中国科学院空间应用工程与技术中心 | 遥感场景数据集的标注方法、系统、存储介质及电子设备 |
CN115035406B (zh) * | 2022-06-08 | 2023-08-04 | 中国科学院空间应用工程与技术中心 | 遥感场景数据集的标注方法、系统、存储介质及电子设备 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN113344852A (zh) | 一种电力场景通用物品的目标检测方法、装置及存储介质 | |
CN111339858B (zh) | 一种基于神经网络的油气管道标志物识别方法 | |
CN112199993B (zh) | 基于人工智能识别任意方向变电站绝缘子红外图像检测模型的方法 | |
CN111738110A (zh) | 基于多尺度注意力机制的遥感图像车辆目标检测方法 | |
CN108009515B (zh) | 一种基于fcn的无人机航拍图像的输电线定位识别方法 | |
CN112734692A (zh) | 一种变电设备缺陷识别方法及装置 | |
CN111695622A (zh) | 变电作业场景的标识模型训练方法、标识方法及装置 | |
CN108010025B (zh) | 基于rcnn的屏柜的开关与指示灯定位和状态识别方法 | |
CN114241332A (zh) | 一种基于深度学习的固废场识别方法、装置以及存储介质 | |
CN115546223A (zh) | 一种列车车下设备紧固螺栓的缺失检测方法和系统 | |
CN116071315A (zh) | 一种基于机器视觉的产品可视缺陷检测方法及系统 | |
CN111507398A (zh) | 基于目标检测的变电站金属仪器锈蚀识别方法 | |
CN114693963A (zh) | 基于电力数据特征提取的识别模型训练、识别方法及装置 | |
CN113052103A (zh) | 一种基于神经网络的电气设备缺陷检测方法及装置 | |
CN117456441A (zh) | 一种结合变化区域识别的锈斑面积扩大的监测方法与系统 | |
CN111582332A (zh) | 一种高速铁路接触网吊弦部件的图片识别方法 | |
CN114596433A (zh) | 一种绝缘子识别方法 | |
CN112287740B (zh) | 基于YOLOv3-tiny实现的输电线路的目标检测方法、装置、无人机 | |
CN116503398B (zh) | 绝缘子污闪检测方法、装置、电子设备及存储介质 | |
CN112667832B (zh) | 一种未知室内环境下基于视觉的互定位方法 | |
CN114648736B (zh) | 基于目标检测的鲁棒工程车辆识别方法及系统 | |
CN117911317A (zh) | 板材图像的缺陷检测方法、装置、电子设备及存储介质 | |
CN117333476A (zh) | 一种基于微服务的硅钢片视觉质量检测方法及实验系统 | |
CN117152139A (zh) | 一种基于实例分割技术的贴片电感缺陷检测方法 | |
CN116958065A (zh) | 一种基于滑动窗口亲和力的芯片载体封装漏装检测方法 |
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 | ||
WW01 | Invention patent application withdrawn after publication |
Application publication date: 20210903 |
|
WW01 | Invention patent application withdrawn after publication |