diff --git a/configs/yolov3_mobilenet_v1_roadsign.yml b/configs/yolov3_mobilenet_v1_roadsign.yml index 294542f0fee5679aeb2e2a40bf211b1e34ebdda2..89fd3e7cb8705864a0d4e1ff95043e1c740caf48 100644 --- a/configs/yolov3_mobilenet_v1_roadsign.yml +++ b/configs/yolov3_mobilenet_v1_roadsign.yml @@ -144,7 +144,7 @@ EvalReader: - !Permute to_bgr: false channel_first: True - batch_size: 8 + batch_size: 1 drop_empty: false worker_num: 4 bufsize: 2 diff --git a/deploy/cpp/CMakeLists.txt b/deploy/cpp/CMakeLists.txt index de2f9fe66a35addc0a73af120153230a0022282c..0517825795dce9ae4c550e174eb642bd2273d6bd 100644 --- a/deploy/cpp/CMakeLists.txt +++ b/deploy/cpp/CMakeLists.txt @@ -113,9 +113,8 @@ endif() if (NOT WIN32) if (WITH_TENSORRT AND WITH_GPU) - include_directories("${TENSORRT_INC_DIR}/include") - #link_directories("${TENSORRT_LIB_DIR}/lib") - link_directories("${TENSORRT_LIB_DIR}/") + include_directories("${TENSORRT_INC_DIR}") + link_directories("${TENSORRT_LIB_DIR}") endif() endif(NOT WIN32) @@ -149,7 +148,11 @@ if(WITH_MKL) endif () endif() else() - set(MATH_LIB ${PADDLE_DIR}/third_party/install/openblas/lib/libopenblas${CMAKE_STATIC_LIBRARY_SUFFIX}) + if (WIN32) + set(MATH_LIB ${PADDLE_DIR}/third_party/install/openblas/lib/openblas${CMAKE_STATIC_LIBRARY_SUFFIX}) + else() + set(MATH_LIB ${PADDLE_DIR}/third_party/install/openblas/lib/libopenblas${CMAKE_STATIC_LIBRARY_SUFFIX}) + endif() endif() if (WIN32)