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)