TensorRT
_yuki_
一路打怪升级,直至见到大魔王
展开
-
TensorRT (五)yoloV5 训练模型
环境:tensorRT 6 / tensorRT 7 Ubuntu 18.04 CUDA 10.1一、下载项目yoloV5 项目:地址二、安装相应环境pip install -r requirements.txt训练关键依赖:Cythonmatplotlibnumpyopencv-pythonpillowPyYAMLscipytensorboardtorch>=1.6.0torchvision>=0.7.0tqdmps:t...原创 2020-11-12 18:19:43 · 2932 阅读 · 5 评论 -
TensorRT (四)yoloV4 / yoloV4-tiny 模型转化
环境:tensorRT 6 / tennsorRT 7 Ubuntu 18.04 CUDA 10 CUDNN 7.6.3依赖:numpy 1.16.2 onnx 1.4.1 pycuda 2019.1.1一、下载模型与参数yolov4.weights:下载 yolov4.cfg:下载 yolov4-tiny.weights:下载 yolov4-tiny.cfg:下载二、Weights 转换 ONNXfrom __future__ import print...原创 2020-07-24 14:53:49 · 6672 阅读 · 40 评论 -
TensorRT (三)yoloV3 / yoloV3-tiny onnx 转 tensorRT
环境:tensorRT 6 Ubuntu 18.04 CUDA 10 CUDNN 7.6.3前面的文章用的 tensorRT 7 为何 换用 6 ?主要 用 tensorRT 7 调用会报错,但查询官网文档是有该API:(无奈暂时只能这样)engine.serialize()一、数据化处理 data_processing.py( 可修改 #!!!!!!...原创 2020-04-10 16:44:43 · 1798 阅读 · 1 评论 -
TensorRT (一) Ubuntu 安装配置
环境:Ubuntu 18.04 CUDA 10 CUDNN 7.6.3 TensorRT 7一、 下载tar包TensorRT官方:地址二、 解压tar包$ tar xzvf TensorRT-7.x.x.x.<os>.<arch>-gnu.cuda-x.x.cudnn7.x.tar.gz-其中 6.x.x.x is y...原创 2020-04-09 16:44:19 · 1227 阅读 · 1 评论 -
TensorRT (二)Python3 yoloV3 / yoloV3-tiny 转 onnx
环境: python3.6 /anacoda cuda10.0 cudnn7.6.4一、所需的包pip install numpy #1.16.1 如果你有tensorflowpip install onnx==1.4.1pip install pycuda==2019.1.1pip install Pillow==...原创 2020-04-09 13:18:29 · 1890 阅读 · 2 评论 -
TensorRT (一) Win10 安装配置
我的环境:Anaconda3 /Python 3.7 Tensorflow-gpu 1.4 TensorRT 7 CUDA 10 CUDNN 7.6 VS 2019一、下载TensorRTTensorRT 官方下载:地址二、添加环境变量PATH:D:\it\TensorRT-7.0.0.11\lib (这是我的)三、安装uff和graphsu...原创 2020-03-20 12:34:13 · 8221 阅读 · 6 评论