CMakeLists.txt 1.04 KB
Newer Older
lijian6's avatar
lijian6 committed
1
2
3
4
5
6
7
8
9
10
cmake_minimum_required(VERSION 3.16)
project(Video_Onnx)

set(CMAKE_CXX_STANDARD 11)
set(CMAKE_CXX_COMPILER g++)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++17 -pthread")
add_definitions(-DUNICODE -D_UNICODE)
add_definitions("-Wall")

# project include
lijian6's avatar
lijian6 committed
11
set(INCLUDE_PATH    /usr/local/include/opencv4
lijian6's avatar
lijian6 committed
12
13
14
15
16
17
18
19
20
21
22
                    ${CMAKE_CURRENT_SOURCE_DIR}/include
                    $ENV{ROCM_PATH}/include)
include_directories(${INCLUDE_PATH})

# project lib path
set(LIBRARY_PATH    /usr/local/lib
                    ${CMAKE_CURRENT_SOURCE_DIR}/lib
                    $ENV{ROCM_PATH}/lib)
link_directories(${LIBRARY_PATH})

# project lib
lijian6's avatar
lijian6 committed
23
set(LIBRARY     avcodec avdevice avfilter avformat avutil swresample swscale
lijian6's avatar
lijian6 committed
24
25
26
27
28
29
30
31
32
33
                opencv_core opencv_imgproc opencv_imgcodecs opencv_dnn opencv_highgui opencv_videoio
                onnxruntime)
link_libraries(${LIBRARY})

# project source files
file(GLOB _SRC ${CMAKE_CURRENT_SOURCE_DIR}/src/*.cpp)
set(_COMPILE_CODE ${_SRC})

add_executable(Video_Onnx ${_COMPILE_CODE})
target_link_libraries(Video_Onnx decode Queue)