用 nanodet 训练口罩检测模型,并在 jetson nano 下部署测试
一 nanodet 介绍
最近比较火的 one stage 物体定位 和检测模型, 作者不光是取得了比较高的 学术 score ( 精度和性能), 并且把工程部署的细节也落地了, 在 android 下也有不错的方案。
真是牛的可以, 感觉工程师们再不努力就没饭吃了 。。
https://blog.csdn.net/amusi1994/article/details/110072440
二 数据集准备:
可以看上面链接的介绍 ,比较详细
也可以参考我的另外一个博文, 里面 github 地址里 包含了nanodet 数据集的准备
https://blog.csdn.net/walletiger/article/details/110733182
nanodet 需要 coco 格式数据集,标注文件格式不大一样, train 和 val 放在两个 独立的 json 文件中。
除了数据集,还要有个 训练 模型描述文件, 我放在 config/nanodet-facemask.yaml
从 nanodet-m.yml copy 过来, 改了如下:
三 训练
python tools/train.py nanodet-facemask.yaml
nandet 训练 过程出错解决:
一个是 coco 需要 图片文件名是整数, 我的数据集很冤。报错了, 这个根据报错代码行数 简单粗暴的忽略了。 (不该这么简单粗暴)
第二个是 trainer.py 训练 报错:
list object has no method cpu ..
根据自己的理解 修改了下, it works ...
四 部署 nanodet 模型到 jetson nano 并验证
先在, 然后把训练 好的模型放上去
jetson nano下载代码:
git clone https://github.com/RangiLyu/nanodet
上传训练 好的模型,在:
output/model_best/model_best.pth
我修改了 demo (demo2), 更换了 camera 类来适配 jetson nano 的 gstreamer camera backend
摄像头采集类:
https://github.com/walletiger/jetson_nano_py/blob/master/camera.py
python demo/demo2.py webcam --config nanodet-facemask.yml --model model_best.pth --camid jetcam
demo 效果:
jetson nano 口罩佩戴检测
pth 模型直接跑 获得了 640x360 @15fps 的性能,初步的测试速度比 yolov5s 我验证的结果 要快! 后续 能否 通过 onnx , onnx->tensorRT 。。 还有多少提升空间。 再研究吧
用 nanodet 训练口罩检测模型,并在 jetson nano 下部署测试相关推荐
- 无需代码即可训练和可视化口罩检测模型
介绍 新冠改变了人们的生活,出门佩戴口罩也已经成为了大部分人的习惯.但并不是每一个人都能遵守新冠时代下的行为规范.为了更好地防控新型冠状病毒的传播,监控探头和物联网摄像头被广泛的应用于监督人们在公共场 ...
- 手把手教你使用 YOLOV5 训练目标检测模型
作者 | 肆十二 来源 | CSDN博客 这次要使用YOLOV5来训练一个口罩检测模型,比较契合当下的疫情,并且目标检测涉及到的知识点也比较多. 先来看看我们要实现的效果,我们将会通过数据来训练一个口 ...
- 使用opencv训练目标检测模型基于cascade模型
使用opencv训练目标检测模型基于cascade模型 基于Haar特征的cascade分类器(classifiers) 是Paul Viola和 Michael Jone在2001年,论文" ...
- MSCOCO数据集转VOC数据集训练目标检测模型
MSCOCO数据集转VOC数据集训练目标检测模型 Images 2014 Train images [83K/13GB] 2014 Val images [41K/6GB] 2014 Test ima ...
- Python口罩识别检测全网最全OpenCV训练分类器具体步骤(以训练口罩检测分类器为例)附分类器和数据集下载地址
声明 本文仅在CSDN发布,其他均为盗版.请支持正版! 正版链接: https://blog.csdn.net/meenr/article/details/115825671 OpenCV训练口罩检测 ...
- caffe使用训练好的模型对自己的一张图片进行测试
前面学习了如何训练模型,也学了如何用测试集测试模型.但是好像还不会拿一张自己的图片去模型中进行测试.这篇文章就主要聊聊如何使用训练好的模型对自己的一张图片进行测试!(前面提到了做的项目是藏文识别, ...
- R语言构建文本分类模型并使用LIME进行模型解释实战:文本数据预处理、构建词袋模型、构建xgboost文本分类模型、基于文本训练数据以及模型构建LIME解释器解释多个测试语料的预测结果并可视化
R语言构建文本分类模型并使用LIME进行模型解释实战:文本数据预处理.构建词袋模型.构建xgboost文本分类模型.基于文本训练数据以及模型构建LIME解释器解释多个测试语料的预测结果并可视化 目录
- 基于OpenCV训练口罩检测数据集并测试
以下内容是利用opencv自带的训练器opencv_traincascade.exe与opencv_createsamples.exe,来对口罩数据集进行训练.内容是自己操作过程中的笔记,可能会有些杂 ...
- WiderFace数据集用于训练人脸检测模型
下载链接: http://mmlab.ie.cuhk.edu.hk/projects/WIDERFace/ 该数据集可用于进行人脸检测模型的训练 如果做训练,可以点击网页中的 (国内还是百度吧,会快一 ...
最新文章
- 日常该怎么处理繁杂的工作?用智办事更有序、高效!
- VTK一个面向对象的可视化类库
- php怎样下载网上的文件,php怎样实现文件下载
- Java LinkedList公共对象pollLast()方法(带示例)
- jquery和css的区别是什么?
- 汽油与消费需求问题的MonteCarlo求解方法
- NLP提效,除了选择合适的模型,就是数据增强了
- vue 中使用 js-xlsx 导出数据导 excle, 支持多 sheet
- 软件工程之图书管理系统总体设计
- 测试技术-兼容性测试
- (生物信息学)R语言与统计学入门(九)—— 单因素cox回归分析
- 如何查看linux的系统配置,多少个核心,多少个线程?CPU的主频 查看内存
- ODrive配置云台电机GIMBLE_TYPE
- 【庄碰辉】万般滋味,皆是生活常态
- 分布式数据库之TiDB
- linux 命令行别名,bash命令行实用的别名-alias命令
- NFC手机手环脱机模拟加密门禁卡
- 阿里技术大神「多隆」的成神之路
- Android动态化方案
- location.hash 原生js页面跳转/路由