Win10+libtorch1.1+opencv 筆記

這幾天剛剛把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文件

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市纺棺,隨后出現(xiàn)的幾起案子榄笙,更是在濱河造成了極大的恐慌,老刑警劉巖祷蝌,帶你破解...
    沈念sama閱讀 222,464評論 6 517
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件茅撞,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機米丘,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,033評論 3 399
  • 文/潘曉璐 我一進店門剑令,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人拄查,你說我怎么就攤上這事吁津。” “怎么了堕扶?”我有些...
    開封第一講書人閱讀 169,078評論 0 362
  • 文/不壞的土叔 我叫張陵碍脏,是天一觀的道長。 經(jīng)常有香客問我稍算,道長典尾,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 59,979評論 1 299
  • 正文 為了忘掉前任糊探,我火速辦了婚禮钾埂,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘科平。我一直安慰自己褥紫,他們只是感情好,可當我...
    茶點故事閱讀 69,001評論 6 398
  • 文/花漫 我一把揭開白布匠抗。 她就那樣靜靜地躺著故源,像睡著了一般污抬。 火紅的嫁衣襯著肌膚如雪汞贸。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 52,584評論 1 312
  • 那天印机,我揣著相機與錄音矢腻,去河邊找鬼。 笑死射赛,一個胖子當著我的面吹牛多柑,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播楣责,決...
    沈念sama閱讀 41,085評論 3 422
  • 文/蒼蘭香墨 我猛地睜開眼竣灌,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了秆麸?” 一聲冷哼從身側(cè)響起初嘹,我...
    開封第一講書人閱讀 40,023評論 0 277
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎沮趣,沒想到半個月后屯烦,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,555評論 1 319
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,626評論 3 342
  • 正文 我和宋清朗相戀三年驻龟,在試婚紗的時候發(fā)現(xiàn)自己被綠了温眉。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 40,769評論 1 353
  • 序言:一個原本活蹦亂跳的男人離奇死亡翁狐,死狀恐怖类溢,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情谴蔑,我是刑警寧澤豌骏,帶...
    沈念sama閱讀 36,439評論 5 351
  • 正文 年R本政府宣布,位于F島的核電站隐锭,受9級特大地震影響窃躲,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜钦睡,卻給世界環(huán)境...
    茶點故事閱讀 42,115評論 3 335
  • 文/蒙蒙 一蒂窒、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧荞怒,春花似錦洒琢、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,601評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至荧嵌,卻和暖如春呛踊,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背啦撮。 一陣腳步聲響...
    開封第一講書人閱讀 33,702評論 1 274
  • 我被黑心中介騙來泰國打工谭网, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人赃春。 一個月前我還...
    沈念sama閱讀 49,191評論 3 378
  • 正文 我出身青樓愉择,卻偏偏與公主長得像,于是被迫代替她去往敵國和親织中。 傳聞我的和親對象是個殘疾皇子锥涕,可洞房花燭夜當晚...
    茶點故事閱讀 45,781評論 2 361

推薦閱讀更多精彩內(nèi)容