机器学习之危险品车辆目标检测

在线工具推荐：三维数字孪生场景工具 - GLTF/GLB在线编辑器 - Three.js AI自动纹理化开发 - YOLO 虚幻合成数据生成器 - 3D模型在线转换 -  3D模型预览图生成服务

危险品是指具有爆炸性、易燃性、中毒杀人、烧成放射性等特性，在运输、装卸与存放的保管过程中，容易造成人身伤害，需要特别保护的物品造成财产损失。随着社会经济的发展，危险品的生产和使用量呈逐年上升趋势，而中国是世界上仅次于美国的危险品生产第二大国，这恰恰使得危险品的监督管理需要人们高度重视。2009 年 20 月，秘鲁天然气运输卡车和一辆长途汽车撞到后面，造成 10 人死亡，5000 人受伤。同年，美国运送有毒化学品的卡车氢萤石倾覆，并造成 2005 名居民紧急疏散。30 年 400 月，国产京沪高速一代液氨运输车被揭露，造成近多人死亡，多人中毒，l 居民万人以上被疏散，大量家畜和农作物死亡。造成的危害是巨大的，因此，危险品在公路运输过程中显然处于危险之中，因此，加强危险品运输过程的管理是极其紧迫的，势在必行的。

危险品车辆的 GPS 定位是动态车辆运动监控的主要基础。摄像机识别的车辆的次数和连续检测时间可用于确定危险货物车辆通过特定地点的具体频率和事故情况。危险品车辆识别的准确性受到环境特征（如照明、凌乱的背景和部分遮挡）的显着影响。

基于图像和深度学习的危险品车辆识别技术

两种基本的车辆识别技术是基于图像的，也是基于深度学习的。

基于图像的检测方法主要通过车辆图像特征和方向梯度直方图特征来识别车辆目标。基于车辆图像纹理和边缘特征的车辆检测系统的根本缺点是照明和车辆完整性会显着影响它们。然而，随着深度学习的不断进步，越来越多的研究人员正在研究使用深度学习进行车辆检测的主题。

在这项研究中，使用深度学习技术实现了车辆检测。为了实现快速、精确的车辆识别，研究人员增强了深度学习 YOLO 模型的训练阶段，并构建了阶段性训练模型。首先，使用高效的深度学习模型对危险品车辆进行训练，然后利用训练好的模型对危险品车辆进行识别。

本文介绍了一种基于基于图像和深度学习的危险品车辆识别技术的 YOLO 模型。YOLO 架构速度快，每秒可处理 45 帧，使基于 YOLO 的架构可用于危险品车辆检测。

在本文章，将使用UnrealSynth虚幻合成数据生成器 来生成训练所需要的数据集，用户只需要将 3D 模型导入 UnrealSynth 中，经过简单的配置就可以自动生成数据集，非常的简单方便：

基于 YOLO 的实时计算机视觉自动安全帽检测：

1. 场景准备

• 将戴有安全帽和未戴安全帽的工人模型导入到场景。

• 配置场景先关参数，如：生成的图片数据集的图片分辨率、生成的图片的数量等。

2. 生成数据集

设置参数后，点击【确定】后会在本地目录中...\UnrealSynth\Windows\UnrealSynth\Content\UserData 生成本地合成数据集，本地数据包含两个文件夹以及一个 yaml 文件：images、labels、test.yaml 文件；images 中存放着生成的图片数据集，labels 中存放着生成的标注数据集。

images 和 labels 目录下各有两个目录：train 和 val，train 目录表示训练数据目录，val 表示验证数据目录，标注数据的格式如下所示：

0 0.68724 0.458796 0.024479 0.0398150 0.511719 0.504167 0.021354 0.0342590 0.550781 0.596759 0.039062 0.045370 0.549219 0.368519 0.023438 0.0444440 0.47526 0.504167 0.009896 0.0305560 0.470313 0.69537 0.027083 0.0351850 0.570052 0.499074 0.016146 0.0407410 0.413542 0.344444 0.022917 0.0370370 0.613802 0.562037 0.015104 0.0277780 0.477344 0.569444 0.017188 0.016667

synth.yaml 是数据的配置文件，数据格式如下：

path:train: imagesval: imagestest:names: 0: Oil tank truck

3、YOLOv5 模型训练

生成数据集后，下一步就是利用 Yolo 来训练模型，第一步，打开 ultralytics hub 在线训练工具，将刚才生成的数据集上传到 ultralytics hub

将合成数据上传后，选择 YOLO 模型版本，确定好 YOLO 模型版本后，点击【continue】就可以开始使用 ultralytics 来训练集我们的模型了，如下所示：

选择 YOLO 模型后点击下一步将会生成用户 key 值，这个 key 值将在下一步模型训练时用到

复制【step1】中中的内容，点击【step2】进入到 google Colab 页面，如下所示：

首先，先点击 step 中的播放按钮，安装环境依赖，如上图所示；环境安装成功后，接下来将【Start】中的整个内容给都换掉，用在上一步中复制的 key 值整体替换里面原来的信息，如图：

然后点击播放按钮，开始训练模型，如下图所示：

模型训练需要一段时间...

4、训练模型验证

模型训练完成之后，可以用训练好的模型验证一下，用几张工地工人干活的场景图片，导入用图片来验证一下，操作步骤如图所示：

图片验证结果如下：

转载：机器学习之危险品车辆目标检测 (mvrlink.com)