yolov7转换tensorrt问题
yolov7
YOLOv7 - 实现了一种新的实时目标检测算法,用于图像识别和处理。
项目地址:https://gitcode.com/gh_mirrors/yo/yolov7
免费下载资源
·
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 年前
更多推荐
已为社区贡献8条内容
所有评论(0)