From 65be036aa6562848449abe901dfcf4fdbbc4db77 Mon Sep 17 00:00:00 2001
From: nihuini <nihuini@tencent.com>
Date: Mon, 3 Dec 2018 19:43:11 +0800
Subject: [PATCH] mxnet-ssd done

---
 src/layer/detectionoutput.cpp | 12 +++++++++---
 src/layer/priorbox.cpp        |  2 --
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/layer/detectionoutput.cpp b/src/layer/detectionoutput.cpp
index e5143a3e1..8d424b97e 100644
--- a/src/layer/detectionoutput.cpp
+++ b/src/layer/detectionoutput.cpp
@@ -151,11 +151,14 @@ int DetectionOutput::forward(const std::vector<Mat>& bottom_blobs, std::vector<M
     const Mat& confidence = bottom_blobs[1];
     const Mat& priorbox = bottom_blobs[2];
 
-    const int num_prior = priorbox.w / 4;
+    bool mxnet_ssd_style = priorbox.dims == 2;
+
+    // mxnet-ssd _contrib_MultiBoxDetection
+    const int num_prior = mxnet_ssd_style ? priorbox.h : priorbox.w / 4;
 
     int num_class_copy = num_class;
     if (num_class_copy == -233)
-        num_class_copy = priorbox.w / num_prior;
+        num_class_copy = confidence.h;
 
     // apply location with priorbox
     Mat bboxes;
@@ -209,7 +212,10 @@ int DetectionOutput::forward(const std::vector<Mat>& bottom_blobs, std::vector<M
 
         for (int j = 0; j < num_prior; j++)
         {
-            float score = confidence[j * num_class_copy + i];
+            // prob data layout
+            // caffe-ssd = num_class x num_prior
+            // mxnet-ssd = num_prior x num_class
+            float score = mxnet_ssd_style ? confidence[i * num_prior + j] : confidence[j * num_class_copy + i];
 
             if (score > confidence_threshold)
             {
diff --git a/src/layer/priorbox.cpp b/src/layer/priorbox.cpp
index 2abd5fee0..a104186ea 100644
--- a/src/layer/priorbox.cpp
+++ b/src/layer/priorbox.cpp
@@ -54,8 +54,6 @@ int PriorBox::forward(const std::vector<Mat>& bottom_blobs, std::vector<Mat>& to
     if (bottom_blobs.size() == 1 && image_width == -233 && image_height == -233 && max_sizes.empty())
     {
         // mxnet style _contrib_MultiBoxPrior
-        fprintf(stderr, "mxnet style _contrib_MultiBoxPrior\n");
-
         float step_w = step_width;
         float step_h = step_height;
         if (step_w == -233)