Yolov5 口罩识别
自定义口罩数据集进行目标检测
目录
- 自定义口罩数据集进行目标检测
- Yolov5 代码地址
- 数据集
- 一、手动收集标记数据集
- 二、使用公开数据集
- 改动源码
- 训练结果
- 迁移学习
- 制作自定义数据集的注意事项
Yolov5 代码地址
https://github.com/ultralytics
数据集
数据集可以手动标记或者使用公开数据集
一、手动收集标记数据集
使用Labelimg软件标记,网上有很多介绍
软件下载
二、使用公开数据集
https://public.roboflow.com/
以口罩数据集为例: 口罩数据集
训练图片只有105张,比较少,仅用作实验,实际做项目时每个类别的图片至少有1500张,并且图片应用场景要多种多样。
下载解压后后改名为MASK,内容如下(可能有区别),要用的只有 train和valid文件夹。
改动源码
Yolov5
下载好yolov5源码,打开后在data文件夹下新建mask.yaml,填入以下信息,前三行改为自己MASK文件的绝对路径
train: /media/system/dataset/MASK/train/images
val: /media/system/dataset/MASK/valid/images
test: /media/system/dataset/MASK/test/imagesnc: 2
names: ['mask', 'no-mask']
数据集中只有mask和no-mask2个类别,修改yolov5/models/yolov5s.yaml
,将第四行的 nc = 80修改为nc = 2
接下来执行训练命令
python3 train.py --img 640 --batch 8 --epochs 500 --data data/mask.yaml --cfg models/yolov5s.yaml --weights ''
会自动下载Yolov5s的预训练模型进行训练
训练结果
打开wandb注册账号,登录,按照提示操作
训练过程会在wandb中清楚地显示出来,方便调参。
总的结果
训练结果测试
mAP
train的三个损失函数
val的三个损失函数
GPU监控
学习速率
可以看出虽然训练样本很少,但是训练的结果还是不错的。这主要是因为用了yolov5在coco数据集上预训练的模型,特征比较丰富,而且口罩检测的类别少,只有两类,区分度较大,带不带口罩很容易看出来。
迁移学习
为了加快训练速度,可以采用迁移学习,冻结前面的特征层,只训练后面的检测层或者其他自定义的卷积层
data/yolov5s.yaml
# YOLOv5
Yolov5 口罩识别相关推荐
- Python+Yolov5人脸口罩识别
程序示例精选 Python+Yolov5人脸口罩识别 如需安装运行环境或远程调试,见文章底部微信名片,由专业技术人员远程协助! 前言 Yolov5比较Yolov4,Yolov3等其他识别框架,速度快 ...
- 人脸口罩识别的项目总结
人脸口罩识别的项目总结 最近一段时间,国内部分地区又爆发了新冠疫情,传播速度很快.但最近天气炎热,人们戴口罩的自觉性不如从前了,在商场等公共场所,需要专门的人员去提醒顾客佩戴好口罩.因此萌发了使用计算 ...
- 基于深度学习的口罩识别与检测PyTorch实现
基于深度学习的口罩识别与检测PyTorch实现 1. 设计思路 1.1 两阶段检测器:先检测人脸,然后将人脸进行分类,戴口罩与不戴口罩. 1.2 一阶段检测器:直接训练口罩检测器,训练样本为人脸的标注 ...
- 基于OpenCV实现口罩识别
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达 昨天在GitHub上看到了一个开源的项目,是利用深度学习来检测是否 ...
- 免费GPU支持!口罩识别、安全帽识别算法赛同步开启,丰厚奖品等你来战!
2020年,极市隆重推出 [极市平台视觉算法季度赛] ,旨在邀请广大视觉开发者用算法解决实际问题,第一季算法赛于2020年3月3日正式启动! 本次活动由极视角主办.英特尔联合主办,由UCloud提供技 ...
- python利用tensorflow识别圆_RaspberryPi上实现佩戴口罩识别——2020电赛F题小记
今年的电赛题目非常反常,传统控制题目基本没有,新增加了测距题目,甚至物联网题目,Ai题目都出来了.直接电赛变算法+钞能力大赛.看到F题:测温+人脸识别,碰巧我们手头有调好的红外测温模块+树莓派+ope ...
- 人脸、微笑、口罩识别
一.人脸图像特征提取的方法 1.HOG 特征 方向梯度直方图(Histogram of Oriented Gradient, HOG)特征是一种在计算机视觉和图 像处理中用来进行物体检测的特征描述子. ...
- 基于dlib库进行微笑识别和口罩识别
环境配置 tensorflow和keras参考我之前的博客https://blog.csdn.net/A981012/article/details/106650686 下载dlib库不能从anaco ...
- 树莓派 --- 人脸口罩识别智能监控
目录 初步效果设想 像SIRI那样的语音唤醒,开启服务 只有当摄像头检测到人脸时,舵机才开始调整位置,让人脸保持C位 检测人脸是否戴口罩了,没带就发出"提示",戴了就不管 长时间不 ...
最新文章
- 【知识便利贴】ImageNet得的比Best Paper还牛的Longuet-Higgins奖是什么?
- PHP 拓展 开发,开发php扩展
- 页面设计四大基本原则
- Google Analytics 搜索引擎来源
- 【待完成】并发包下常用到线程工具类汇总
- Mysql中Innodb大量插入数据时SQL语句的优化
- oracle 交叉链接,Oracle里的交叉SQL语句写法-数据库专栏,ORACLE
- 用汇编的眼光看C++(之虚函数)
- SAP License:数据导入时的问题与总结
- [合作] 钢结构结构健康监测研究与实验
- sqlserver201无效的许可证书_sql提示无效数字
- [Leetcode] longest common prefix 最长共同前缀
- 86版五笔-字根速记
- Linux内存查看通用方法
- 数据结构期末作业思路分享——库存管理小程序(云数据库)
- hello ,酷狗音乐产品体验报告
- C语言 删除单链表中值为e的全部元素
- Kate Spade_百度百科
- (软件测试工程师)一线大厂新人工作指南!
- Rodrigues' Rotation Matrix(罗德里格旋转矩阵)
热门文章