pytorch yolov3 代码详解_PyTorch C++ libtorch的使用方法(1)-nightly 版本的 libtorch

问题描述：

按照PyTorch中文教程的【在 C++ 中加载 PYTORCH 模型】一文，尝试调用 PyTorch模型。

1. 例子来源

在 C++ 中加载 PYTORCH 模型

我是使用Qt新建纯C++工程，然后CMake编译、运行的，直接贴代码：

1.1 C++代码

main.cpp

#include <torch/script.h> // One-stop header.
#include <torch/torch.h>
#include <iostream>
#include <memory>using namespace std;int main()
{//    std::shared_ptr<torch::jit::script::Module> module = torch::jit::load(
//                "/home/book/Jupyterlab work/model.pt");torch::jit::script::Module module = torch::jit::load("/home/book/Jupyterlab work/model.pt");// Create a vector of inputs.std::vector<torch::jit::IValue> inputs;inputs.push_back(torch::ones({1, 3, 224, 224}));// Execute the model and turn its output into a tensor.at::Tensor output = module.forward(inputs).toTensor();std::cout << output.slice(/*dim=*/1, /*start=*/0, /*end=*/5) << 'n'; //与前面python的输出是一样的！std::cout << "okn";
}

那么我遇到的与原例子冲突的地方是：

//原例子中的代码：
std::shared_ptr<torch::jit::script::Module> module = torch::jit::load(argv[1]);//我改后的代码：
torch::jit::script::Module module = torch::jit::load(argv[1]);

使用原例子中的代码，会报出如下的错误：

/home/book/qt5.12/example-app/main.cpp:16: error:
conversion from ‘torch::jit::script::Module’ to non-scalar type ‘std::shared_ptr<torch::jit::script::Module>’ requested
std::shared_ptr<torch::jit::script::Module> module = torch::jit::load(                                                                           ^

报错内容

解决方案：

conversion to non-scalar type torch::jit::load("model.pt") · Issue #22382 · pytorch/pytorch

亲测有效！

但是，这也引入了新的问题，'nightly build' 表示什么？联系到前面下载LibTorch时的下载地址，其中也出现了'nightly'，二者之间是不是有所关联，难道我下载的LibTorch版本不对？

貌似nightly build 版本意思是没有正式发布的，这种说法有待考证。

nightly build 版本的 libtorch

1.2 CMakeLists.txt

cmake_minimum_required(VERSION 3.5)project(example-app LANGUAGES CXX)set(CMAKE_CXX_STANDARD 11)
set(CMAKE_CXX_STANDARD_REQUIRED ON)set(Torch_DIR /home/book/libtorch/share/cmake/Torch)
find_package(Torch REQUIRED)add_executable(example-app main.cpp)
target_link_libraries(example-app "${TORCH_LIBRARIES}")
set_property(TARGET example-app PROPERTY CXX_STANDARD 11)

可能会用到的链接：

（1）libtorch初体验 - 一度逍遥 - 博客园（加载 "xxx.pt" 模型）

（2）ubuntu16.04改变桌面图标大小显示-百度经验（ubuntu 使用技巧）

（3）pytorch/pytorch

（4）Update our cpp export tutorial by Krovatkin · Pull Request #560 · pytorch/tutorials

（5）conversion to non-scalar type torch::jit::load("model.pt") · Issue #22382 · pytorch/pytorch （靠这个解决的问题！）

（6）walktree/libtorch-yolov3