文章目录
- 准备工作
- 文件目录
- 进行编译
准备工作
- cmake
- libTorch
文件目录
进行编译
CMakeLists.txt
cmake_minimum_required(VERSION 3.12 FATAL_ERROR)
project(example-app)
find_package(Torch REQUIRED)
find_package(OpenCV REQUIRED)
if(NOT Torch_FOUND)
message(FATAL_ERROR "Pytorch Not Found!")
endif(NOT Torch_FOUND)
message(STATUS "Pytorch status:")
message(STATUS " libraries: ${TORCH_LIBRARIES}")
message(STATUS "OpenCV library status:")
message(STATUS " version: ${OpenCV_VERSION}")
message(STATUS " libraries: ${OpenCV_LIBS}")
message(STATUS " include path: ${OpenCV_INCLUDE_DIRS}")
add_executable(example-app example.cpp)
target_link_libraries(example-app ${TORCH_LIBRARIES} ${OpenCV_LIBS})
set_property(TARGET example-app PROPERTY CXX_STANDARD 11)
example.cpp
// One-stop header.
int main(int argc, const char* argv[]) {
if (argc != 2) {
std::cerr << "usage: example-app <path-to-exported-script-module>\n";
return -1;
}
// Deserialize the ScriptModule from a file using torch::jit::load().
torch::jit::script::Module module = torch::jit::load(argv[1]);
assert(module != nullptr);
std::cout << "ok";
}
设置libTorch和opencv的根目录
cmake
-DCMAKE_PREFIX_PATH=D:\MyWorkSpace\Lib\opencv\build\x64\vc15\lib;D:\MyWorkSpace\Lib\libtorch
-DCMAKE_BUILD_TYPE=Release -G “Visual Studio 15 Win64” …