Skip to content

Commit 089b0dc

Browse files
committed
Fix: NCNN compilation.
1 parent 02df07d commit 089b0dc

File tree

3 files changed

+17
-11
lines changed

3 files changed

+17
-11
lines changed

CMakeLists.txt

+10-10
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ if (USE_DLIB)
157157
PREFIX dlib
158158
INSTALL_DIR ${CMAKE_BINARY_DIR}
159159
URL https://github.com/davisking/dlib/archive/${DLIB_RELEASE_VERSION}.tar.gz
160-
CONFIGURE_COMMAND cd dlib && mkdir build && cd build && cmake .. -DDLIB_NO_GUI_SUPPORT=ON -DDLIB_USE_CUDA=ON -DCMAKE_INSTALL_PREFIX=${CMAKE_BINARY_DIR}/dlib/build && cmake --build . --config Release --target install
160+
CONFIGURE_COMMAND cd dlib && mkdir -p build && cd build && cmake .. -DDLIB_NO_GUI_SUPPORT=ON -DDLIB_USE_CUDA=ON -DCMAKE_INSTALL_PREFIX=${CMAKE_BINARY_DIR}/dlib/build && cmake --build . --config Release --target install
161161
BUILD_COMMAND ""
162162
INSTALL_COMMAND ""
163163
BUILD_IN_SOURCE 1
@@ -170,7 +170,7 @@ if (USE_DLIB)
170170
PREFIX dlib
171171
INSTALL_DIR ${CMAKE_BINARY_DIR}
172172
URL https://github.com/davisking/dlib/archive/${DLIB_RELEASE_VERSION}.tar.gz
173-
CONFIGURE_COMMAND cd dlib && mkdir build && cd build && cmake .. -DDLIB_NO_GUI_SUPPORT=ON -DDLIB_USE_CUDA=OFF -DCMAKE_INSTALL_PREFIX=${CMAKE_BINARY_DIR}/dlib/build && cmake --build . --config Release --target install
173+
CONFIGURE_COMMAND cd dlib && mkdir -p build && cd build && cmake .. -DDLIB_NO_GUI_SUPPORT=ON -DDLIB_USE_CUDA=OFF -DCMAKE_INSTALL_PREFIX=${CMAKE_BINARY_DIR}/dlib/build && cmake --build . --config Release --target install
174174
BUILD_COMMAND ""
175175
INSTALL_COMMAND ""
176176
BUILD_IN_SOURCE 1
@@ -283,7 +283,7 @@ if (USE_CAFFE2)
283283
set(PROTOBUF_INCLUDE_DIR ${CMAKE_BINARY_DIR}/protobuf/include)
284284
set(PROTOBUF_LIB_DIR ${CMAKE_BINARY_DIR}/protobuf/lib)
285285
if (NOT EXISTS ${CMAKE_BINARY_DIR}/protobuf)
286-
execute_process(COMMAND mkdir ${CMAKE_BINARY_DIR}/protobuf)
286+
execute_process(COMMAND mkdir -p ${CMAKE_BINARY_DIR}/protobuf)
287287
execute_process(COMMAND ln -s ${PYTORCH_PATH}/third_party/protobuf/src ${PROTOBUF_INCLUDE_DIR})
288288
execute_process(COMMAND ln -s ${PYTORCH_PATH}-build/lib ${PROTOBUF_LIB_DIR})
289289
execute_process(COMMAND ln -s ${PYTORCH_PATH}-build/bin ${CMAKE_BINARY_DIR}/protobuf/bin)
@@ -307,7 +307,7 @@ if (USE_CAFFE2)
307307

