qq_46144078的博客

原创 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主文件夹中。