1.首先pt转换wts
代码下载地址:https://github.com/wang-xinyu/tensorrtx
使用的是py脚本gen_wts 下完后,目录结构是这样的,在你选的模型里找到wts生成脚本
python gen_wts.py -w XXXXXXX.pt -o XXXXXXX.wts -t seg
2.wts转eng
先mkdir个build在build目录里cmake…make就有了可执行文件,
./yolov8_seg -s yolov8s-seg.wts yolov8s-seg.engine s