YOLOX Window10 TensorRT 全面部署教程
配置流程
官网这部分写的非常笼统,我把完整教程分享给大家。
环境:
window10+GTX1650+TensorRT-7.2.3.4-CUDA11.1
yolox程序在ubuntu上,所以部分内容是ubuntu上操作
其他环境不保证
训练自己的模型
训练部分请参照官网流程进行训练,或者直接使用官方提供的几个模型也可。得到.pth文件。例如这里我的是:best_ckpt.pth
安装torch2trt
yolox官网这里真的懒,就给了个链接,好在之前我用过,自己的环境中也有这个模块。下面来说说这个模块。
首先还是要先进入到这个git地址:https://github.com/NVIDIA-AI-IOT/torch2trt
注意注意,他的安装教程可没有写在最前面,最前面是一些功能的简单的介绍,吹一下他这个工具多好用,然后支持多少多少模型等等,我们直接跳过去,看setup部分。
安装之前要确保自己的环境中已经安装了tensorrt,这些之前我都装过ÿ