在电脑上(x_86平台)上获取wts文件,并将该文件转移到jetson平台上,这里以yolov5.wts为例:
(备注):获取wts文件方法可参考以下链接(只要获得wts文件,则链接内剩余内容无需看):
Pytorch模型–>wts 文件–>TensorRT模
- 下载tensorrtx项目(这里使用的是yolo5-v3.0版本):
- 将yolov5.wts文件复制到~/tensorrtx/yolov5/文件夹下,并建立build文件夹(假设yolov5.wts文件在home目录下):
- 在build文件夹内生成engine文件
执行以上命令后需等待较长时间才能生成engine文件
- 将生成的engine文件(yolov5s.engine)移动到CDR项目中:
- 至此yolov5s.engine文件生成结束,并已移植到需要的目录中。