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

GitHub 加速计划 / yo / yolov7
13.13 K
4.14 K
下载
YOLOv7 - 实现了一种新的实时目标检测算法,用于图像识别和处理。
最近提交(Master分支:3 个月前 )
a207844b - 1 年前
2c612d33 - 1 年前
Logo

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

更多推荐