利用C++来调用PyTorch训练好的模型,其实也是利用官方强大的LibTorch库。
一、LibTorch的安装
虽然说安装,其实就是下载官方的LibTorch包而已,从官方网站中选择PyTorch(1.1),libtorch,以及cuda的版本,其中会出现下载链接,这里为cuda9.0的链接:https://download.pytorch.org/libtorch/cu90/libtorch-shared-with-deps-latest.zip。下载好找个路径解压。解压完放在那不动!!
二、PyTorch模型训练
这里我使用了最为简单ResNet50的预训练模型,其中保存跟踪模型的代码如下:
import torch
import torchvision.models as models
from PIL import