Merge pull request #110 from CodingBoO/test_branch

FIX: get step size for priorbox layer
alibaba · Jun 10, 2019 · 2170047 · 2170047
2 parents 0b085d6 + 965f793
commit 2170047
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tools/converter/source/caffe/PriorBox.cpp b/tools/converter/source/caffe/PriorBox.cpp
@@ -58,5 +58,15 @@ void PrioxBox::run(MNN::OpT* dstOp, const caffe::LayerParameter& parameters, con
     if (caffePrior.has_offset()) {
         prior->offset = caffePrior.offset();
     }
+    if(caffePrior.has_step()){
+        prior->stepWidth=caffePrior.step();
+        prior->stepHeight=caffePrior.step();
+    }else if(caffePrior.has_step_h() && caffePrior.has_step_w()){
+        prior->stepWidth=caffePrior.step_w();
+        prior->stepHeight=caffePrior.step_h();
+    }else{
+        prior->stepWidth=0;
+        prior->stepHeight=0;
+    }
 }
 static OpConverterRegister<PrioxBox> a("PriorBox");