27 lines
911 B
CMake
27 lines
911 B
CMake
|
# Copyright (c) OpenMMLab. All rights reserved.
|
||
|
|
|
||
|
|
cmake_minimum_required(VERSION 3.14)
|
||
|
|
project(mmdeploy_mmocr_cuda_impl)
|
||
|
|
|
||
|
|
if ("cuda" IN_LIST MMDEPLOY_TARGET_DEVICES)
|
||
|
|
include(${CMAKE_SOURCE_DIR}/cmake/MMDeploy.cmake)
|
||
|
|
include(${CMAKE_SOURCE_DIR}/cmake/cuda.cmake NO_POLICY_SCOPE)
|
||
|
|
enable_language(CUDA)
|
||
|
|
add_library(${PROJECT_NAME} OBJECT
|
||
|
|
connected_component.cu
|
||
|
|
utils.cu
|
||
|
|
dbnet.cpp
|
||
|
|
panet.cpp
|
||
|
|
psenet.cpp)
|
||
|
|
set_target_properties(${PROJECT_NAME} PROPERTIES POSITION_INDEPENDENT_CODE 1)
|
||
|
|
target_include_directories(${PROJECT_NAME} PRIVATE
|
||
|
|
${CUDA_INCLUDE_DIRS})
|
||
|
|
target_link_libraries(${PROJECT_NAME} PRIVATE
|
||
|
|
mmdeploy_opencv_utils
|
||
|
|
mmdeploy::core
|
||
|
|
${CUDA_LIBRARIES})
|
||
|
|
target_link_libraries(mmdeploy_mmocr PRIVATE
|
||
|
|
${PROJECT_NAME})
|
||
|
|
mmdeploy_export(${PROJECT_NAME})
|
||
|
|
endif ()
|