CMakeLists.txt

cmake_minimum_required(VERSION 3.16)
project(resnet)

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

# find_package(OpenCV REQUIRED)
# 头文件
set(INCLUDE_PATH    
    ${CMAKE_CURRENT_SOURCE_DIR}/include
    /usr/local/include/opencv4
    ${ROCM_PATH}/onnxruntime/include)
include_directories(${INCLUDE_PATH})

# 库文件
set(LIB_PATH
    ${ROCM_PATH}/onnxruntime/lib
    /usr/local/lib    
    $ENV{DTKROOT}/lib/
)
link_directories(${LIB_PATH})

# 库文件链接
set(LIB onnxruntime opencv_core opencv_imgproc opencv_highgui opencv_imgcodecs)
link_libraries(${LIB})

set(source_cpp  ${CMAKE_CURRENT_SOURCE_DIR}/src/resnet50.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/src/Helpers.cpp)

add_executable(resnet  ${source_cpp})