|
@@ -17,16 +17,6 @@ link_directories("${CMAKE_CURRENT_BINARY_DIR}")
|
|
|
include_directories("${CMAKE_CURRENT_BINARY_DIR}/ext/yaml-cpp/src/ext-yaml-cpp/include")
|
|
include_directories("${CMAKE_CURRENT_BINARY_DIR}/ext/yaml-cpp/src/ext-yaml-cpp/include")
|
|
|
link_directories("${CMAKE_CURRENT_BINARY_DIR}/ext/yaml-cpp/lib")
|
|
link_directories("${CMAKE_CURRENT_BINARY_DIR}/ext/yaml-cpp/lib")
|
|
|
|
|
|
|
|
-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(${flag_var} MATCHES "/MD")
|
|
|
|
|
- endforeach(flag_var)
|
|
|
|
|
-endmacro()
|
|
|
|
|
-
|
|
|
|
|
if (NOT DEFINED LITE_DIR OR ${LITE_DIR} STREQUAL "")
|
|
if (NOT DEFINED LITE_DIR OR ${LITE_DIR} STREQUAL "")
|
|
|
message(FATAL_ERROR "please set LITE_DIR with -LITE_DIR=/path/influence_engine")
|
|
message(FATAL_ERROR "please set LITE_DIR with -LITE_DIR=/path/influence_engine")
|
|
|
endif()
|
|
endif()
|
|
@@ -40,7 +30,7 @@ if (NOT DEFINED GFLAGS_DIR OR ${GFLAGS_DIR} STREQUAL "")
|
|
|
endif()
|
|
endif()
|
|
|
|
|
|
|
|
if (NOT DEFINED ARCH OR ${ARCH} STREQUAL "")
|
|
if (NOT DEFINED ARCH OR ${ARCH} STREQUAL "")
|
|
|
- message(FATAL_ERROR "please set ARCH with ARCH=armv7 or ARCH=armv8")
|
|
|
|
|
|
|
+ message(FATAL_ERROR "please set ARCH with ARCH=armv7-a or ARCH=armv8-a")
|
|
|
endif()
|
|
endif()
|
|
|
|
|
|
|
|
if (NOT DEFINED LITE OR ${LITE} STREQUAL "")
|
|
if (NOT DEFINED LITE OR ${LITE} STREQUAL "")
|
|
@@ -60,40 +50,30 @@ include_directories("${GFLAGS_DIR}/include")
|
|
|
find_package(OpenCV REQUIRED PATHS ${OPENCV_DIR}/share/OpenCV NO_DEFAULT_PATH)
|
|
find_package(OpenCV REQUIRED PATHS ${OPENCV_DIR}/share/OpenCV NO_DEFAULT_PATH)
|
|
|
include_directories(${OpenCV_INCLUDE_DIRS})
|
|
include_directories(${OpenCV_INCLUDE_DIRS})
|
|
|
|
|
|
|
|
-if (${ARCH} STREQUAL "armv7")
|
|
|
|
|
- set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -march=armv7-a -mfloat-abi=hard -mfpu=neon-vfpv4 -g -o2 -fopenmp -std=c++11")
|
|
|
|
|
|
|
+if (${ARCH} STREQUAL "armv7-a")
|
|
|
|
|
+ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -mfloat-abi=hard -mfpu=neon-vfpv4 -g -o2 -fopenmp -std=c++11")
|
|
|
set(CMAKE_STATIC_LIBRARY_PREFIX "")
|
|
set(CMAKE_STATIC_LIBRARY_PREFIX "")
|
|
|
-elseif(${ARC}H STREQUAL "armv8")
|
|
|
|
|
- set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -march=armb8-a -g -o2 -fopenmp -std=c++11")
|
|
|
|
|
|
|
+elseif(${ARC}H STREQUAL "armv8-a")
|
|
|
|
|
+ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -o2 -fopenmp -std=c++11")
|
|
|
set(CMAKE_STATIC_LIBRARY_PREFIX "")
|
|
set(CMAKE_STATIC_LIBRARY_PREFIX "")
|
|
|
else()
|
|
else()
|
|
|
message(FATAL_ERROR "Unknown arch abi ${ARCH}, only support armv8 and armv7")
|
|
message(FATAL_ERROR "Unknown arch abi ${ARCH}, only support armv8 and armv7")
|
|
|
endif()
|
|
endif()
|
|
|
|
|
|
|
|
|
|
|
|
|
-if(WITH_STATIC_LIB)
|
|
|
|
|
- if(${LITE} STREQUAL "full")
|
|
|
|
|
- set(DEPS ${LITE_DIR}/lib/libpaddle_full_api_shared${CMAKE_STATIC_LIBRARY_SUFFIX})
|
|
|
|
|
- elseif(${LITE} STREQUAL "light")
|
|
|
|
|
- set(DEPS ${LITE_DIR}/lib/libpaddle_light_api_shared${CMAKE_STATIC_LIBRARY_SUFFIX})
|
|
|
|
|
- else()
|
|
|
|
|
- message(FATAL_ERROR "Unkown LITE release ${LITE}, only support light and full")
|
|
|
|
|
- endif()
|
|
|
|
|
|
|
+if(${LITE} STREQUAL "full")
|
|
|
|
|
+ set(DEPS ${LITE_DIR}/lib/libpaddle_full_api_shared${CMAKE_SHARED_LIBRARY_SUFFIX})
|
|
|
|
|
+elseif(${LITE} STREQUAL "light")
|
|
|
|
|
+ set(DEPS ${LITE_DIR}/lib/libpaddle_light_api_shared${CMAKE_SHARED_LIBRARY_SUFFIX})
|
|
|
else()
|
|
else()
|
|
|
- if(${LITE} STREQUAL "full")
|
|
|
|
|
- set(DEPS ${LITE_DIR}/lib/libpaddle_full_api_shared${CMAKE_SHARED_LIBRARY_SUFFIX})
|
|
|
|
|
- elseif(${LITE} STREQUAL "light")
|
|
|
|
|
- set(DEPS ${LITE_DIR}/lib/libpaddle_light_api_shared${CMAKE_SHARED_LIBRARY_SUFFIX})
|
|
|
|
|
- else()
|
|
|
|
|
- message(FATAL_ERROR "Unkown LITE release ${LITE}, only support light and full")
|
|
|
|
|
- endif()
|
|
|
|
|
|
|
+ message(FATAL_ERROR "Unkown LITE release ${LITE}, only support light and full")
|
|
|
endif()
|
|
endif()
|
|
|
|
|
|
|
|
set(DEPS ${DEPS} gflags yaml-cpp)
|
|
set(DEPS ${DEPS} gflags yaml-cpp)
|
|
|
set(EXTERNAL_LIB "-ldl -lrt -lgomp -lz -lm -lpthread")
|
|
set(EXTERNAL_LIB "-ldl -lrt -lgomp -lz -lm -lpthread")
|
|
|
set(DEPS ${DEPS} ${EXTERNAL_LIB})
|
|
set(DEPS ${DEPS} ${EXTERNAL_LIB})
|
|
|
-
|
|
|
|
|
set(DEPS ${DEPS} ${OpenCV_LIBS})
|
|
set(DEPS ${DEPS} ${OpenCV_LIBS})
|
|
|
|
|
+
|
|
|
add_executable(classifier demo/classifier.cpp src/transforms.cpp src/paddlex.cpp)
|
|
add_executable(classifier demo/classifier.cpp src/transforms.cpp src/paddlex.cpp)
|
|
|
ADD_DEPENDENCIES(classifier ext-yaml-cpp)
|
|
ADD_DEPENDENCIES(classifier ext-yaml-cpp)
|
|
|
target_link_libraries(classifier ${DEPS})
|
|
target_link_libraries(classifier ${DEPS})
|