博文中会给出本项目全套源码+运行文档+讲解文件的网盘链接,开发SDK接口,可同步迁移至其他C++项目或基于自定义训练网络进行部署。此教程可大幅度减少试错成本,一步一步跟着本文教学,若不成功,博主永不更文。
首先给出实测性能指标:
- 在1080显卡上,两张105×105图像一次相似度预测耗时约6ms,较原始Pytorch推理提升约5倍。
- 推理精度和Pytorch源码保持高度一致。
文末源码相关配置:
- TensorRT-8.2.1.8.Windows10.x86_64.cuda-10.2.cudnn8.2
- Cuda-10.2
- Cudnn-10.2-windows10-x64-v8.2.1.32
- Opencv-4.5<



