pt格式文件转engine小记【yolov5-6.0版本】

背景

项目是使用yolov5-6.0的版本，需要加一个新模型进去，yolov5提供的类别有很多，我这里使用chair椅子。第一步就是先把提供的pt文件转化为tensorrt所需要的engine格式的文件，在官网上有提供转换方法。（似乎高版本的yolo可以直接导出engine格式文件了）
tensorrt github地址
yolov5-6.0版本

下载所需要的文件

因为版本不一样，使用git命令下载的只能下最新版，下载压缩包解压使用


解压完毕之后，看对应版本的操作流程，如下：

照着这个步骤一步步来就可以。其中需要注意点如下，在生成engine的时候可能会出现错误，就是build tensorrtx/yolov5 and run环节里，其中我做的修改如下，找到yolov5的目录下CMakeLists.txt文件，修改tensorrt的目录成自己的。生成完engine文件之后可以进行测试。