CN113688709A - 一种安全帽佩戴智能检测方法、系统、终端及介质 - Google Patents
一种安全帽佩戴智能检测方法、系统、终端及介质 Download PDFInfo
- Publication number
- CN113688709A CN113688709A CN202110943360.5A CN202110943360A CN113688709A CN 113688709 A CN113688709 A CN 113688709A CN 202110943360 A CN202110943360 A CN 202110943360A CN 113688709 A CN113688709 A CN 113688709A
- Authority
- CN
- China
- Prior art keywords
- model
- yolo
- scale
- safety helmet
- wearing
- 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.)
- Granted
Links
- 238000001514 detection method Methods 0.000 title claims abstract description 91
- 238000012549 training Methods 0.000 claims abstract description 41
- 238000000034 method Methods 0.000 claims abstract description 29
- 238000012545 processing Methods 0.000 claims abstract description 20
- 238000007781 pre-processing Methods 0.000 claims abstract description 9
- 230000008569 process Effects 0.000 claims abstract description 9
- 238000005070 sampling Methods 0.000 claims abstract description 6
- 238000004422 calculation algorithm Methods 0.000 claims description 24
- 238000004590 computer program Methods 0.000 claims description 14
- 238000012360 testing method Methods 0.000 claims description 12
- 230000004927 fusion Effects 0.000 claims description 8
- 230000006870 function Effects 0.000 claims description 7
- 238000004458 analytical method Methods 0.000 claims description 6
- 238000012544 monitoring process Methods 0.000 claims description 6
- 238000000605 extraction Methods 0.000 claims description 4
- 230000004913 activation Effects 0.000 claims description 3
- 238000011161 development Methods 0.000 claims description 3
- 238000010586 diagram Methods 0.000 description 8
- 230000000694 effects Effects 0.000 description 8
- 238000010276 construction Methods 0.000 description 5
- 238000011897 real-time detection Methods 0.000 description 4
- 238000004364 calculation method Methods 0.000 description 3
- 230000007547 defect Effects 0.000 description 2
- 238000011156 evaluation Methods 0.000 description 2
- 230000009286 beneficial effect Effects 0.000 description 1
- 238000013500 data storage Methods 0.000 description 1
- 238000013135 deep learning Methods 0.000 description 1
- 238000013461 design Methods 0.000 description 1
- 230000008034 disappearance Effects 0.000 description 1
- 238000002474 experimental method Methods 0.000 description 1
- 238000004880 explosion Methods 0.000 description 1
- 238000005286 illumination Methods 0.000 description 1
- 238000002372 labelling Methods 0.000 description 1
- 238000004519 manufacturing process Methods 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 210000002569 neuron Anatomy 0.000 description 1
- 230000003287 optical effect Effects 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 238000007790 scraping Methods 0.000 description 1
- 230000035945 sensitivity Effects 0.000 description 1
- 238000010206 sensitivity analysis Methods 0.000 description 1
- 238000012795 verification Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/23—Clustering techniques
- G06F18/232—Non-hierarchical techniques
-
- 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
-
- Y—GENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
- Y02—TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
- Y02P—CLIMATE CHANGE MITIGATION TECHNOLOGIES IN THE PRODUCTION OR PROCESSING OF GOODS
- Y02P90/00—Enabling technologies with a potential contribution to greenhouse gas [GHG] emissions mitigation
- Y02P90/30—Computing systems specially adapted for manufacturing
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Data Mining & Analysis (AREA)
- Physics & Mathematics (AREA)
- Evolutionary Computation (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Physics & Mathematics (AREA)
- General Engineering & Computer Science (AREA)
- General Health & Medical Sciences (AREA)
- Software Systems (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- Biophysics (AREA)
- Biomedical Technology (AREA)
- Mathematical Physics (AREA)
- Computational Linguistics (AREA)
- Health & Medical Sciences (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Bioinformatics & Computational Biology (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Evolutionary Biology (AREA)
- Image Analysis (AREA)
Abstract
本发明公开了一种安全帽佩戴智能检测方法、系统、终端及介质,涉及计算机视觉识别技术领域,其技术方案要点是:对采集图像数据预处理后建立数据集;依据训练集对改进YOLO v3模型训练后得到训练后的改进YOLO v3模型,并通过改进YOLO v3模型检测目标图像中的工作人员是否佩戴安全帽;改进YOLOv3模型进行检测的过程具体为:通过Darknet‑53网络的图像特征金字塔对输入图像进行多尺度处理,得到多尺度特征图;将原始YOLO v3模型中的全连接操作替换为全卷积操作,并以上采样方法将低层特征和高层特征融合,得到新特征图进行安全帽佩戴检测。本发明均衡提升了检测准确度和速度。
Description
技术领域
本发明涉及计算机视觉识别技术领域,更具体地说,它涉及一种安全帽佩戴智能检测方法、系统、终端及介质。
背景技术
施工现场工人佩戴安全帽以在遇到突发事件时有效地保护工人的头部。然而,工人可能因为安全意识薄弱等原因,在施工现场并未佩戴安全帽,从而为工人带来了较高安全隐患。因此,对施工现场工人是否佩戴安全帽进行检测非常有必要。
目前,对于安全帽佩戴智能检测的方法主要有Faster R-CNN、SSD、Light_YOLO_v3,但SSD有时会受环境和图片背景的影响出现误检。而在多目标、小目标并存的情况下,SSD算法会出现漏检的情况,Faster R-CNN、Light_YOLO_v3算法对小目标和多目标检测表现相对较好,但Faster R-CNN目标检测算法是two stage模型,对于候选区域的确定需要进行大量计算,在检测速度上有待提升。而Light_YOLO_v3算法在检测准确度上有待进一步提升。
因此,如何研究设计一种能够克服上述缺陷的安全帽佩戴智能检测方法、系统、终端及介质是我们目前急需解决的问题。
发明内容
为解决现有技术中的不足,本发明的目的是提供一种安全帽佩戴智能检测方法、系统、终端及介质。
本发明的上述技术目的是通过以下技术方案得以实现的:
第一方面,提供了一种安全帽佩戴智能检测方法,包括以下步骤:
对采集图像数据预处理后建立数据集,数据集划分为训练集和测试集;
依据训练集对改进YOLO v3模型训练后得到训练后的改进YOLO v3模型,并通过改进YOLO v3模型检测目标图像中的工作人员是否佩戴安全帽;
改进YOLO v3模型进行检测的过程具体为:
通过Darknet-53网络的图像特征金字塔对输入图像进行多尺度处理,得到多尺度特征图;
将原始YOLO v3模型中的全连接操作替换为全卷积操作,并以上采样方法将低层特征和高层特征融合,得到新特征图进行安全帽佩戴检测。
进一步的,所述改进YOLO v3模型的网络结构具体为:
通过Darknet-53的图像特征金字塔,对第53层卷积层进行1×1及3×3的卷积操作,并重复进行一次1×1及3×3的卷积操作,得到新的小尺度YOLO层;
将新的小尺度YOLO层采取上采样操作,并与Darknet-53的第45个卷积层进行卷积和,使用连续两次的1×1和3×3卷积操作,得到新的中尺度YOLO层;
将新的中尺度YOLO层与Darknet-53的第29个卷积层进行卷积和,连续使用两次1×1和3×3卷积操作,得到新的大尺度YOLO层;
以三组新的不同尺度的YOLO层进行安全帽佩戴检测。
进一步的,所述改进YOLO v3模型的基础学习率为0.001,激活函数选用ReLU,Batch-size为32,每迭代100次计算一次损失值。
进一步的,所述Darknet-53网络具体为:
将基于标准卷积的Darknet-53的特征提取网络替换为基于深度可分离卷积的Darknet-53网络;
同时保留原基础网络中的残差网络结构,得到最终的Darknet-53网络。
进一步的,该方法还包括:
采用k-means维度聚类算法对训练集进行聚类分析,得到多组新的目标先验框;
将多组新的目标先验框按从小达到的顺序均分到不同尺度的特征图,并将较小的先验框分配给尺度更大的特征图。
进一步的,所述改进YOLO v3模型的训练过程具体为:
根据训练集对原始YOLO v3模型进行训练,先进行前向传播训练,再反向传播修正权值,得到与检测目标接近的新权重;
对改进YOLO v3模型进行网络结构调整后将新权重复制到改进模型中,得到最终的改进YOLO v3模型。
进一步的,所述采集图像数据的预处理具体为:
采用开发工具将工业企业的视频监控转换为图片格式,并采用Faster R-CNN算法筛选出图片数据中包含行人的图片作为数据集图片,放入待标记数据集文件夹中;
采用labelImg工具对待标记数据集中图像的目标位置以矩形框进行标注,得到预处理后的数据集。
第二方面,提供了一种安全帽佩戴智能检测系统,包括:
数据处理模块,用于对采集图像数据预处理后建立数据集,数据集划分为训练集和测试集;
模型检测模块,用于依据训练集对改进YOLO v3模型训练后得到训练后的改进YOLO v3模型,并通过改进YOLO v3模型检测目标图像中的工作人员是否佩戴安全帽;
模型检测模块包括多尺度处理单元和特征融合单元;
多尺度处理单元,用于通过Darknet-53网络的图像特征金字塔对输入图像进行多尺度处理,得到多尺度特征图;
特征融合单元,用于将原始YOLO v3模型中的全连接操作替换为全卷积操作,并以上采样方法将低层特征和高层特征融合,得到新特征图进行安全帽佩戴检测。
第三方面,提供了一种计算机终端,包含存储器、处理器及存储在存储器并可在处理器上运行的计算机程序,所述处理器执行所述程序时实现如第一方面中任意一项所述的一种安全帽佩戴智能检测方法。
第四方面,提供了一种计算机可读介质,其上存储有计算机程序,所述计算机程序被处理器执行可实现如第一方面中任意一项所述的一种安全帽佩戴智能检测方法。
与现有技术相比,本发明具有以下有益效果:
1、本发明提出的改进YOLO v3模型,使用全卷积层替换掉全连接层后,对于不同尺寸的输入图像进行训练,可以提高网络对输入图像尺寸的鲁棒性。
2、本发明提出的改进YOLO v3模型,对不同分辨率的输入图像进行检测,结果检测准确率明显有提高,其对安全帽佩戴检测的多尺度训练能够有效提高算法鲁棒性;
3、本发明提出的改进YOLO v3模型相比于原始YOLO v3模型而言,对不同目标尺寸的检测准确率更高,通过使用基于图像金字塔的多层特征融合的策略,在对安全帽等小目标进行检测的效果显著;
4、本发明将基于标准卷积的Darknet-53的特征提取网络替换为基于深度可分离卷积的Darknet-53网络,同时保留原基础网络中的残差网络结构,有效避免了梯度消失、梯度爆炸或信息丢失问题出现;
5、本发明提出的改进YOLO v3模型的mAP值为高达87.12%,虽然其检测准确率不及Faster R-CNN,但在检测速度方面,比Faster R-CNN明显快的多,所以改进YOLO v3模型对于实时检测任务会表现更好;此外,改进YOLO v3模型在mAP和识别帧率两项指标上比SSD算法和原始YOLO v3模型均有提高,可见,改进YOLO v3模型在检测准确率和检测速度上都达到了优异的水平,能够满足工业企业智能监控系统实时性检测要求。
附图说明
此处所说明的附图用来提供对本发明实施例的进一步理解,构成本申请的一部分,并不构成对本发明实施例的限定。在附图中:
图1是本发明实施例中小目标检测敏感性分析结果图;
图2是本发明实施例中低中高分辨率敏感性分析结果图。
具体实施方式
为使本发明的目的、技术方案和优点更加清楚明白,下面结合实施例和附图,对本发明作进一步的详细说明,本发明的示意性实施方式及其说明仅用于解释本发明,并不作为对本发明的限定。
实施例1:一种安全帽佩戴智能检测方法,具体由以下步骤实现。
一、数据集构建
1、数据集建立
为了保证自建数据集的准确性,满足训练和测试要求,在数据采集时要满足以下要求:一是数据集的图像必须有多样性,包括图片拍摄时的光照强度、拍摄距离、拍摄季节等,收集的数据有差异性,训练的模型才能有更好的鲁棒性;而是采集的数据要有一定的数量,以达到训练的规模。按照以上数据采集要求,可以通过到工业企业去提取现场监控视频;还可以通过网络扒取获得相应图片。图片所包含的要素是必须有作业人员,其中有部分图片中的作业人员未佩戴安全帽。最终,收集的数据集共有5200张图片,包含待检测目标15000个左右,自建数据集名称为“FacImage”。
2、数据预处理和标记
采用开发工具将工业企业的视频监控转换为图片格式,并采用Faster R-CNN算法筛选出图片数据中包含行人的图片作为数据集图片,放入待标记数据集文件夹中;采用labelImg工具对待标记数据集中图像的目标位置以矩形框进行标注,得到预处理后的数据集。
labelImg工具操作过程具体为:打开图片所在文件夹,选择标记后数据存储文件夹,在右边区块的标签中,增加预设的标签,然后选择标签,在图像中使用矩形框标注出该标签的目标位置,当所有矩形框标注完后,选择下一个图片进行标记。对于自建“FacImage”数据集,标签分为2类:一类为Helmat即已佩戴安全帽人员,另一类为alarm即未佩戴安全帽人员,这些标记数据将以XML文件保存,Python可直接读取相关数据。
由于训练和测试的需要,本发明将自建数据集随机划分为两部分,其中训练集占整个数据集的70%,测试集占整个数据集的30%。在训练时,因改进YOLO v3模型较原始YOLO v3模型增加了数个卷积层,所以设置原始YOLO v3模型的基础学习率为0.001,模型其他参数均保持不变,激活函数选用ReLU,Batch-size为32,每迭代100次计算一次损失值。
二、模型构建
改进YOLO v3模型的网络结构具体为:通过Darknet-53的图像特征金字塔,对第53层卷积层进行1×1及3×3的卷积操作,并重复进行一次1×1及3×3的卷积操作,得到新的小尺度YOLO层;将新的小尺度YOLO层采取上采样操作,并与Darknet-53的第45个卷积层进行卷积和,使用连续两次的1×1和3×3卷积操作,得到新的中尺度YOLO层;将新的中尺度YOLO层与Darknet-53的第29个卷积层进行卷积和,连续使用两次1×1和3×3卷积操作,得到新的大尺度YOLO层;以三组新的不同尺度的YOLO层进行安全帽佩戴检测。
Darknet-53网络具体为:将基于标准卷积的Darknet-53的特征提取网络替换为基于深度可分离卷积的Darknet-53网络;同时保留原基础网络中的残差网络结构,得到最终的Darknet-53网络。
此外,本发明还采用k-means维度聚类算法对训练集进行聚类分析,得到多组新的目标先验框;将多组新的目标先验框按从小达到的顺序均分到不同尺度的特征图,并将较小的先验框分配给尺度更大的特征图。
例如,在公用数据机上初始训练计算得到的9组先验框维度分别为(10,13),(16,30),(33,23),(30,61),(62,45),(59,119),(116,90),(156,198),(373,326),这些先验框在安全帽佩戴检测中没有针对性,不能达到理想的检测精度。根据自建安全帽佩戴检测数据集,待检测的目标框尺寸一般在n×n(n≦150像素)范围内,因此,原有的YOLO v3先验框很难准确检测到安全帽的目标框信息。
因此,本发明针对自建的安全帽佩戴检测数据集“FacImage”,使用k-means维度聚类算法其进行聚类分析,得到9组新的目标先验框,分别为:(59,22),(68,30),(75,35),(88,37),(90,40),(99,45),(108,54),(119,57),(128,63),将这9组新的目标先验框按从小达到的顺序均分到三个不同尺度的特征图上,注意将较小的先验框分配给尺度更大的特征图。最后,我们用新的目标先验框进行安全帽佩戴检测实验。
为了使Light_YOLO_v3模型具有更好的鲁棒性,在自建安全帽检测数据集中,加入了不同尺寸的图片数据,这些图片使整个网络进行了多迟钝训练,所以将全连接层替换为卷积层的过程。
例如,当416×416的图片输入后,经过Darknet-53网络得到13×13×1024的特征图,该特征图经过全连接后,会得到一组规格为4096×1的特征图;同样的13×13×1024的特征图,不经过全连接,而是使用4096个13×13的卷积核进行全卷积操作,最后得到一组1×1×4096的特征图。在实际检测中,当输入图像尺寸为416×416时,全连接和全卷积操作都能得到结果,且结果基本一致。而当输入图片的尺寸与416×416不同时,比如输入图片大小为608×608,经过Darknet-53网络得到19×19×1024的特征图,接下来该特征图会与4096个神经元进行全连接,但由于构架尺寸的不同,全连接层将无法运行。对于全卷积操作替换掉全连接层后的网络,该网络会得到一个7×7×4096的输出,网络能够正常运行。所以,使用全卷积层替换掉全连接层后,对于不同尺寸的输入图像进行训练,可以提高网络对输入图像尺寸的鲁棒性。
除此之外,使用全卷积层替换掉全连接层可以提高网络检测性能。当使用全连接层进行目标预测时,整个前向传播算法可能需要7亿个参数,当目标检测网络使用全卷积操作进行分类时,只需要七十万个参数,可见全卷积层替换掉全连接层可以减少参数数量,减少运算量,提升检测性能。
改进YOLO v3模型的训练过程具体为:根据训练集对原始YOLO v3模型进行训练,先进行前向传播训练,再反向传播修正权值,得到与检测目标接近的新权重;对改进YOLOv3模型进行网络结构调整后将新权重复制到改进模型中,得到最终的改进YOLO v3模型。
改进YOLO v3模型称为改进Light_YOLO_v3,也可以称为改进MobileNet_YOLO_v3。
实施例2:一种安全帽佩戴智能检测系统,包括数据处理模块和模型检测模块。数据处理模块,用于对采集图像数据预处理后建立数据集,数据集划分为训练集和测试集。模型检测模块,用于依据训练集对改进YOLO v3模型训练后得到训练后的改进YOLO v3模型,并通过改进YOLO v3模型检测目标图像中的工作人员是否佩戴安全帽。
模型检测模块包括多尺度处理单元和特征融合单元。多尺度处理单元,用于通过Darknet-53网络的图像特征金字塔对输入图像进行多尺度处理,得到多尺度特征图。特征融合单元,用于将原始YOLO v3模型中的全连接操作替换为全卷积操作,并以上采样方法将低层特征和高层特征融合,得到新特征图进行安全帽佩戴检测。
实验验证:
基于深度学习的目标检测评价指标主要有四类,分别是平均准确率AP(AveragePrecision)、类别均衡准确率mAP(mean Average Precision)以及准确率P(Precision)和召回率R(Recall)。
1、将安全帽佩戴测试数据集按照目标尺寸大小进行排序,将其目标尺寸中0~10%,10%~30%,30%~70%,70%~90%,90~100%的目标分为XS、S、M、L、XL 5个子类,分别代表了不同大小尺寸的目标,通过对这些数据的测试,验证其对小目标检测性能的检测效果。图1中列出了原始YOLO v3模型和改进YOLO v3模型对不同尺寸大小目标的检测性能效果
由图1可知,改进YOLO v3模型比原始YOLO v3模型对不同目标尺寸的检测准确率更高,因此,使用基于图像金字塔的多层特征融合的策略,在对安全帽这种小目标的检测取得了不错的效果。
2、本发明中基于改进YOLO v3模型的多尺度训练方法,模型对于不同分辨率图像检测的鲁棒性明显提高。在测试过程中,将自制安全帽佩戴测试集图片均分为3种不同分辨率大小,即{320,608,1024},用于验证模型对不同分辨率输入图像的检测效果,3种不同分别率大小的图像分别代表了低分辨率、中分辨率、高分辨率三类,图2列出了两种算法针对不同分辨率图片的检测准确率的性能效果。
如图2所示,和原始YOLO v3模型比较,改进YOLO v3模型对不同分辨率的输入图像进行检测,结果检测准确率明显有提高,由此可见,针对安全帽佩戴检测的多尺度训练达到了提高算法鲁棒性的效果。
3、对比SSD、Faster R-CNN和YOLO v3等算法,安全帽佩戴检测效果评价结果(mAP和每秒识别帧数)如表1所示。
表1实验结果对比表
从实验结果来看,Faster R-CNN的mAP值为94.72%,是四种模型中最高的,而改进YOLO v3模型的mAP值为87.12%,虽然它的检测准确率不及Faster R-CNN,但在检测速度方面,比Faster R-CNN明显快的多。因此,改进YOLO v3模型对于实时检测任务会表现更好。改进YOLO v3模型在mAP和识别帧率两项指标上比SSD算法和原始YOLO v3模型都要提高,可见,改进YOLO v3模型在检测准确率和检测速度上都达到了优异的水平,该算法能够满足工业企业智能监控系统实时性检测要求。
经过四类算法进行对比分析,Faster R-CNN、SSD、Light_YOLO_v3以及改进Light_YOLO_v3算法均能够将安全帽佩戴情况准确识别出来,但SSD有时会受环境和图片背景的影响出现误检。而在多目标、小目标并存的情况下,SSD算法会出现漏检的情况,Faster R-CNN、原始YOLO v3模型和改进YOLO v3模型对小目标和多目标检测表现较好。综合来说,Faster R-CNN和改进YOLO v3模型在目标检测准确率上均优于其他两种算法,但Faster R-CNN目标检测算法是two stage模型,对于候选区域的确定需要进行大量计算,在检测速度上无法与改进YOLO v3模型相比。因此,针对工业企业作业人员安全帽佩戴检测任务,本发明提出的改进YOLO v3模型能够有效的均衡检测速率和检测准确率,能够满足作业现场实际检测的需求。
本领域内的技术人员应明白,本申请的实施例可提供为方法、系统、或计算机程序产品。因此,本申请可采用完全硬件实施例、完全软件实施例、或结合软件和硬件方面的实施例的形式。而且,本申请可采用在一个或多个其中包含有计算机可用程序代码的计算机可用存储介质(包括但不限于磁盘存储器、CD-ROM、光学存储器等)上实施的计算机程序产品的形式。
本申请是参照根据本申请实施例的方法、设备(系统)、和计算机程序产品的流程图和/或方框图来描述的。应理解可由计算机程序指令实现流程图和/或方框图中的每一流程和/或方框、以及流程图和/或方框图中的流程和/或方框的结合。可提供这些计算机程序指令到通用计算机、专用计算机、嵌入式处理机或其他可编程数据处理设备的处理器以产生一个机器,使得通过计算机或其他可编程数据处理设备的处理器执行的指令产生用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的装置。
这些计算机程序指令也可存储在能引导计算机或其他可编程数据处理设备以特定方式工作的计算机可读存储器中,使得存储在该计算机可读存储器中的指令产生包括指令装置的制造品,该指令装置实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能。
这些计算机程序指令也可装载到计算机或其他可编程数据处理设备上,使得在计算机或其他可编程设备上执行一系列操作步骤以产生计算机实现的处理,从而在计算机或其他可编程设备上执行的指令提供用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的步骤。
以上的具体实施方式,对本发明的目的、技术方案和有益效果进行了进一步详细说明,所应理解的是,以上仅为本发明的具体实施方式而已,并不用于限定本发明的保护范围,凡在本发明的精神和原则之内,所做的任何修改、等同替换、改进等,均应包含在本发明的保护范围之内。
Claims (10)
1.一种安全帽佩戴智能检测方法,其特征是,包括以下步骤:
对采集图像数据预处理后建立数据集,数据集划分为训练集和测试集;
依据训练集对改进YOLOv3模型训练后得到训练后的改进YOLOv3模型,并通过改进YOLOv3模型检测目标图像中的工作人员是否佩戴安全帽;
改进YOLOv3模型进行检测的过程具体为:
通过Darknet-53网络的图像特征金字塔对输入图像进行多尺度处理,得到多尺度特征图;
将原始YOLOv3模型中的全连接操作替换为全卷积操作,并以上采样方法将低层特征和高层特征融合,得到新特征图进行安全帽佩戴检测。
2.根据权利要求1所述的一种安全帽佩戴智能检测方法,其特征是,所述改进YOLOv3模型的网络结构具体为:
通过Darknet-53的图像特征金字塔,对第53层卷积层进行1×1及3×3的卷积操作,并重复进行一次1×1及3×3的卷积操作,得到新的小尺度YOLO层;
将新的小尺度YOLO层采取上采样操作,并与Darknet-53的第45个卷积层进行卷积和,使用连续两次的1×1和3×3卷积操作,得到新的中尺度YOLO层;
将新的中尺度YOLO层与Darknet-53的第29个卷积层进行卷积和,连续使用两次1×1和3×3卷积操作,得到新的大尺度YOLO层;
以三组新的不同尺度的YOLO层进行安全帽佩戴检测。
3.根据权利要求1所述的一种安全帽佩戴智能检测方法,其特征是,所述改进YOLOv3模型的基础学习率为0.001,激活函数选用ReLU,Batch-size为32,每迭代100次计算一次损失值。
4.根据权利要求1所述的一种安全帽佩戴智能检测方法,其特征是,所述Darknet-53网络具体为:
将基于标准卷积的Darknet-53的特征提取网络替换为基于深度可分离卷积的Darknet-53网络;
同时保留原基础网络中的残差网络结构,得到最终的Darknet-53网络。
5.根据权利要求1所述的一种安全帽佩戴智能检测方法,其特征是,该方法还包括:
采用k-means维度聚类算法对训练集进行聚类分析,得到多组新的目标先验框;
将多组新的目标先验框按从小达到的顺序均分到不同尺度的特征图,并将较小的先验框分配给尺度更大的特征图。
6.根据权利要求1所述的一种安全帽佩戴智能检测方法,其特征是,所述改进YOLOv3模型的训练过程具体为:
根据训练集对原始YOLOv3模型进行训练,先进行前向传播训练,再反向传播修正权值,得到与检测目标接近的新权重;
对改进YOLOv3模型进行网络结构调整后将新权重复制到改进模型中,得到最终的改进YOLOv3模型。
7.根据权利要求1所述的一种安全帽佩戴智能检测方法,其特征是,所述采集图像数据的预处理具体为:
采用开发工具将工业企业的视频监控转换为图片格式,并采用FasterR-CNN算法筛选出图片数据中包含行人的图片作为数据集图片,放入待标记数据集文件夹中;
采用labelImg工具对待标记数据集中图像的目标位置以矩形框进行标注,得到预处理后的数据集。
8.一种安全帽佩戴智能检测系统,其特征是,包括:
数据处理模块,用于对采集图像数据预处理后建立数据集,数据集划分为训练集和测试集;
模型检测模块,用于依据训练集对改进YOLOv3模型训练后得到训练后的改进YOLOv3模型,并通过改进YOLOv3模型检测目标图像中的工作人员是否佩戴安全帽;
模型检测模块包括多尺度处理单元和特征融合单元;
多尺度处理单元,用于通过Darknet-53网络的图像特征金字塔对输入图像进行多尺度处理,得到多尺度特征图;
特征融合单元,用于将原始YOLOv3模型中的全连接操作替换为全卷积操作,并以上采样方法将低层特征和高层特征融合,得到新特征图进行安全帽佩戴检测。
9.一种计算机终端,包含存储器、处理器及存储在存储器并可在处理器上运行的计算机程序,其特征是,所述处理器执行所述程序时实现如权利要求1-7中任意一项所述的一种安全帽佩戴智能检测方法。
10.一种计算机可读介质,其上存储有计算机程序,其特征是,所述计算机程序被处理器执行可实现如权利要求1-7中任意一项所述的一种安全帽佩戴智能检测方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110943360.5A CN113688709B (zh) | 2021-08-17 | 2021-08-17 | 一种安全帽佩戴智能检测方法、系统、终端及介质 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110943360.5A CN113688709B (zh) | 2021-08-17 | 2021-08-17 | 一种安全帽佩戴智能检测方法、系统、终端及介质 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN113688709A true CN113688709A (zh) | 2021-11-23 |
CN113688709B CN113688709B (zh) | 2023-12-05 |
Family
ID=78580294
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110943360.5A Active CN113688709B (zh) | 2021-08-17 | 2021-08-17 | 一种安全帽佩戴智能检测方法、系统、终端及介质 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113688709B (zh) |
Cited By (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113553922A (zh) * | 2021-07-05 | 2021-10-26 | 安徽中医药大学 | 一种基于改进型卷积神经网络口罩佩戴状态检测方法 |
CN114170421A (zh) * | 2022-02-10 | 2022-03-11 | 青岛海尔工业智能研究院有限公司 | 图像检测方法、装置、设备及存储介质 |
CN114623727A (zh) * | 2022-03-14 | 2022-06-14 | 北京理工大学 | 一种激光成像近程探测目标识别方法 |
CN114937232A (zh) * | 2022-07-25 | 2022-08-23 | 浙江大学 | 医废处理人员防护用具穿戴检测方法、系统和设备 |
Citations (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108460403A (zh) * | 2018-01-23 | 2018-08-28 | 上海交通大学 | 一种图像中多尺度特征融合的目标检测方法与系统 |
CN109389057A (zh) * | 2018-09-22 | 2019-02-26 | 天津大学 | 一种基于多尺度高级语义融合网络的物体检测方法 |
CN109685008A (zh) * | 2018-12-25 | 2019-04-26 | 云南大学 | 一种实时的视频目标检测方法 |
CN111881730A (zh) * | 2020-06-16 | 2020-11-03 | 北京华电天仁电力控制技术有限公司 | 一种火电厂现场安全帽佩戴检测方法 |
CN112926405A (zh) * | 2021-02-01 | 2021-06-08 | 西安建筑科技大学 | 一种安全帽佩戴检测方法、系统、设备及存储介质 |
CN113033289A (zh) * | 2021-01-29 | 2021-06-25 | 南瑞集团有限公司 | 一种基于dssd算法的安全帽佩戴检查方法、装置与系统 |
-
2021
- 2021-08-17 CN CN202110943360.5A patent/CN113688709B/zh active Active
Patent Citations (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108460403A (zh) * | 2018-01-23 | 2018-08-28 | 上海交通大学 | 一种图像中多尺度特征融合的目标检测方法与系统 |
CN109389057A (zh) * | 2018-09-22 | 2019-02-26 | 天津大学 | 一种基于多尺度高级语义融合网络的物体检测方法 |
CN109685008A (zh) * | 2018-12-25 | 2019-04-26 | 云南大学 | 一种实时的视频目标检测方法 |
CN111881730A (zh) * | 2020-06-16 | 2020-11-03 | 北京华电天仁电力控制技术有限公司 | 一种火电厂现场安全帽佩戴检测方法 |
CN113033289A (zh) * | 2021-01-29 | 2021-06-25 | 南瑞集团有限公司 | 一种基于dssd算法的安全帽佩戴检查方法、装置与系统 |
CN112926405A (zh) * | 2021-02-01 | 2021-06-08 | 西安建筑科技大学 | 一种安全帽佩戴检测方法、系统、设备及存储介质 |
Cited By (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113553922A (zh) * | 2021-07-05 | 2021-10-26 | 安徽中医药大学 | 一种基于改进型卷积神经网络口罩佩戴状态检测方法 |
CN114170421A (zh) * | 2022-02-10 | 2022-03-11 | 青岛海尔工业智能研究院有限公司 | 图像检测方法、装置、设备及存储介质 |
CN114170421B (zh) * | 2022-02-10 | 2022-06-17 | 卡奥斯工业智能研究院(青岛)有限公司 | 图像检测方法、装置、设备及存储介质 |
CN114623727A (zh) * | 2022-03-14 | 2022-06-14 | 北京理工大学 | 一种激光成像近程探测目标识别方法 |
CN114937232A (zh) * | 2022-07-25 | 2022-08-23 | 浙江大学 | 医废处理人员防护用具穿戴检测方法、系统和设备 |
Also Published As
Publication number | Publication date |
---|---|
CN113688709B (zh) | 2023-12-05 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN113688709B (zh) | 一种安全帽佩戴智能检测方法、系统、终端及介质 | |
CN109636848B (zh) | 一种基于无人机的油气管道巡检方法 | |
CN105809146B (zh) | 一种图像场景识别方法和装置 | |
CN114495029B (zh) | 一种基于改进YOLOv4的交通目标检测方法及系统 | |
CN102682301B (zh) | 使用附加分类器来调节畅通路径检测 | |
CN109858547A (zh) | 一种基于bssd的目标检测方法与装置 | |
CN112381005A (zh) | 一种面向复杂场景的安全帽检测系统 | |
CN115294528A (zh) | 一种行人安全监控方法及装置 | |
CN115512333A (zh) | 一种sar图像的车辆目标检测方法和系统 | |
CN116824135A (zh) | 基于机器视觉的大气自然环境试验工业产品识别分割方法 | |
CN107358246A (zh) | 一种使用卷积神经网络对物体进行细分类的方法 | |
CN110866931A (zh) | 图像分割模型训练方法及基于分类的强化图像分割方法 | |
CN112784675B (zh) | 目标检测方法及装置、存储介质、终端 | |
CN112861678A (zh) | 一种图像识别方法及装置 | |
Tran et al. | A fully automated vision-based system for real-time personal protective detection and monitoring | |
CN112257752A (zh) | 快递物品的识别分类方法、装置、设备和存储介质 | |
Li et al. | An efficient method for DPM code localization based on depthwise separable convolution | |
CN112016542A (zh) | 城市积涝智能检测方法及系统 | |
CN115311885B (zh) | 一种评测方法、系统、电子设备及存储介质 | |
CN103473562B (zh) | 特定人体动作的自动训练和识别系统 | |
CN115953744A (zh) | 一种基于深度学习的车辆识别追踪方法 | |
Bellam et al. | A Practical Approach of Recognizing and Detecting Traffic Signs using Deep Neural Network Model | |
CN111626193A (zh) | 一种面部识别方法、面部识别装置及可读存储介质 | |
Jia et al. | Coarse annotation refinement for segmentation of dot-matrix batchcodes | |
CN114266887B (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 | ||
TA01 | Transfer of patent application right |
Effective date of registration: 20231108 Address after: 524088 No. 1 Haida Road, Mazhang District, Zhanjiang City, Guangdong Province Applicant after: Guangdong Ocean University Address before: 434023 No. 1 South Ring Road, Hubei, Jingzhou Applicant before: Yangtze University |
|
TA01 | Transfer of patent application right | ||
GR01 | Patent grant | ||
GR01 | Patent grant |