pytorch模型解析

本文主要介绍关于pytorch,c++,python的知识点，对机器视觉系列4：C++部署pytorch模型和pytorch模型解析有兴趣的朋友可以看下由【Arvin L】投稿的技术文章，希望该技术和经验能帮到你解决你所遇的机器视觉,人工智能相关技术问题。

pytorch模型解析

系列文章目录

第一章：Visual Studio 2019 动态链接库DLL建立

第二章：VS动态链接库DLL调试

第三章：VS2019 OpenCV环境配置 

第四章：C++部署pytorch模型

---

系列文章目录

前言

一、C++部署pytorch？

二、Libtorch配置

1.下载Libtorch

2.VS2019配置Libtorch

2.1配置VC++目录

 2.2配置链接器

 2.3Libtorch环境变量配置

三、pytorch模型转换为Libtorch使用的pt

四、C++中Libtorch的使用

参考文献

---

前言

环境：visual studio 2019；OpenCV4.5.5；pytorch1.8；libtorch10.2；

一、C++部署pytorch？

pytorch模型在C++部署我知道的一般有两种方式，一种是讲pytorch模型转为onnx，使用opencv的DNN模块部署。一种是使用pytorch对应版本的Libtorch部署。onnx试验发现，转换模型过程之后，语义分割精度相差太大，最终选择Libtorch部署。

二、Libtorch配置

注意事项：1，Libtorch版本与pytorch版本要对应

                  2，Libtorch与pytorch的CPU/GPU要对应

1.下载Libtorch

Pytorch官网下载，有Release版本和Debug版本

https://pytorch.org/

windows+VS2019+PyTorchLib配置使用攻略 - 简书 (jianshu.com)

https://www.jianshu.com/p/2371ee8b45f0

 

本文《机器视觉系列4：C++部署pytorch模型》版权归Arvin L所有，引用机器视觉系列4：C++部署pytorch模型需遵循CC 4.0 BY-SA版权协议。