超级会员免费看
该博客介绍了如何在Ubuntu环境下，利用TensorRT将ResNet50模型转化为FP16精度的Engine文件，以实现模型的部署和加速。通过运行Python脚本得到权重文件，再用C++工程配置和编译，最终生成可执行的推理引擎。
#TensorRT模型部署# ResNet50基于TensorRT FP16生成Engnie文件的C++工程，ubuntu运行，可自行修改为win下使用，推理代码可自行根据自己的实际情况书写，需要的小伙伴自取哈。
执行步骤简单：
第一步：先运行resnet50.py文件得到resnet50.pth权重文件；
第二步：运行inference.py文件得到resnet50.wts权重文件；
第三步：配置CmakeLists.txt文件，主要是修改其中Tensorrt的路径
第四步：在resnet_50_trt目录下创建build文件夹（mkdir build&cd build）
第五步：cmake …
第六步：make
第七步：./resnet50，结束后即可生成resnet50.engine
====
源代码下载 resnet_50_trt.zip
文件结构：
关注公众号，回复1001，获取文件下载动态密码。
链接: https://pan.baidu.com/s/1jxAEqU9qfQvaA-uyfYgHkg 提取码: hvlb
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
