本文主要是介绍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文件之后可以进行测试。
这篇关于pt格式文件转engine小记【yolov5-6.0版本】的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!