0 背景
在医院、机场、车站等场所中,需要对行人进行实时口罩检测。本文基于NVIDIA TLT工具,介绍口罩检测模型的训练及部署方法,实现在嵌入式端实时处理的效果。在嵌入式上部署推理性能表如下,可以看到裁剪后的模型,即使在 Jetson Nano平台上也能跑搭配20帧,做到实时处理的效果。
- 注意:这个帧率是使用deepstream获得的端到端帧率,其中nano是使用的FP16精度,Xavier NX和AGX使用的是int8精度
- NX和AGX有两种推理模块,分别是GPU和DLA,当在DLA上推理的时候,GPU是空闲的,可以处理其它任务
流程如下
官方博客: