博文中会给出本项目全套源码+运行文档+讲解文件的网盘链接,开发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
版权归原作者 @会飞的毛毛虫 所有, 如有侵权,请联系我们删除。
版权归原作者 @会飞的毛毛虫 所有, 如有侵权,请联系我们删除。