文章作者:激萌の小宅
促销:¥0
价格:¥0
配送方式:
购买后立即生效(如购买异常,请联系站长)
付款之后一定要等待自动跳转结束,否则购买可能会失败
-
0 天
有效期
-
0
总销量
-
0
累计评价
YOLOv5 pt转engine文件
在前面的环境基础上,继续安装依赖包
下载TensorRT-8.6.1.6.Windows10.x86_64.cuda-12.0,博主是win11环境cuda 12.1
https://developer.nvidia.com/nvidia-tensorrt-8x-download#trt861
安装这个文件:TensorRT-8.6.1.6.Windows10.x86_64.cuda-12.0\TensorRT-8.6.1.6\python\tensorrt-8.6.1-cp38-none-win_amd64.whl
pip install tensorrt-8.6.1-cp38-none-win_amd64.whl
屏蔽掉export.py文件中的这两段代码,不然没法转换
然后执行转换命令
# 必须要先导出onnx文件才行 python export.py --weights yolov5s.pt --include onnx --device 0 # 然后再执行engine转换 python export.py --weights yolov5s.pt --include engine --device 0