基于Win10+TensorRT+C++部署Siamese图像相似度网络实战篇

博文中会给出本项目全套源码+运行文档+讲解文件的网盘链接,开发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<

你可能感兴趣的:(零/小样本学习干货分享,TensorRT,C++,图像相似度,Siamese,深度学习)