没有合适的资源?快使用搜索试试~ 我知道了~
cmake file for demo
需积分: 5 0 下载量 85 浏览量
2020-03-04
14:27:54
上传
评论
收藏 1KB TXT 举报
温馨提示
试读
2页
demo cpp inference using libtorch for realtime semantic segmentation
资源推荐
资源详情
资源评论
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(inference)
# Application required
find_package(OpenCV 3.4.7 COMPONENTS photo core highgui imgproc cudaimgproc cudawarping cudaarithm cudafilters)
if(NOT OpenCV_FOUND)
message(WARNING "The predict program will not be built")
endif()
# check opencv since we may need it
if (OpenCV_FOUND)
find_package(CUDA REQUIRED)
include_directories(SYSTEM ${OpenCV_INCLUDE_DIRS})
include_directories(SYSTEM ${CUDA_TOOLKIT_INCLUDE})
link_directories(${CUDA_TOOLKIT_ROOT_DIR}/lib64)
set ( OpenCV_DIR CACHE PATH ${OpenCV_CONFIG_PATH})
endif() # OpenCV_FOUND
find_package(Torch REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
add_executable(demo demo.cpp)
target_link_libraries(demo "${TORCH_LIBRARIES}" ${OpenCV_LIBS})
set_property(TARGET demo PROPERTY CXX_STANDARD 14)
# The following code block is suggested to be used on Windows.
project(inference)
# Application required
find_package(OpenCV 3.4.7 COMPONENTS photo core highgui imgproc cudaimgproc cudawarping cudaarithm cudafilters)
if(NOT OpenCV_FOUND)
message(WARNING "The predict program will not be built")
endif()
# check opencv since we may need it
if (OpenCV_FOUND)
find_package(CUDA REQUIRED)
include_directories(SYSTEM ${OpenCV_INCLUDE_DIRS})
include_directories(SYSTEM ${CUDA_TOOLKIT_INCLUDE})
link_directories(${CUDA_TOOLKIT_ROOT_DIR}/lib64)
set ( OpenCV_DIR CACHE PATH ${OpenCV_CONFIG_PATH})
endif() # OpenCV_FOUND
find_package(Torch REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
add_executable(demo demo.cpp)
target_link_libraries(demo "${TORCH_LIBRARIES}" ${OpenCV_LIBS})
set_property(TARGET demo PROPERTY CXX_STANDARD 14)
# The following code block is suggested to be used on Windows.
资源评论
RudyCHiP
- 粉丝: 0
- 资源: 2
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功