C++에서 Torchvision 사용

안녕하세요. 저는 MaskRCNN을 C언어로 구현하기 프로그램 개발중에 있습니다.

기존 파이썬으로 Pretrained된 모델을 C언어에서 사용하기 위해 개발 중 Torchvision의 Header file을 include 해야하는 것을 알게 되었습니다.

#include <torchvision/vision.h>

find_package(TorchVision REQUIRED)
target_link_libraries(my-target PUBLIC TorchVision::TorchVision)

Torchvision cpp를 컴파일이 정상적으로 생성되어 구동테스트를 하려 하였으나, Cmake Configuration 단계에서 다음과 같은 에러가 발생하여 도움을 요청드립니다.

  1. CMakeList 내역
    cmake_minimum_required(VERSION 3.10)
    project(hello-world)
    find_package(TorchVision REQUIRED)
    find_package(Python3 COMPONENTS Development)
    add_executable(hello-world main.cpp)
    target_compile_features(hello-world PUBLIC cxx_range_for)
    target_link_libraries(hello-world TorchVision::TorchVision)
    set_property(TARGET hello-world PROPERTY CXX_STANDARD 14)

  2. 에러내역
    [CMake] CMake Error at D:/vision-mainr1/build/TorchVisionConfig.cmake:50 (include):
    1> [CMake] include could not find requested file:
    1> [CMake]
    1> [CMake] D:/vision-mainr1/build/TorchVisionTargets.cmake
    1> [CMake] Call Stack (most recent call first):
    1> [CMake] CMakeLists.txt:7 (find_package)
    1> [CMake] CMake Error at D:/vision-mainr1/build/TorchVisionConfig.cmake:52 (target_include_directories):
    1> [CMake] Cannot specify include directories for target “TorchVision::TorchVision”
    1> [CMake] which is not built by this project.
    1> [CMake] Call Stack (most recent call first):
    1> [CMake] CMakeLists.txt:7 (find_package)
    1> [CMake] CMake Error at D:/vision-mainr1/build/TorchVisionConfig.cmake:55 (target_compile_definitions):
    1> [CMake] Cannot specify compile definitions for target “TorchVision::TorchVision”
    1> [CMake] which is not built by this project.
    1> [CMake] Call Stack (most recent call first):
    1> [CMake] CMakeLists.txt:7 (find_package)
    1> [CMake] CMake Error at D:/vision-mainr1/build/TorchVisionConfig.cmake:59 (target_link_libraries):
    1> [CMake] Cannot specify link libraries for target “TorchVision::TorchVision” which
    1> [CMake] is not built by this project.
    1> [CMake] Call Stack (most recent call first):
    1> [CMake] CMakeLists.txt:7 (find_package)
    1> [CMake] – Configuring incomplete, errors occurred!

:grinning:… 해결하였습니다.

1개의 좋아요

이 글은 마지막 댓글이 달린지 오래(30일)되어 자동으로 닫혔습니다. 댓글 대신 새로운 글을 작성해주세요! :slight_smile: