From 9d764e023d134078db28204101deb90837ad9727 Mon Sep 17 00:00:00 2001
From: xiebaiyuan <xiebaiyuan@139.com>
Date: Wed, 15 Aug 2018 17:49:47 +0800
Subject: [PATCH] add simple build for mobilenetssd  close #773

---
 tools/build.sh |  2 +-
 tools/op.cmake | 22 ++++++++++++++++++++++
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/tools/build.sh b/tools/build.sh
index bf3545ef16..4d877720a3 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 a3b40d07e2..8f5ffb52ae 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)
-- 
GitLab