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

作者：Monodyee | 2024-04-14 07:33:38

踩

背景

项目是使用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文件之后可以进行测试。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Monodyee/article/detail/420874