点击查看系列文章目录
0 背景
deepstream 5.0 GA 版本的 SDK 中包含有 yolov3 的检测,还没有正式加上 yolov4 的功能,不过 nvidia 官方也在 github 上发布了相关的代码,本文对使用方法做一个介绍
官方介绍:DeepStream SDK FAQ - #8 by ersheng - DeepStream SDK - NVIDIA Developer Forums
1 模型准备
官方没有提供转换后的模型,需要我们自己转换
git clone https://github.com/Tianxiaomo/pytorch-YOLOv4.git
cd pytorch-YOLOv4
建议创建 conda 环境来安装部署相关软件