CMakeLists.txt 1.32 KB
Newer Older
yangql's avatar
yangql committed
1
2
3
4
5
6
7
8
9
10
11
# 设置cmake的最低版本
cmake_minimum_required(VERSION 3.5)

# 设置项目名
project(ResNet50)

# 设置编译器
set(CMAKE_CXX_COMPILER g++)
set(CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS} -std=c++17) # 2.2版本以上需要c++17
set(CMAKE_BUILD_TYPE release)

yangql's avatar
yangql committed
12
13
# 配置OpenCV环境
find_package(OpenCV REQUIRED)
yangql's avatar
yangql committed
14
15
16
17
# 添加头文件路径
set(INCLUDE_PATH    ${CMAKE_CURRENT_SOURCE_DIR}/Src/
                    ${CMAKE_CURRENT_SOURCE_DIR}/Src/Utility/ 
                    $ENV{DTKROOT}/include/
18
                    $ENV{DTKROOT}/onnxruntime/include/
yangql's avatar
yangql committed
19
                    ${OpenCV_INCLUDE_DIRS}
yangql's avatar
yangql committed
20
21
22
23
24
25
26
27
28
                    ${CMAKE_CURRENT_SOURCE_DIR}/depend/include/)
include_directories(${INCLUDE_PATH})

# 添加依赖库路径
set(LIBRARY_PATH ${CMAKE_CURRENT_SOURCE_DIR}/depend/lib64/
                $ENV{DTKROOT}/lib/)
link_directories(${LIBRARY_PATH})

# 添加依赖库
yangql's avatar
yangql committed
29
set(LIBRARY 
yangql's avatar
yangql committed
30
31
32
33
34
35
36
37
38
39
40
41
            onnxruntime
            )
link_libraries(${LIBRARY})

# 添加源文件
set(SOURCE_FILES ${CMAKE_CURRENT_SOURCE_DIR}/Src/main.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/Src/Classifier.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/Src/Utility/CommonUtility.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/Src/Utility/Filesystem.cpp)

# 添加可执行目标
add_executable(ResNet50 ${SOURCE_FILES})
yangql's avatar
yangql committed
42
43

target_link_libraries(ResNet50 ${OpenCV_LIBS} onnxruntime)