yolov7转换tensorrt问题
·
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
更多推荐
已为社区贡献8条内容
所有评论(0)