308308
# Python path
309309
if (NOT EXISTS ${CMAKE_BINARY_DIR}/python_path)
310-
execute_process(COMMAND mkdir ${CMAKE_BINARY_DIR}/python_path)
310+
execute_process(COMMAND mkdir -p ${CMAKE_BINARY_DIR}/python_path)
311311
execute_process(COMMAND ln -s ${PYTORCH_PATH}-build
312312
${CMAKE_BINARY_DIR}/python_path/pytorch)
313313
execute_process(COMMAND ln -s ${DETECTRON_PATH}
@@ -332,7 +332,7 @@ if (USE_TF)
332332
PREFIX tensorflow_cc
333333
INSTALL_DIR ${CMAKE_BINARY_DIR}
334334
DOWNLOAD_COMMAND git clone https://github.com/beniz/tensorflow_cc.git
335-
CONFIGURE_COMMAND cd tensorflow_cc && mkdir build && cd build && cmake -DTENSORFLOW_STATIC=OFF -DTENSORFLOW_SHARED=ON .. && make && ln -s ${CMAKE_BINARY_DIR}/tensorflow_cc/src/tensorflow_cc/tensorflow_cc/build/tensorflow/tensorflow/contrib/makefile/gen/protobuf ${CMAKE_BINARY_DIR}/protobuf
335+
CONFIGURE_COMMAND cd tensorflow_cc && mkdir -p build && cd build && cmake -DTENSORFLOW_STATIC=OFF -DTENSORFLOW_SHARED=ON .. && make && ln -s ${CMAKE_BINARY_DIR}/tensorflow_cc/src/tensorflow_cc/tensorflow_cc/build/tensorflow/tensorflow/contrib/makefile/gen/protobuf ${CMAKE_BINARY_DIR}/protobuf
336336
BUILD_COMMAND ""
337337
INSTALL_COMMAND ""
338338
BUILD_IN_SOURCE 1
@@ -344,7 +344,7 @@ if (USE_TF)
344344
PREFIX tensorflow_cc
345345
INSTALL_DIR ${CMAKE_BINARY_DIR}
346346
DOWNLOAD_COMMAND git clone https://github.com/beniz/tensorflow_cc.git
347-
CONFIGURE_COMMAND cd tensorflow_cc && mkdir build && cd build && cmake -DTENSORFLOW_STATIC=OFF -DTENSORFLOW_SHARED_CPU=ON .. && make && ln -s ${CMAKE_BINARY_DIR}/tensorflow_cc/src/tensorflow_cc/tensorflow_cc/build/tensorflow/tensorflow/contrib/makefile/gen/protobuf ${CMAKE_BINARY_DIR}/protobuf
347+
CONFIGURE_COMMAND cd tensorflow_cc && mkdir -p build && cd build && cmake -DTENSORFLOW_STATIC=OFF -DTENSORFLOW_SHARED_CPU=ON .. && make && ln -s ${CMAKE_BINARY_DIR}/tensorflow_cc/src/tensorflow_cc/tensorflow_cc/build/tensorflow/tensorflow/contrib/makefile/gen/protobuf ${CMAKE_BINARY_DIR}/protobuf
348348
BUILD_COMMAND ""
349349
INSTALL_COMMAND ""
350350
BUILD_IN_SOURCE 1
@@ -511,7 +511,7 @@ if (USE_NCNN)
511511
INSTALL_DIR ${CMAKE_BINARY_DIR}
512512
DOWNLOAD_COMMAND git clone --recursive https://github.com/Tencent/ncnn.git
513513
PATCH_COMMAND git reset --hard ${NCNN_COMMIT_HASH}
514-
CONFIGURE_COMMAND mkdir build && cd build && cmake .. -DPI3=ON -DCMAKE_TOOLCHAIN_FILE=../pi3.toolchain.cmake
514+
CONFIGURE_COMMAND mkdir -p build && cd build && cmake .. -DPI3=ON -DCMAKE_TOOLCHAIN_FILE=../pi3.toolchain.cmake
515515
BUILD_COMMAND cd build/ && make -j${N}
516516
INSTALL_COMMAND ""
517517
BUILD_IN_SOURCE 1
@@ -523,7 +523,7 @@ if (USE_NCNN)
523523
INSTALL_DIR ${CMAKE_BINARY_DIR}
524524
DOWNLOAD_COMMAND git clone --recursive https://github.com/Tencent/ncnn.git
525525
PATCH_COMMAND git reset --hard ${NCNN_COMMIT_HASH}
526-
CONFIGURE_COMMAND mkdir build && cd build && cmake ..
526+
CONFIGURE_COMMAND mkdir -p build && cd build && cmake ..
527527
BUILD_COMMAND cd build/ && make -j${N}
528528
INSTALL_COMMAND ""
529529
BUILD_IN_SOURCE 1
@@ -548,7 +548,7 @@ if (USE_XGBOOST)
548548
PREFIX xgboost
549549
INSTALL_DIR ${CMAKE_BINARY_DIR}
550550
DOWNLOAD_COMMAND git clone --recursive https://github.com/dmlc/xgboost.git
551-
CONFIGURE_COMMAND wget https://github.com/NVlabs/cub/archive/1.6.4.zip && unzip 1.6.4.zip && rm 1.6.4.zip && mkdir build && cd build &&
551+
CONFIGURE_COMMAND wget https://github.com/NVlabs/cub/archive/1.6.4.zip && unzip 1.6.4.zip && rm 1.6.4.zip && mkdir -p build && cd build &&
552552
cmake .. -DPLUGIN_UPDATER_GPU=ON -DCUB_DIRECTORY=${CMAKE_BINARY_DIR}/xgboost/src/xgboost/cub-1.6.4/ -DCUDA_NVCC_FLAGS=-Xcompiler\ -fPIC\ --expt-extended-lambda\ -gencode\ arch=compute_30,code=compute_30\ -gencode\ arch=compute_35,code=compute_35\ -gencode\ arch=compute_50,code=compute_50\ -gencode\ arch=compute_52,code=compute_52\ -gencode\ arch=compute_61,code=compute_61 && make && make
553553
INSTALL_COMMAND ""
554554
BUILD_IN_SOURCE 1
@@ -577,7 +577,7 @@ if (USE_TSNE)
577577
PREFIX Multicore-TSNE
578578
INSTALL_DIR ${CMAKE_BINARY_DIR}
579579
DOWNLOAD_COMMAND git clone https://github.com/beniz/Multicore-TSNE.git
580-
CONFIGURE_COMMAND cd multicore_tsne && mkdir build && cd build && cmake ..
580+
CONFIGURE_COMMAND cd multicore_tsne && mkdir -p build && cd build && cmake ..
581581
BUILD_COMMAND cd multicore_tsne/build/ && make
582582
INSTALL_COMMAND ""
583583
BUILD_IN_SOURCE 1

src/CMakeLists.txt

+6
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,12 @@ endif()
4141
if (USE_DLIB)
4242
list(APPEND ddetect_SOURCES backends/dlib/DNNStructures.h backends/dlib/dliblib.cc backends/dlib/dliblib.h backends/dlib/dlibmodel.cc backends/dlib/dlibmodel.h backends/dlib/dlibinputconns.h)
4343
endif()
44+
if (USE_NCNN)
45+
list(APPEND ddetect_SOURCES
46+
backends/ncnn/ncnnlib.cc
47+
backends/ncnn/ncnnmodel.cc
48+
)
49+
endif()
4450
add_library(ddetect ${ddetect_SOURCES})
4551
if (USE_CAFFE)
4652
add_dependencies(ddetect caffe_dd)

src/backends/ncnn/ncnnlib.cc

+1-1
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ namespace dd
7676
if (ad.has("threads"))
7777
_threads = ad.get("threads").get<int>();
7878
else
79-
_threads = dd_utils::hardware_concurrency();
79+
_threads = dd_utils::my_hardware_concurrency();
8080

8181
_blob_pool_allocator.set_size_compare_ratio(0.0f);
8282
_workspace_pool_allocator.set_size_compare_ratio(0.5f);

0 commit comments

Comments
 (0)