wts文件生成engine文件的方法

本文详细指导如何从x86平台获取Yolov5.wts文件,通过TensorRT迁移至Jetson,涉及git clone、编译及engine文件生成过程,适用于深度学习模型在嵌入式设备部署。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

在电脑上(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文件生成结束,并已移植到需要的目录中。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值