wts文件生成engine文件的方法

在电脑上(x_86平台)上获取wts文件，并将该文件转移到jetson平台上，这里以yolov5.wts为例：

(备注)：获取wts文件方法可参考以下链接(只要获得wts文件，则链接内剩余内容无需看)： Pytorch模型--＞wts 文件--＞TensorRT模

• 下载tensorrtx项目(这里使用的是yolo5-v3.0版本)：

cd ~/

git clone -b yolov5-v3.0 https://github.com/wang-xinyu/tensorrtx.git

• 将yolov5.wts文件复制到~/tensorrtx/yolov5/文件夹下，并建立build文件夹(假设yolov5.wts文件在home目录下)：

cp ~/yolov5.wts ~/tensorrtx/yolov5/

cd ~/tensorrtx/yolov5/

mkdir build

cd build

cmake ..

make

• 在build文件夹内生成engine文件

cd ~/tensorrtx/yolov5/build/

sudo ./yolov5 -s

执行以上命令后需等待较长时间才能生成engine文件

• 将生成的engine文件(yolov5s.engine)移动到CDR项目中：

cd ~/tensorrtx/yolov5/build/

cp yolov5s.engine /opt/nvidia/deepstream/deepstream-5.1/sources/yolov5-ros/

• 至此yolov5s.engine文件生成结束，并已移植到需要的目录中。