diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt index 1033cfa180ac6928b2edf6b0cef2885dd0e72a8c..a144e553236e300ad24501420183df01dd15aad5 100644 --- a/test/CMakeLists.txt +++ b/test/CMakeLists.txt @@ -31,7 +31,10 @@ elseif("FPGAnets" IN_LIST NET) # 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) 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 () # gen test diff --git a/tools/build.sh b/tools/build.sh index bf3545ef162c86c16c0877f5f25f3a1e09de1fd4..4d877720a32c6c79df6aca60a62a4da7e997d803 100755 --- a/tools/build.sh +++ b/tools/build.sh @@ -1,6 +1,6 @@ #!/usr/bin/env bash NETS="" -declare -a supportedNets=("googlenet" "mobilenet" "yolo" "squeezenet" "resnet") +declare -a supportedNets=("googlenet" "mobilenet" "yolo" "squeezenet" "resnet" "mobilenetssd") build_for_mac() { if [ ! `which brew` ]; then diff --git a/tools/op.cmake b/tools/op.cmake index a3b40d07e2d5c1a241a7fb2887f7fe07645b3173..8f5ffb52aeae29c76d0d456a1392b5411cb5d04a 100644 --- a/tools/op.cmake +++ b/tools/op.cmake @@ -33,6 +33,28 @@ if ("mobilenet" IN_LIST NET) 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) message("yolo enabled") set(BATCHNORM_OP ON)