链接: 利用LibTorch部署PyTorch模型.
大体是按照教程来做的,但由于作者没有详细说明,初学者也会踩了不少坑。
如果不准备使用cmake,可以参考链接: Win10+VS2017配置LibTorch开发环境,不过麻烦的是需要自己去导入各种需要的包,opencv,torch这些
大体上步骤和利用 cmake 工具生成 Visual Studio 工程文件类似。
比如,我在此处创建了一个PytorchC的文件夹,在Cmake之前还需要在其下创建一个文件夹,不如build(对于编译源码是这样,此处只是为了创建一个工程,就按照下面的vsprojects创建了)
在Configure之前,需要更改CMakefile如下,同时需要在vsprojects文件夹下创建main.cpp文件。不然按照demo中提供的程序无法找到opencv库。,当然此处的路径是自己的opencv和libtorch的路径了。
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(example_torch)
set(CMAKE_PREFIX_PATH "G:/libtorch")
set(OpenCV_DIR "E:/opencv/opencv/build")
find_package(Torch REQUIRED)
find_package(OpenCV QUIET)
if(NOT OpenCV_FOUND)
find_package(OpenCV 2.4.3 QUIET)
if(NOT OpenCV_FOUND)
message(FATAL_ERROR "OpenCV > 2.4.3 not found.")
endif()
endif()
add_executable(${PROJECT_NAME} "main.cpp")
target_link_libraries(${PROJECT_NAME} ${TORCH_LIBRARIES} ${OpenCV_LIBS})
set_property(TARGET ${PROJECT_NAME} PROPERTY CXX_STANDARD 11)
Configure之后可以点击Generate,然后再文件夹里面双击.sin文件就可以打开工程。
libtorch error C2440: “初始化”: 无法从“torch::jit::script::Module”转换…的问题
C++部署Pytorch(Libtorch)出现问题、错误汇总
至此,应该是可以正确运行了。