正文

使用ymir生产基于yolov5的头盔检测模型

fzp8656342  fzp8656342  2022-12-07  349

关键词：

使用YMIR生产基于yolov5的头盔检测模型

YOLOv5(物体检测/Pytorch训练)

最近发现了个好玩的东西，直接使用无代码进行检测模型开发，在上面试了一下头盔检测的效果，效果不错，下面附一下教程，这个平台不仅可以用来训练模型，还能将训练好的模型直接发布到端设备上，废话不多说先上关键信息。

YMIR平台主页：https://github.com/IndustryEssentials/ymir
京东商城页面：https://i-item.jd.com/10065116628109.html

1.概述

YOLOv5共有s,m,l,x四个模型，相同的数据集场景下，一般YOLOv5-s模型训练和推理性能最佳，YOLOv5-x模型mAP指标最佳。针对提供带有物体框标注的数据集，该算法会预先载入在COCO数据集下训练过的模型，训练后生成的模型可直接在YMIR平台部署到相机。

2.YOLOV5结构解析

上图即Yolov5的网络结构图，可以看出，分为输入端、Backbone、Neck、Prediction四个部分。
（1）输入端：Mosaic数据增强、自适应锚框计算、自适应图片缩放
（2）Backbone：Focus结构，CSP结构
（3）Neck：FPN+PAN结构
（4）Prediction：GIOU_Loss

YOLOV5在coco数据集性能测试图

3.算法基本信息

适用场景：物体检测，大中小目标检测

适用算法：头盔检测，行人检测，车辆检测，人脸检测，其他检测相关
算法输入：
- 已选择的数据集，用户可自行分配训练、测试数据集
- YOLOv5的COCO预训练模型。
算法输出：
- 已训练完成的模型文件，可直接一键发布到算法平台
推理速度参考：

训练参数说明
预训练模型：可设置预训练模型选择上一个训练作业完成的模型
- 举例：对一个已经完成的训练作业（假设训练了50个epochs），想要
  训练更多的epochs（追加30个epochs）, 选择上一个已完成训练的模型如model_group_v3

动手实测

查看训练、测试数据集

在项目列表里选择头盔项目，导入数据完成后即可查看数据集
如下图所示该数据集共包含三个类别，分别为person（人体），head（人头），helmet（安全帽），可以粗略地浏览一下数据集中bbox的尺寸，可以看到perosn的bbox最大，head和helmet人头bbox大小相近，三个目标在图中的比例都相对较大，总体来讲head和helmet算中等大小目标，person是大目标，所以本次使用的超参img_size可使用默认超参640，如果图像中小目标较多，可以适当增加img_size，注意：img_size必须是32的倍数，
可供参考的选项如：
小目标：800，960
中目标：640，604
大目标：320，416

模型训练启动页面

模型运行中性能、进度曲线变化

启动训练任务后即可查看训练过程

模型发布、部署

REF：

https://github.com/IndustryEssentials/ymir
https://zhuanlan.zhihu.com/p/260400612

基于b/s架构搭建一个支持多路摄像头的实时处理系统----基于websocket使用yolov5系列模型

文章大纲多路摄像头使用yolov5增加视频检测能力调整yolov5的目录结构符合自己的项目yolov5调用本地模型文件进行视频图像的推理torchhubloadcustom方法yolov5的函数位置变化，letter_box,dataset去哪了？添加yolo检测能力获得当前图像的推... 查看详情

yolov5+gsconv：基于yolov5gsconv得小目标检测算法训练

目标检测是计算机视觉中一项艰巨的下游任务。对于车载边缘计算平台，大模型很难达到实时检测的要求。而且，由大量深度可分离卷积层构建的轻量级模型无法达到足够的准确性。因此本文引入了一种新方法GSConv来减轻模型的... 查看详情

python基于融合spd-conv改进yolov5与原生yolov5模型实践路面裂痕裂缝检测

