fix leakyrelu with default type

Tencent · Nov 28, 2018 · d9d7d6b · d9d7d6b
1 parent a941701
commit d9d7d6b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tools/mxnet/mxnet2ncnn.cpp b/tools/mxnet/mxnet2ncnn.cpp
@@ -728,9 +728,9 @@ int main(int argc, char** argv)
         if (n.op == "_contrib_MultiBoxDetection")
         {
             // reorder input blob
-            int temp = inputs[0];
-            inputs[0] = inputs[1];
-            inputs[1] = temp;
+            int temp = n.inputs[0];
+            n.inputs[0] = n.inputs[1];
+            n.inputs[1] = temp;
         }
 
         // input
@@ -1081,7 +1081,7 @@ int main(int argc, char** argv)
             {
                 fprintf(pp, "%-16s", "ELU");
             }
-            else if (type == "leaky")
+            else if (type == "leaky" || type.empty())
             {
                 fprintf(pp, "%-16s", "ReLU");
             }
@@ -1801,7 +1801,7 @@ int main(int argc, char** argv)
                 float slope = n.has_attr("slope") ? n.attr("slope") : 0.25f;
                 fprintf(pp, " 0=%f", slope);
             }
-            else if (type == "leaky")
+            else if (type == "leaky" || type.empty())
             {
                 float slope = n.has_attr("slope") ? n.attr("slope") : 0.25f;
                 fprintf(pp, " 0=%f", slope);