基于深度学习的安全帽识别检测系统(python OpenCV yolov5)

收藏和点赞,您的关注是我创作的动力

文章目录

    • 概要
  • 一、研究的内容与方法
  • 二、基于深度学习的安全帽识别算法
    • 2.1 深度学习
    • 2.2 算法流程
    • 2.3 目标检测算法
    • 2.3.1 Faster R-CNN
    • 2.3.2 SSD
    • 2.3.3 YOLO v3
  • 三 实验与结果分析
    • 3.1 实验数据集
    • 3.1.1 实验数据集的构建
    • 3.1.2 数据集的分类
    • 3.1.3 增强数据集
  • 四 原型系统实现
    • 4.1 生成系统的Web页面
    • 4.2 上传检测图片
  • 五 结 论
    • 目录

概要

  
  基于深度学习算法,以PaddlePaddle深度学习框架作为实验环境,选取了开源的安全帽识别数据库和实地拍摄的安全帽佩戴照片,使用样本扩增增加了实验数据集的样本数,选取了Faster R-CNN、SSD与YOLO v3三种深度神经网络模型,构建出安全帽智能识别模型。
  在实验数据集上对三种模型分别实验,对比实验结果。结果显示,基于YOLOv3的模型具有识别精度高,识别速率快等特点,识别准确率达到了99.97%。为了验证了本文提出方法的有效性,使用Python语言开发了安全帽佩戴识别的原型系统。
  
关键词:深度学习;安全帽识别;Python;YOLO v3

一、研究的内容与方法

  本课题对基于深度学习的安全帽佩戴识别方法进行了研究,以PaddleHub深度学习框架作为实验环境,选取了开源的安全帽识别数据库和实地拍摄的安全帽佩戴照片,使用样本扩增增加了实验数据集的样本数,选取了Faster RCNN、SSD与YOLOv3三种深度神经网络模型,构建出安全帽智能识别模型。在实验数据集上对三种模型进行了对比实验。分别计算了准确率、速率与稳定性,对实验结果进行了对比分析。为了验证了本文提出方法的有效性,使用Python语言开发了安全帽佩戴识别的原型系统,用于模拟建筑工地上识别安全帽的过程。

二、基于深度学习的安全帽识别算法

2.1 深度学习

深度学习,是实现人工智能的必经之路——机械学习中的一种,是人工神经网络研究不断深入后的产物之一,它的结构特点是包含多个隐藏层感知器。深度学习中更为抽象的高层次表示特征是由多个低层次的特征组合而成,这使深度学习模型更容易发现数据特征的表示结构。由于深度学习模型的构造理念是通过模拟人脑来建立高效分析与学习的神经网络,因此深度学习具有通过模拟人脑的思考方式来解释图像、声音等数据的能力。深度学习是一种常用的模式分析方法的统称,根据研究的具体内容,主要使用三种方法:
  一是卷积神经网络,也就是基于卷积运算的神经网络系统;二是基于多层神经元的自编码神经网络,主要包括自编码与稀疏编码两类;三是深度执行网络,主要通过多层自编码神经网络进行预训练,继而与鉴别信息结合对神经网络权值进行优化。
  深度学习在通过多层处理后,能够将初始“低层”的特征表示逐渐向“高层”的特征表示转化,以达到用相对简单、基础的模型完成更为困难、复杂的学习任务。从这一角度,也可以把深度学习理解为特征学习或是表示学习
  在以往的机械学习实现实际任务时,通常依赖专家对描述样本的特征进行设计,这使得专家对样本特征设计的好坏,直接影响了实际系统的性能。而特征学习则通过机械学习技术来弥补这一缺点,这使得深度学习拜托了对专家的依赖。

2.2 算法流程

