mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
5e51739485
* add SDK support for PANet * fix panet * fix panet * simplify panet * add PSENet support * fix-psenet * add CUDA post-processing for DBNet * fix dbnet * fix dbnet * add cpu support for PANet * fix panet * add CUDA support for PANet * fix panet * format * add cpu impls for PSENet * fix psenet * add cuda impl for PSENet * fix psenet * add param parsing * simplify impls * simplify impls * clean-up * fix lint * fix cuda-10 build * fix cuda-10 build
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 ()
|