2024年6月3日
TensorRT c++部署onnx模型
摘要： 在了解一些概念之前一直看不懂上交22年开源的TRTModule.cpp和.hpp，好在交爷写的足够模块化，可以配好环境开箱即用，移植很简单。最近稍微了解了神经网络的一些概念，又看了TensorRT的一些api，遂试着部署一下自己在MNIST手写数字数据集上训练的一个LeNet模型，识别率大概有98. 阅读全文
posted @ 2024-06-03 22:18 凪风sama 阅读(870) 评论(1) 推荐(0) 编辑