本课题的算法流程图如图1所示
在这里插入图片描述
图1 算法流程图
  首先,将开源的安全帽数据集与工地实地拍摄获取的安全帽原始图片按比例分为训练集与测试集,并对它们分别进行数据增强。然后再用数据增强后训练集对深度模型进行训练。训练完成后,用数据增强后的测试集安全帽图片对已训练的模型进行测试,最后输出识别结果。

2.3 目标检测算法

本文选用了Faster R-CNN、SSD与YOLO v3三种深度模型。Faster R-CNN模型在ILSVRV与COCO等竞赛中均获得多项第一,SSD是在Faster RCNN与YOLO v1 之后的高级的物体检测模型。与 Faster RCNN与YOLO v1相比,SSD在识别速度和性能上都得到了明显的提升。下面对三种模型分别进行介绍。

2.3.1 Faster R-CNN

Faster R-CNN在目标检测领域表现出极强的生命力,虽然Faster R-CNN是2015年建立的算法,但它现在仍然是许多的目标监测系统的基础,并且在解决拥挤或是小物体等问题上仍然具有较大的有优势,这在更新迅速的深度学习领域是十分少见的,不仅如此,Faster R-CNN的利用范围也十分广阔,例如:水果种类检测,宠物种类检测以及本文的安全帽检测等等。

2.3.2 SSD

SSD的全称为Single Shot MultiBox Detector是WeiLiu在ECCV上提出的深度模型,是一种目标检测模型。虽然SSD是在2016年提出的模型,但现在仍然是主要的目标检测算法之一,兼顾了Faster R-CNN与YOLOv1的优缺点,在速度上优于Faster R-CNN,在精度上优于YOLOv1;并且在处理小目标时对比YOLO有着结构上的优势。

2.3.3 YOLO v3

YOLO v3是在YOLOv2与YOLOv1的基础上建立的一种目标检测算法采用了残差跳连接和上采样等新架构值,YOLO v3最为突出的特点是它能够检测三种不同的尺度。YOLO是一种全卷积网络,它是通过在特征映射上使用1x1的核来产生最终的输出,而YOLOv3在此基础上完成了优化和改进,它能够在网络的三个位于不同位置的大小不同的特征图使用1x1核来进行检测。

三 实验与结果分析

3.1 实验数据集

