0


基于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<

本文转载自: https://blog.csdn.net/qq_42308217/article/details/125728812
版权归原作者 @会飞的毛毛虫 所有, 如有侵权,请联系我们删除。

“基于Win10+TensorRT+C++部署Siamese图像相似度网络实战篇”的评论:

还没有评论