未验证 提交 7ccdf7fe 编写于 作者: X xiebaiyuan 提交者: GitHub

Merge pull request #774 from xiebaiyuan/develop

add simple build for mobilenetssd  close #773
...@@ -31,7 +31,10 @@ elseif("FPGAnets" IN_LIST NET) ...@@ -31,7 +31,10 @@ elseif("FPGAnets" IN_LIST NET)
# target_link_libraries(test-resnet paddle-mobile) # target_link_libraries(test-resnet paddle-mobile)
ADD_EXECUTABLE(test-tensor-quant fpga/test_tensor_quant.cpp test_helper.h test_include.h executor_for_test.h) ADD_EXECUTABLE(test-tensor-quant fpga/test_tensor_quant.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-tensor-quant paddle-mobile) target_link_libraries(test-tensor-quant paddle-mobile)
elseif("mobilenetssd" IN_LIST NET)
# gen test
ADD_EXECUTABLE(test-mobilenetssd net/test_mobilenet+ssd.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-mobilenetssd paddle-mobile)
else () else ()
# gen test # gen test
......
#!/usr/bin/env bash #!/usr/bin/env bash
NETS="" NETS=""
declare -a supportedNets=("googlenet" "mobilenet" "yolo" "squeezenet" "resnet") declare -a supportedNets=("googlenet" "mobilenet" "yolo" "squeezenet" "resnet" "mobilenetssd")
build_for_mac() { build_for_mac() {
if [ ! `which brew` ]; then if [ ! `which brew` ]; then
......
...@@ -33,6 +33,28 @@ if ("mobilenet" IN_LIST NET) ...@@ -33,6 +33,28 @@ if ("mobilenet" IN_LIST NET)
endif() endif()
if ("mobilenetssd" IN_LIST NET)
message("mobilenetssd enabled")
set(FUSION_CONVBNRELU_OP ON)
set(FUSION_CONVBNRELU_OP ON)
set(FUSION_DWCONVBNRELU_OP ON)
set(FUSION_CONVADD_OP ON)
set(MULTICLASSNMS_OP ON)
set(SOFTMAX_OP ON)
set(TRANSPOSE_OP ON)
#feed
set(PRIORBOX_OP ON)
set(CONCAT_OP ON)
set(BOXCODER_OP ON)
set(RESHAPE_OP ON)
#fetch
#total
set(FOUND_MATCH ON)
endif()
if ("yolo" IN_LIST NET) if ("yolo" IN_LIST NET)
message("yolo enabled") message("yolo enabled")
set(BATCHNORM_OP ON) set(BATCHNORM_OP ON)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册