yolov7_tensorrt

代做

code  资源释放、前后处理代码优化(其中维度尺寸等信息直接写死了，需要调整)。 部署  plugin

介绍

yolov7的tensorrt部署

环境搭建

1. 训练环境搭建
    参考子仓README.md
2. 部署环境搭建
    1)onnxRunTime部署
     训练环境中pip(或conda)安装OnnxRunTime即可。
    2)tensorRt部署
     宿主机安装nvidia驱动、dockers以及nvidia container toolkit；
     拉取nvidia/cuda docker(devel);
     docker内安装tensorRt以及cudnn。

模型训练

1. 参考子仓README.md

模型部署

mkdir build
cd build
cmake ..
make

结果测试