3.1.1 实验数据集的构建

    实验的数据集包括两部分,一部分来源于一个开源的安全帽识别数据集(https://github.com/njvisionpower/Safety-Helmet-Wearing-Dataset),该数据集已经完成了图片样本的标注。另外一部分是本人在工地现场拍摄的照片,也进行了标注。
  佩戴安全帽两个集合,构成有标签的数据样本。随后再将数据集随机划分成训练集与测试集两个部分。两个部分的比例为训练集:测试集=8:2。得到的训练集和测试集的样本数量分布如图2所示。

3.1.2 数据集的分类

   首先将数据集中本人拍摄的安全帽图片样本进行了人工分类,与开源数据集样本一起,划分为佩戴安全帽与未佩戴安全帽两个集合,构成有标签的数据样本。随后再将数据集随机划分成训练集与测试集两个部分。两个部分的比例为训练集:测试集=8:2。得到的训练集和测试集的样本数量分布如图2所示。

在这里插入图片描述

3.1.3 增强数据集

   为了得到更多的训练图片,实验使用Tensorflow开发库对数据进行了增强,目的是达到扩充原始数据集的效果。Tensorflow开发库是深度学习库,它是开源的,支持所有流行语言,如Python、C++、Java、R和Go,并且它受到所有云服务(AWS、Google和Azure)的支持。Tensorflow内部包含有大量的工具,可以对模型进行设计和调试,同时也提供了基本的数据增强工具。
  本课题使用了Tensorflow库对数据集进行了旋转、变形等随机操作,得到增强后的数据集。增强后的数据集包括:佩戴安全帽图片样本总数达到3143张,未佩戴安全帽的图片样本总数达到2271张。在佩戴安全帽的图片样本中,将2514张作为训练集,将629张作为测试集;未佩戴安全帽的图片中将1816张作为训练集,将455张作为测试集。训练集和测试集的样本统计结果如表1所示。图3和图4分别为佩戴安全帽图片样本和未佩戴安全帽图片的样本示例。
表1 训练集和测试集的样本统计

在这里插入图片描述在这里插入图片描述

四 原型系统实现

   因为在实验过程中发现,YOLOv3是三个对比实验中最为突出的一个,在损失函数曲线图中的曲线收敛速度相比另外两种算法更快,而且网络波动的幅度也较小,收敛的值比较小,且准确率在三者中也是最高的一个,得到了YOLOv3在安全帽智能检测领域的适用性较高的结果。因此,本课题基于训练好的YOLOv3模型使用Python语言开发了安全帽佩戴识别的原型系统,用于模拟建筑工地上识别安全帽的过程。该系统实现为一个Web页面。

4.1 生成系统的Web页面

    通过Python程序编写生成了该系统的Web页面。在Python程序运行结束时会给出生成的页面的本地地址和端口号,一般为http://localhost:8000。在浏览器上输入该地址和端口号就可以打开系统的Web页面,进行安全帽的识别操作。下面是该Python程序的源代码。

在这里插入图片描述图8是Python程序的运行结果。输出了系统页面的地址。
在这里插入图片描述

图8 运行结果
打开地址http://localhost:8000就可以实际进行安全帽图像的智能识别操作。

4.2 上传检测图片

识别系统需要上传检测图片,相关代码如下。
在这里插入图片描述

用户只需要点击选择文件就可以打开浏览窗口对图片进行选择,如图10所示。
在这里插入图片描述

图10 选择图片
  在选择了图片之后,运行界面就会改变,便于用户对选择的图片进行确认,以及进行下一步的操作,如图11所示。
在这里插入图片描述
在这里插入图片描述

图11图片确认

五 结 论

  本课题针对企业作业和工地施工过程佩戴安全帽的自动识别问题,基于深度学习,提出了一种安全帽佩戴识别方法。该方法基于深度学习算法,以PaddlePaddle深度学习框架作为实验环境,选取了开源的安全帽识别数据库和实地拍摄的安全帽佩戴照片,使用样本扩增增加了实验数据集的样本数,选取了Faster R-CNN、SSD与YOLO v3三种深度神经网络模型,构建出安全帽智能识别模型。在实验数据集上对三种模型分别实验,对比实验结果。结果显示,基于YOLOv3的模型具有识别精度高,识别速率快等特点,识别准确率达到了99.97%。为了验证了本文提出方法的有效性,使用Python语言开发了安全帽佩戴识别的原型系统。
  在本课题的研究与实现过程中,遇到了许多困难在设计这个系统之前,例如,算法调用失败的经历就多次遇到,图像显示、参数传递出现的问题更是数不胜数,在浏览各大网站之后慢慢摸索,逐渐改善了研究,解决了问题,经过了这一系列的努力,不仅懂得了学习一项技术最重要的是能够脚踏实地,勇敢面对失败才能更加靠近成功,而且提高了自身的专业能力,赋予自己面对将来挑战的信心。由于能力和时间有限,系统仍有着诸多不足,在将来的学习工作中将不断完善。

目录

目 录
内容摘要I
AbstractII
1 绪论1
1.1 目的及意义1
1.2 国内外的发展现状1
1.3 研究的内容与方法2
2 基于深度学习的安全帽识别算法3
2.1 深度学习3
2.2 算法流程4
2.3 目标检测算法4
3 实验结果分析6
3.1 数据集6
3.2 实验环境搭建8
3.3 实验设置9
3.4 实验结果分析10
4 原型系统实现13
4.1 模块调用以及运行网址呼出13
4.2 检测图片上传14
4.3 检测结果显示15
5 结论与研究成果17
参考文献18

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.rhkb.cn/news/176740.html

如若内容造成侵权/违法违规/事实不符,请联系长河编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

基于松鼠算法的无人机航迹规划-附代码

基于松鼠算法的无人机航迹规划 文章目录 基于松鼠算法的无人机航迹规划1.松鼠搜索算法2.无人机飞行环境建模3.无人机航迹规划建模4.实验结果4.1地图创建4.2 航迹规划 5.参考文献6.Matlab代码 摘要:本文主要介绍利用松鼠算法来优化无人机航迹规划。 1.松鼠搜索算法 …

安装pytorch报错torch.cuda.is_available()=false的解决方法

参考文章: https://blog.csdn.net/qq_46126258/article/details/112708781 https://blog.csdn.net/Andy_Luke/article/details/122503884 https://blog.csdn.net/anmin8888/article/details/127910084 https://blog.csdn.net/zcs2632008/article/details/127025294 …

Linux内核是如何创建进程?

目录 1.Linux如何创建进程 2.fork函数原理 2.1 fork函数原型 2.2 fork函数实现原理 2.3 父子进程虚拟地址空间(mm_struct)之间的关系 2.4 写时拷贝(copy-on-write)技术 2.5 父子进程如何共享文件(files_struct&…

IDEA优雅自动生成类注释和快捷键生成方法注释

生成类注释 Preferences->Editor->File and Code Templates-> Includes ->File Header 注释模板: /*** Classname ${NAME}* Description ${description}* Date ${DATE} ${TIME}* Created by ZouLiPing*/生成方法和字段注释 查看IDEA自动配置java快捷…

山西电力市场日前价格预测【2023-11-02】

日前价格预测 预测说明: 如上图所示,预测明日(2023-11-02)山西电力市场全天平均日前电价为151.67元/MWh。其中,最高日前电价为280.23元/MWh,预计出现在22:15。最低日前电价为0.00元/MWh,预计出…

杂记 | 使用numpy计算欧氏距离和马氏距离示例

文章目录 一、欧氏距离与马氏距离二、使用numpy计算欧氏距离三、使用numpy计算马氏距离 一、欧氏距离与马氏距离 这里引用GPT4的回答: 二、使用numpy计算欧氏距离 import numpy as npp1 [1, 2, 3] p2 [4, 5, 6] point1 np.array(p1) point2 np.array(p2)euc…

Leetcode刷题---多数元素(计数、哈希表、摩尔投票)

题目描述: 思路一: 可以使用暴力破解来解决此问题,但是暴力破解的时间复杂度很高,在leetcode测试的时候爆出来超出时间限制的问题。 代码: class Solution {public int majorityElement(int[] nums) {int nnums.leng…

低代码,程序员提高生产力的开发工具

目录 一、什么是低代码? 二、低代码的本质是什么? 三、低代码平台的搭建能力 四、写在最后 一、什么是低代码? 简单来说,低代码是一种用于应用程序开发的模块化方法,它能有效减少应用程序的开发时间。基于可重用的、组…

新手小白如何自学制作电子画册并分享各大平台

一提到电子画册制作,相信很多人都开始头大了,毕竟对于新手小白来说还是比较难的,那么,新手小白要如何自学制作电子画册并快速分享出去呢,其实只需要一个在线电子画册制作平台FLBOOK就可以了 。 小白该怎样制作电子画册…

数据结构 | 顺序表专题

数据结构 | 顺序表专题 文章目录 数据结构 | 顺序表专题课前准备1. 目标2. 需要的储备知识3. 数据结构相关概念 开始顺序表1、顺序表的概念及结构2、顺序表分类3、动态顺序表的实现初始化顺序表顺序表的销毁顺序表的尾插顺序表的头插检查容量顺序表的尾删打印顺序表顺序表的头删…

塞尔帕替尼/塞普替尼是什么,这篇文章用大白话帮你深入了解

(图片来源于网络) 塞尔帕替尼是一种靶向药物,用于治疗某些类型的癌症。它的全名是Selpercatinib,也被称为LOXO-292。这种药物主要针对一种叫做RET的基因,该基因在某些癌症中发生异常。 那么,为什么要针对…

微信小程序去掉Button自带边框

前言 微信button自带边框 去掉边框后效果 实现 html代码 <view><button class"contactBtn" open-type"contact" contact"handleContact"session-from"sessionFrom">意见反馈</button> </view>css代码 …

oracle 重启步骤及踩坑经验

oracle 重启步骤及踩坑经验 标准重启步骤 切换到oracle用户 su - oracle关闭监听 lsnrctl stop杀掉oracle有关进程 ps -ef|grep $ORACLE_SID|grep -v ora_|grep LOCALNO|awk {print $2}|xargs kill -9#查询pid ps -ef|grep $ORACLE_SID|grep -v ora_|grep LOCALNO|awk {p…

介绍几款Linux 下终极SSH客户端

安全外壳协议&#xff08;Secure Shell&#xff0c;简称 SSH&#xff09;是一种网络连接协议&#xff0c;允许您通过网络远程控制计算机。特别是在Linux命令行模式下&#xff0c;使用SSH&#xff0c;可以很方便管理linux上的运维工作。以下是一些最受欢迎的Linux SSH客户端&…

Jetpack:025-Jetpack中的多点触控事件

文章目录 1. 概念介绍2. 使用方法2.1 缩放事件2.2 旋转事件2.3 平移事件2.4 综合事件 3. 示例代码4. 内容总结 我们在上一章回中介绍了Jetpack中滚动事件相关的内容&#xff0c;本章回中主要介绍 多点解控事件。闲话休提&#xff0c;让我们一起Talk Android Jetpack吧&#xf…

音视频开发:音频编码原理+采集+编码实战

原理&#xff1a; 消除冗余信息&#xff0c;压缩量最大&#xff0c;也叫有损压缩 剔除人耳听觉范围外的音频信号20Hz以下和20000Hz以上&#xff1b;去除被掩蔽的音频信号&#xff0c;信号的遮蔽可以分为频域遮蔽和时域遮蔽&#xff1b;频域遮蔽效应 屏蔽70分贝以下&#xff0…

Qt 使用Quazip解压缩、压缩文件

1.环境搭建 Quazip&#xff0c;是在zlib基础上进行了简单封装的开源库&#xff0c;适用于多种平台&#xff0c;利用它可以很方便将单个或多个文件打包为zip文件&#xff0c;且打包后的zip文件可以通过其它工具打开。 下载Quazip QuaZIP download | SourceForge.net 解压后&…

利用ChatGPT进行股票走势分析

文章目录 1. 股票分析2. 技巧分析3. 分析技巧21. 股票分析 这张图片显示了一个股票交易软件的界面。以下是根据图片内容的一些解读: 股票代码: 图片右上角显示的代码是“600517”,这是股票的代码。 图形解读: 该图展示了股票的日K线图。其中,蜡烛图表示每日的开盘、收盘、最…

力扣第62题 不同路径 c++ 动态规划 dp二维 + dp一维 解法

题目 62. 不同路径 中等 相关标签 数学 动态规划 组合数学 一个机器人位于一个 m x n 网格的左上角 &#xff08;起始点在下图中标记为 “Start” &#xff09;。 机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角&#xff08;在下图中标记为 “Fini…

电商课堂|5分钟了解电商数据分析完整流程,建议收藏!

账户效果下降&#xff0c;如何能够快速找到问题并优化调整&#xff1f; 相信百分之90%的竞价员都会说&#xff1a;“做数据分析。” 没错&#xff0c;数据分析能够帮助我们快速锁定问题所在&#xff0c;确定优化方向&#xff0c;还可以帮助我们找到流量控制的方向。那么做电商&…