- 博客(1)
- 收藏
- 关注
原创 TX2+Yolov5之tensorRT加速
(2)如果上一步没问题,那大概率就是你的tensorRT的版本和你训练模型的版本不同,可以从官网下载个tensorRT版本的yolov5.pt测试一下。5.接下来去到目录tensorrtx下的yolov5文件夹,创建一个build文件,并生成生成makeFile,如果用的自己的权重文件请看8.6.将第4步生成的.wts文件复制到tensorrtx/yolov5/build里,运行代码生成.engine文件。3.将官方的权重文件'yolov5s.pt',或自己训练出来的模型,也放在yolov5主文件夹中。
2023-06-02 21:35:52 506
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人