#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