提交 9d764e02 编写于 作者: xiebaiyuan's avatar xiebaiyuan

add simple build for mobilenetssd close #773

上级 df289bfc
#!/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
......
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册