2021-12-07 10:57:55 +08:00
|
|
|
# Copyright (c) OpenMMLab. All rights reserved.
|
|
|
|
cmake_minimum_required(VERSION 3.14)
|
|
|
|
project(model)
|
|
|
|
|
2022-02-24 20:08:44 +08:00
|
|
|
include(${CMAKE_SOURCE_DIR}/cmake/MMDeploy.cmake)
|
2021-12-07 10:57:55 +08:00
|
|
|
|
2021-12-21 21:32:39 +08:00
|
|
|
set(MODEL_NAMES "directory_model")
|
2021-12-07 10:57:55 +08:00
|
|
|
if (${MMDEPLOY_ZIP_MODEL})
|
2021-12-21 21:32:39 +08:00
|
|
|
set(MODEL_NAMES ${MODEL_NAMES} "zip_model")
|
2021-12-07 10:57:55 +08:00
|
|
|
endif ()
|
|
|
|
|
2021-12-21 21:32:39 +08:00
|
|
|
foreach (MODEL_NAME ${MODEL_NAMES})
|
|
|
|
set(TARGET_MODEL_NAME mmdeploy_${MODEL_NAME})
|
2022-02-24 20:08:44 +08:00
|
|
|
mmdeploy_add_module(${TARGET_MODEL_NAME} ${MODEL_NAME}_impl.cpp)
|
2022-03-25 23:28:16 +08:00
|
|
|
if (NOT (MSVC OR ANDROID))
|
2022-02-24 20:08:44 +08:00
|
|
|
target_link_libraries(${TARGET_MODEL_NAME} PUBLIC stdc++fs)
|
|
|
|
endif ()
|
2021-12-21 21:32:39 +08:00
|
|
|
if (${MODEL_NAME} STREQUAL "zip_model")
|
2021-12-07 10:57:55 +08:00
|
|
|
find_package(libzip QUIET)
|
|
|
|
if (libzip_FOUND)
|
2021-12-21 21:32:39 +08:00
|
|
|
target_link_libraries(${TARGET_MODEL_NAME} PUBLIC libzip::zip)
|
2022-02-24 20:08:44 +08:00
|
|
|
else ()
|
2021-12-21 21:32:39 +08:00
|
|
|
target_link_libraries(${TARGET_MODEL_NAME} PUBLIC zip)
|
2022-02-24 20:08:44 +08:00
|
|
|
endif ()
|
2021-12-07 10:57:55 +08:00
|
|
|
endif ()
|
2021-12-21 21:32:39 +08:00
|
|
|
add_library(mmdeploy::${MODEL_NAME} ALIAS ${TARGET_MODEL_NAME})
|
2021-12-07 10:57:55 +08:00
|
|
|
endforeach ()
|