接着上篇~~
3:准备使用tensorrt加速模型,onnx-tensortrt
将yolov5生成 pt文件先转成onnx文件,这个直接使用 yolov5源码里的 model/export.py转就成不难。
下面使用jetson nano搭建 onnx-tensorrt踩坑记录。
原本是想用ubuntu直接转成trt到时候再直接调用...
但ubuntu的环境搭建也挺麻烦的,不如直接就在jetson nano上转了。
jetson nano jetpack4.4 自带 tensorrt7.1 opencv4.1 cuda10.2 cudnn7.5 。
自己做是不可能的...参考文章如下:
jetson nano onnx-tensorrt 7.1,链接:https://blog.csdn.net/walletiger/article/details/109783232
TensorRT7 + Onnx_TensorRT 安装过程记录,链接:https://blog.csdn.net/weixin_44638957/article/details/103704661
出错记录: