DeepStream系列之yolov7部署测试
Posted 宗而研之
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了DeepStream系列之yolov7部署测试相关的知识,希望对你有一定的参考价值。
点击查看系列文章目录
0 背景
测试环境:deepstream 6.1.1,yolov7,T4
1 环境安装
1.1 deepstream 安装
参考《DeepStream系列之6.1版本安装及测试》安装最新版的 deepstream,这里采用的是 docker 方式安装。
1.2 yolov7 安装
安装配置 yolov7 环境的目的是为了生成 onnx 中间模型,onnx 转 tensorrt 的过程需要在容器当中实现
为了避免环境干扰,我们创建一个 conda 虚拟环境
conda create -n yolov7
conda activate yolov7
下载源码
git clone https://github.com/WongKinYiu/yolov7.git
cd yolov7
pip install -r requirements.txt
转换生成 onnx
wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-tiny.pt
python expo
以上是关于DeepStream系列之yolov7部署测试的主要内容,如果未能解决你的问题,请参考以下文章