1。通过试验采用PT-》tensort .engine文件转换一直有问题(网络结构不匹配的问题)这种方法放弃

可采用pt->onnx->engine的方式,直接参考

YOLOv7 Tensorrt Python部署教程_一笑奈何LHY的博客-CSDN博客

1.主要下载yolov7-master中自带的export.py转换成onnx

2.将生成的onnx模型复制到tensorrt安装目录文件下tensorrt/bin文件夹下,利用trtexec文件生成

trtexec --onnx=./yolov7.onnx --saveEngine=./yolov7_fp16.engine --fp16 --workspace=200

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