...一定程度提升模型的检测效果，今天正好有时间就想基于SPD融合yolov5s模型来开发路面裂痕裂缝检测模型，同时与原生的yolov5s模型进行对比分析，首先看下效果图。接下来看下数据情况。标注文件如下：原生yolov5s... 查看详情

基于yolov5n6和tkinker实现的检测模型的可视化界面(代码片段)

项目介绍本项目基于yolov5n6和tkinker实现的检测模型的可视化界面项目连接；https://github.com/MaoliRUNsen/yolov5gui环境：硬件：本人电脑的显卡是RTX3060，并配置Pytorch-GPU关于pytorch安装查看官方文档PyTorchGetStarteddocs软件:Pych... 查看详情

基于yolov5n6和tkinker实现的检测模型的可视化界面(代码片段)

手把手教你使用yolov5训练自己的目标检测模型-口罩检测-视频教程(代码片段)

手把手教你使用YOLOV5训练自己的目标检测模型大家好，这里是肆十二（dejahu），好几个月没有更新了，这两天看了一下关注量，突然多了1k多个朋友关注，想必都是大作业系列教程来的小伙伴。既然有... 查看详情

基于yolov5n6和tkinker实现的检测模型的可视化界面(代码片段)

基于轻量级yolov5开发构建汉字检测识别分析系统

...实际项目的需要，之前的汉字检测模型较为古老了还使用的yolov3时期的模型，检测精度和推理速度都有不小的滞后了，这里要基于yolov5轻量级的模型来开发构建新版的目标检测模型，首先看下效果图：接下来... 查看详情

目标检测yolov5：模型构建解析(代码片段)

...码角度来解析YOLOv5中，模型是如何构建出来的。本文使用的是YOLOv5-5.0版本。模型的深度和宽度在YOLOv5中，模型结构基本是写在了.yaml中，5.0版本的YOLOv5共有yolov5s 查看详情

Yolov5自定义对象检测模型未加载

...oading【发布时间】：2021-09-2917:14:44【问题描述】：我已经使用yolov5为4个类训练了一个自定义对象检测模型。我已经下载了best.pt文件。我仍然对如何使用pytorch加载这个模型感到困惑。我尝试按照Yolov5官方文档运行以下代码来加载... 查看详情

android实现红绿灯检测(含android源码可实时运行)(代码片段)

...(含Android源码可实时运行)1.前言2.红绿灯检测数据集说明3.基于YOLOv5的红绿灯检测模型训练4.红绿灯检测模型Android部署（1）将Pytorch模型转换ONNX模型（2）将ONNX模型转换为TNN模型（3）Android端上部署模型（... 查看详情

基于语义分割groundtruth（gt）转换yolov5目标检测标签（路面积水检测例子）(代码片段)

...测格式转换呢？查遍全网没有很好的方法，因此使用opencv自己写了一个，检验效果还不错。这里的例子是基于极市平台的路面积水检测给出的数据集完成，由于平台只给了分割的示例数据，因此想使用yolo进行... 查看详情

行人检测(人体检测)3：android实现人体检测(含源码，可实时人体检测)(代码片段)

...源码，可实时人体检测)1.前言2.人体检测数据集说明3.基于YOLOv5的人体检测模型训练4.人体检测模型Android部署（1）将Pytorch模型转换ONNX模型（2）将ONNX模型转换为TNN模型（3）Android端上部署模型（4ÿ... 查看详情

行人检测(人体检测)3：android实现人体检测(含源码，可实时人体检测)(代码片段)

深度学习目标检测---使用yolov5训练自己的数据集模型（windows系统）(代码片段)

...教程和一些前辈的博客，实现了深度学习目标检测--使用yolov5网络训练自己的数据集并实现了对目标的识别。这篇博客我也是参考别人的博客教程，复现了使用yolov5训练自己的数据集，我呢主要是想记录下我自己的操... 查看详情

目标识别yolov5针对小目标检测的改进模型/添加帧率检测(代码片段)

...码修改——针对小目标识别所提到的一种改进方案。我所使用的是YOLOv5-5.0版本，数据集采用VisDrone数据集。检测头改进模型方面的修改：作者再模查看详情