-
Notifications
You must be signed in to change notification settings - Fork 0
/
CMakeLists.txt
89 lines (67 loc) · 2.35 KB
/
CMakeLists.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
# CMakeLists for libfacedetectcnn
project(facedetection)
cmake_minimum_required(VERSION 2.8)
option(ENABLE_AVX2 "use avx2" OFF)
option(ENABLE_NEON "whether use neon, if use arm please set it on" OFF)
option(DEMO "build the demo" OFF)
SET(fdt_base_dir ${PROJECT_SOURCE_DIR})
SET(fdt_src_dir ${fdt_base_dir}/src)
SET(fdt_inc_dir ${fdt_base_dir}/src)
SET(fdt_lib_name facedetection)
SET(fdt_lib_static ${fdt_lib_name})
SET(fdt_lib_shared ${fdt_lib_name}_shared)
FILE(GLOB_RECURSE fdt_source_files ${fdt_src_dir}/*.cpp)
LIST(SORT fdt_source_files)
if(ENABLE_AVX2)
message("using avx2")
add_definitions(-D_ENABLE_AVX2)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -mavx2 -mfma")
endif()
if(ENABLE_NEON)
message("using arm")
add_definitions(-D_ENABLE_NEON)
endif()
set(CMAKE_CXX_STANDARD 11)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CXX_EXTENSIONS OFF)
INCLUDE_DIRECTORIES(${fdt_inc_dir})
# Create a static library (.a)
ADD_LIBRARY(${fdt_lib_static} STATIC ${fdt_source_files})
# Create a shared library (.so)
ADD_LIBRARY(${fdt_lib_shared} SHARED ${fdt_source_files})
SET_TARGET_PROPERTIES(${fdt_lib_shared} PROPERTIES OUTPUT_NAME "${fdt_lib_name}")
SET_TARGET_PROPERTIES(${fdt_lib_shared} PROPERTIES PREFIX "lib")
# Create demo. OpenCV is requred.
if (DEMO)
#if(WIN32)
# set(OpenCV_DIR "D:/opencv343/build") # TODO
#endif()
find_package(OpenCV REQUIRED)
include_directories(${OpenCV_INCLUDE_DIRS})
set(fdt_demo_files ${fdt_base_dir}/example/libfacedetectcnn-example.cpp)
add_executable(fdt_demo ${fdt_demo_files})
target_link_libraries(fdt_demo ${fdt_lib_static} ${OpenCV_LIBS})
endif()
if (GSTREAMER)
find_package(OpenCV REQUIRED)
include(FindPkgConfig)
pkg_search_module(GSTREAMER REQUIRED gstreamer-1.0)
pkg_search_module(GSTREAMER_BASE REQUIRED gstreamer-base-1.0)
pkg_search_module(GSTREAMER_VIDEO REQUIRED gstreamer-video-1.0)
add_library(gstfacedetect SHARED
example/libfacedetect.cpp
)
include_directories(gstfacedetect PRIVATE
${GSTREAMER_INCLUDE_DIRS}
${GSTREAMER_BASE_INCLUDE_DIRS}
${GSTREAMER_VIDEO_INCLUDE_DIRS}
${OpenCV_INCLUDE_DIRS}
)
target_link_libraries(gstfacedetect
${GSTREAMER_LIBRARIES}
${GSTREAMER_BASE_LIBRARIES}
${GSTREAMER_VIDEO_LIBRARIES}
${OpenCV_LIBS}
${fdt_lib_shared}
)
endif()