cmake_minimum_required(VERSION 3.16)
project(YoloV5s_ort)

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")

# project include
set(INCLUDE_PATH    /usr/local/include
	            /usr/local/include/opencv4/
                    ${CMAKE_CURRENT_SOURCE_DIR}/include
                    $ENV{ROCM_PATH}/include
		    $ENV{ROCM_PATH}/onnxruntime/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
set(LIBRARY     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(yolov5 ${_COMPILE_CODE})