這幾天剛剛把libtorch加載模型弄明白，記錄一下累魔。
1韵丑、正確安裝VS2017+opencv+cmake +pytorch 1.1
2、官網(wǎng)下載libtorch cpu 1.1版本(注意pytorch與libtorch版本一致)
3锌仅、pytorch 導出模型

import torch
from torchvision import models
model = models.resnet18()

#導入已經(jīng)訓練好的模型
#state = torch.load('latest.pt')
#model.load_state_dict(state['model_state_dict'], strict=True)

#注意模型輸入的尺寸
example = torch.rand(1, 3, 224, 224)
model = model.eval()

traced_script_module = torch.jit.trace(model, example)
output = traced_script_module(torch.ones(1,3,224,224))
traced_script_module.save("model.pt")

4、cmake 編寫

cmake_minimum_required(VERSION 3.14 FATAL_ERROR)
project(custom_ops)

find_package(Torch REQUIRED)
find_package( OpenCV REQUIRED )

include_directories( ${OpenCV_INCLUDE_DIRS} )

add_executable(example-app example-app.cpp)
target_link_libraries(example-app ${TORCH_LIBRARIES} ${OpenCV_LIBS} )
set_property(TARGET example-app PROPERTY CXX_STANDARD 11)

5墙贱、新建build文件夾 并且進入build 打開命令行,這里的Visual Studio 15 Win64是指VS2017

cmake -DCMAKE_PREFIX_PATH=D:\yourpath\opencv\build\x64\vc15\lib;D:\yourpath\libtorch -DCMAKE_BUILD_TYPE=Release -G"Visual Studio 15 Win64" ..

6热芹、打開VS項目sln
7、編寫libtorch代碼加載模型

#include <torch/script.h>
//#include <ATen/ATen.h>
#include <opencv2/opencv.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <iostream>
#include <memory>

using namespace std;

shared_ptr<torch::jit::script::Module> load_model(string model_path)
{
    shared_ptr<torch::jit::script::Module> module = torch::jit::load(model_path);
    //module->to(device);
    assert(module != nullptr);
    std::cout << "load model ok\n";
    return module;
}




int main(int argc, const char* argv[])
{

    if (argc != 3) 
    {
        cerr << "usage : example-app <path-module> <path-image>";
        return -1;
    }

    shared_ptr<torch::jit::script::Module> module = load_model(argv[1]);
    
    cv::Mat image = cv::imread(argv[2]);
    
    cvtColor(image, image, cv::COLOR_BGR2RGB);
    cv::Mat img_float;
    image.convertTo(img_float, CV_32F, 1.0 / 255);
    cv::resize(img_float, img_float, cv::Size(224, 224));
    auto img_tensor = torch::from_blob(img_float.data, { 1, 224, 224, 3 });
    img_tensor = img_tensor.permute({ 0, 3, 1, 2 });

    //輸入
    std::vector<torch::jit::IValue> inputs;

    inputs.push_back(img_tensor);
      // evalute time
    double t = (double)cv::getTickCount();
    auto out = module->forward(inputs).toTensor();
    std::cout << out << std::endl;

    t = (double)cv::getTickCount() - t;
    printf("耗費時間為:  %gs\n", t / cv::getTickFrequency());
    inputs.pop_back();

    return 0;
}

8惨撇、將c10.dll剿吻、caff2.dll、torch.dll串纺、opencv_wordl400.dll放到與exe文件同級目錄
9丽旅、命令行運行exe文件