| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277 |
- cmake_minimum_required(VERSION 3.14)
- project(ppocr CXX)
- set(DEMO_NAME "ppocr")
- set(CMAKE_CXX_STANDARD 11)
- set(CMAKE_CXX_STANDARD_REQUIRED True)
- option(WITH_MKL "Compile demo with MKL/OpenBlas support, default use MKL." ON)
- option(WITH_GPU "Compile demo with GPU/CPU, default use CPU." OFF)
- option(WITH_STATIC_LIB "Compile demo with static/shared library, default use static." ON)
- option(USE_FREETYPE "Enable FreeType support" OFF)
- SET(PADDLE_LIB "" CACHE PATH "Location of libraries")
- SET(OPENCV_DIR "" CACHE PATH "Location of libraries")
- SET(CUDA_LIB "" CACHE PATH "Location of libraries")
- SET(CUDNN_LIB "" CACHE PATH "Location of libraries")
- macro(safe_set_static_flag)
- foreach(flag_var
- CMAKE_CXX_FLAGS CMAKE_CXX_FLAGS_DEBUG CMAKE_CXX_FLAGS_RELEASE
- CMAKE_CXX_FLAGS_MINSIZEREL CMAKE_CXX_FLAGS_RELWITHDEBINFO)
- if(${${flag_var}} MATCHES "/MD")
- string(REGEX REPLACE "/MD" "/MT" ${flag_var} "${${flag_var}}")
- endif()
- endforeach(flag_var)
- endmacro()
- if (WITH_MKL)
- ADD_DEFINITIONS(-DUSE_MKL)
- endif()
- if(NOT DEFINED PADDLE_LIB)
- message(FATAL_ERROR "please set PADDLE_LIB with -DPADDLE_LIB=/path/paddle/lib")
- endif()
- if(NOT DEFINED OPENCV_DIR)
- message(FATAL_ERROR "please set OPENCV_DIR with -DOPENCV_DIR=/path/opencv")
- endif()
- if (WIN32)
- include_directories("${PADDLE_LIB}/paddle/include")
- link_directories("${PADDLE_LIB}/paddle/lib")
- set(CMAKE_CONFIGURATION_TYPES "Debug;Release" CACHE STRING "" FORCE)
- set(OpenCV_DIR "${OPENCV_DIR}/x64/vc16/lib")
- find_package(OpenCV REQUIRED )
- if(USE_FREETYPE)
- if(NOT "opencv_freetype" IN_LIST OpenCV_LIBS)
- message(FATAL_ERROR "OpenCV was not compiled with the freetype module (opencv_freetype) !")
- endif()
- add_definitions(-DUSE_FREETYPE)
- endif()
- else ()
- set(OpenCV_DIR "${OPENCV_DIR}/lib64/cmake/opencv4")
- find_package(OpenCV REQUIRED)
- if(USE_FREETYPE)
- if(NOT "opencv_freetype" IN_LIST OpenCV_LIBS)
- message(FATAL_ERROR "OpenCV was not compiled with the freetype module (opencv_freetype) !")
- endif()
- add_definitions(-DUSE_FREETYPE)
- endif()
- include_directories("${PADDLE_LIB}/paddle/include")
- link_directories("${PADDLE_LIB}/paddle/lib")
- endif ()
- include_directories(${OpenCV_INCLUDE_DIRS})
- if (WIN32)
- add_definitions("/DGOOGLE_GLOG_DLL_DECL=")
- if(WITH_MKL)
- set(FLAG_OPENMP "/openmp")
- endif()
- set(CMAKE_C_FLAGS_DEBUG "${CMAKE_C_FLAGS_DEBUG} /bigobj /MTd ${FLAG_OPENMP}")
- set(CMAKE_C_FLAGS_RELEASE "${CMAKE_C_FLAGS_RELEASE} /bigobj /MT ${FLAG_OPENMP}")
- set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} /bigobj /MTd ${FLAG_OPENMP}")
- set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /bigobj /MT ${FLAG_OPENMP}")
- if (WITH_STATIC_LIB)
- safe_set_static_flag()
- add_definitions(-DSTATIC_LIB)
- add_definitions(-DYAML_CPP_STATIC_DEFINE)
- endif()
- message("cmake c debug flags " ${CMAKE_C_FLAGS_DEBUG})
- message("cmake c release flags " ${CMAKE_C_FLAGS_RELEASE})
- message("cmake cxx debug flags " ${CMAKE_CXX_FLAGS_DEBUG})
- message("cmake cxx release flags " ${CMAKE_CXX_FLAGS_RELEASE})
- else()
- if(WITH_MKL)
- set(FLAG_OPENMP "-fopenmp")
- endif()
- set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -O3 ${FLAG_OPENMP} -std=c++11")
- set(CMAKE_STATIC_LIBRARY_PREFIX "")
- message("cmake cxx flags" ${CMAKE_CXX_FLAGS})
- endif()
- if (WITH_GPU)
- if (NOT DEFINED CUDA_LIB OR ${CUDA_LIB} STREQUAL "")
- message(FATAL_ERROR "please set CUDA_LIB with -DCUDA_LIB=/path/cuda-8.0/lib64")
- endif()
- if (NOT WIN32)
- if (NOT DEFINED CUDNN_LIB)
- message(FATAL_ERROR "please set CUDNN_LIB with -DCUDNN_LIB=/path/cudnn_v7.4/cuda/lib64")
- endif()
- add_definitions(-DWITH_GPU)
- endif(NOT WIN32)
- endif()
- include_directories("${PADDLE_LIB}/third_party/install/protobuf/include")
- include_directories("${PADDLE_LIB}/third_party/install/glog/include")
- include_directories("${PADDLE_LIB}/third_party/install/gflags/include")
- include_directories("${PADDLE_LIB}/third_party/install/xxhash/include")
- include_directories("${PADDLE_LIB}/third_party/install/zlib/include")
- include_directories("${PADDLE_LIB}/third_party/install/onnxruntime/include")
- include_directories("${PADDLE_LIB}/third_party/install/paddle2onnx/include")
- include_directories("${PADDLE_LIB}/third_party/install/yaml-cpp/include")
- include_directories("${PADDLE_LIB}/third_party/install/openvino/include")
- include_directories("${PADDLE_LIB}/third_party/install/tbb/include")
- include_directories("${PADDLE_LIB}/third_party/boost")
- include_directories("${PADDLE_LIB}/third_party/eigen3")
- include_directories("${PADDLE_LIB}/paddle/include/")
- include_directories("${CMAKE_SOURCE_DIR}/")
- link_directories("${PADDLE_LIB}/third_party/install/zlib/lib")
- link_directories("${PADDLE_LIB}/third_party/install/protobuf/lib")
- link_directories("${PADDLE_LIB}/third_party/install/glog/lib")
- link_directories("${PADDLE_LIB}/third_party/install/gflags/lib")
- link_directories("${PADDLE_LIB}/third_party/install/xxhash/lib")
- link_directories("${PADDLE_LIB}/third_party/install/onnxruntime/lib")
- link_directories("${PADDLE_LIB}/third_party/install/paddle2onnx/lib")
- link_directories("${PADDLE_LIB}/third_party/install/yaml-cpp/lib")
- link_directories("${PADDLE_LIB}/third_party/install/openvino/intel64")
- link_directories("${PADDLE_LIB}/third_party/install/tbb/lib")
- link_directories("${PADDLE_LIB}/paddle/lib")
- if(WITH_MKL)
- include_directories("${PADDLE_LIB}/third_party/install/mklml/include")
- if (WIN32)
- set(MATH_LIB ${PADDLE_LIB}/third_party/install/mklml/lib/mklml.lib
- ${PADDLE_LIB}/third_party/install/mklml/lib/libiomp5md.lib)
- else ()
- set(MATH_LIB ${PADDLE_LIB}/third_party/install/mklml/lib/libmklml_intel${CMAKE_SHARED_LIBRARY_SUFFIX}
- ${PADDLE_LIB}/third_party/install/mklml/lib/libiomp5${CMAKE_SHARED_LIBRARY_SUFFIX})
- execute_process(COMMAND cp -r ${PADDLE_LIB}/third_party/install/mklml/lib/libmklml_intel${CMAKE_SHARED_LIBRARY_SUFFIX} /usr/lib)
- endif ()
- set(MKLDNN_PATH "${PADDLE_LIB}/third_party/install/onednn")
- if(EXISTS ${MKLDNN_PATH})
- include_directories("${MKLDNN_PATH}/include")
- if (WIN32)
- set(MKLDNN_LIB ${MKLDNN_PATH}/lib/mkldnn.lib)
- else ()
- set(MKLDNN_LIB ${MKLDNN_PATH}/lib/libdnnl.so.3)
- endif ()
- endif()
- else()
- if (WIN32)
- set(MATH_LIB ${PADDLE_LIB}/third_party/install/openblas/lib/openblas${CMAKE_STATIC_LIBRARY_SUFFIX})
- else ()
- set(MATH_LIB ${PADDLE_LIB}/third_party/install/openblas/lib/libopenblas${CMAKE_STATIC_LIBRARY_SUFFIX})
- endif ()
- endif()
- # Note: libpaddle_inference_api.so/a must put before libpaddle_inference.so/a
- if(WITH_STATIC_LIB)
- if(WIN32)
- set(DEPS
- ${PADDLE_LIB}/paddle/lib/paddle_inference${CMAKE_STATIC_LIBRARY_SUFFIX})
- else()
- set(DEPS
- ${PADDLE_LIB}/paddle/lib/libpaddle_inference${CMAKE_STATIC_LIBRARY_SUFFIX})
- endif()
- else()
- if(WIN32)
- set(DEPS
- ${PADDLE_LIB}/paddle/lib/paddle_inference${CMAKE_SHARED_LIBRARY_SUFFIX})
- else()
- set(DEPS
- ${PADDLE_LIB}/paddle/lib/libpaddle_inference${CMAKE_SHARED_LIBRARY_SUFFIX})
- endif()
- endif(WITH_STATIC_LIB)
- if (NOT WIN32)
- set(DEPS ${DEPS}
- ${MATH_LIB} ${MKLDNN_LIB}
- glog gflags protobuf z xxhash
- )
- if(EXISTS "${PADDLE_LIB}/third_party/install/snappystream/lib")
- set(DEPS ${DEPS} snappystream)
- endif()
- if (EXISTS "${PADDLE_LIB}/third_party/install/snappy/lib")
- set(DEPS ${DEPS} snappy)
- endif()
- else()
- set(DEPS ${DEPS}
- ${MATH_LIB} ${MKLDNN_LIB}
- glog gflags_static libprotobuf xxhash)
- set(DEPS ${DEPS} libcmt shlwapi)
- if (EXISTS "${PADDLE_LIB}/third_party/install/snappy/lib")
- set(DEPS ${DEPS} snappy)
- endif()
- if(EXISTS "${PADDLE_LIB}/third_party/install/snappystream/lib")
- set(DEPS ${DEPS} snappystream)
- endif()
- endif(NOT WIN32)
- if (EXISTS "${PADDLE_LIB}/third_party/install/yaml-cpp/lib")
- set(DEPS ${DEPS} yaml-cpp)
- endif()
- if(WITH_GPU)
- if(NOT WIN32)
- set(DEPS ${DEPS} ${CUDA_LIB}/libcudart${CMAKE_SHARED_LIBRARY_SUFFIX})
- set(DEPS ${DEPS} ${CUDNN_LIB}/libcudnn${CMAKE_SHARED_LIBRARY_SUFFIX})
- else()
- set(DEPS ${DEPS} ${CUDA_LIB}/cudart${CMAKE_STATIC_LIBRARY_SUFFIX} )
- message($DEPS)
- set(DEPS ${DEPS} ${CUDA_LIB}/cublas${CMAKE_STATIC_LIBRARY_SUFFIX} )
- set(DEPS ${DEPS} ${CUDNN_LIB}/cudnn${CMAKE_STATIC_LIBRARY_SUFFIX})
- endif()
- endif()
- if (NOT WIN32)
- set(EXTERNAL_LIB "-ldl -lrt -lgomp -lz -lm -lpthread")
- set(DEPS ${DEPS} ${EXTERNAL_LIB})
- endif()
- set(THIRD_PARTY_PATH ${CMAKE_CURRENT_LIST_DIR}/third_party)
- function(download_and_decompress url filename decompress_dir)
- if(NOT EXISTS "${filename}" AND NOT EXISTS "${decompress_dir}")
- message("Downloading file from ${url} to ${filename} ...")
- file(DOWNLOAD ${url} "${filename}.tmp" SHOW_PROGRESS)
- file(RENAME "${filename}.tmp" ${filename})
- endif()
- if(NOT EXISTS ${decompress_dir})
- file(MAKE_DIRECTORY ${decompress_dir})
- message("Decompress file ${filename} ...")
- execute_process(COMMAND ${CMAKE_COMMAND} -E tar -xf ${filename} WORKING_DIRECTORY ${decompress_dir})
- endif()
- endfunction()
- set(PACKAGE_LIST abseil-cpp clipper_ver6.4.2 nlohmann)
- foreach(PKG ${PACKAGE_LIST})
- set(PKG_URL "https://paddle-model-ecology.bj.bcebos.com/paddlex/cpp/libs/${PKG}.tgz")
- set(PKG_TGZ_PATH "${CMAKE_CURRENT_BINARY_DIR}/${PKG}.tgz")
- set(PKG_DST_PATH "${THIRD_PARTY_PATH}/${PKG}")
- download_and_decompress(${PKG_URL} ${PKG_TGZ_PATH} ${PKG_DST_PATH})
- endforeach()
- add_subdirectory(third_party/abseil-cpp)
- add_subdirectory(third_party/clipper_ver6.4.2/cpp)
- include_directories(${POLYCLIPPING_INCLUDE_DIR})
- set(DEPS ${DEPS} ${OpenCV_LIBS})
- set(DEPS ${DEPS} absl::statusor)
- set(DEPS ${DEPS} polyclipping)
- if(UNIX)
- find_package(Iconv REQUIRED)
- endif()
- file(GLOB_RECURSE SRC_LIST "./src/*.cc")
- set(SRCS cli.cc )
- add_executable(${DEMO_NAME} ${SRCS} ${SRC_LIST} )
- target_link_libraries(${DEMO_NAME} ${DEPS} )
- if (WIN32 AND WITH_MKL)
- add_custom_command(TARGET ${DEMO_NAME} POST_BUILD
- COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_LIB}/third_party/install/mklml/lib/mklml.dll ./mklml.dll
- COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_LIB}/third_party/install/mklml/lib/libiomp5md.dll ./libiomp5md.dll
- COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_LIB}/third_party/install/onednn/lib/mkldnn.dll ./mkldnn.dll
- COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_LIB}/third_party/install/mklml/lib/mklml.dll ./release/mklml.dll
- COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_LIB}/third_party/install/mklml/lib/libiomp5md.dll ./release/libiomp5md.dll
- COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_LIB}/third_party/install/onednn/lib/mkldnn.dll ./release/mkldnn.dll
- )
- endif()
|