diff --git a/tools/mxnet/mxnet2ncnn.cpp b/tools/mxnet/mxnet2ncnn.cpp index 206f8a667..9ba7b03d5 100644 --- a/tools/mxnet/mxnet2ncnn.cpp +++ b/tools/mxnet/mxnet2ncnn.cpp @@ -992,6 +992,10 @@ int main(int argc, char** argv) fprintf(pp, "%-16s", "Input"); } + else if (n.op == "_contrib_BilinearResize2D") + { + fprintf(pp, "%-16s", "Interp"); + } else if (n.op == "_contrib_MultiBoxDetection") { fprintf(pp, "%-16s", "DetectionOutput"); @@ -1396,6 +1400,19 @@ int main(int argc, char** argv) // dummy input shape // fprintf(pp, " 0 0 0"); } + else if (n.op == "_contrib_BilinearResize2D") + { + float scale_height = n.has_attr("scale_height") ? n.attr("scale_height") : 1.f; + float scale_width = n.has_attr("scale_width") ? n.attr("scale_width") : 1.f; + int height = n.has_attr("scale_height") ? 0 : n.attr("height"); + int width = n.has_attr("scale_width") ? 0 : n.attr("width"); + + fprintf(pp, " 0=2"); + fprintf(pp, " 1=%f", scale_height); + fprintf(pp, " 2=%f", scale_width); + fprintf(pp, " 3=%d", height); + fprintf(pp, " 4=%d", width); + } else if (n.op == "_contrib_MultiBoxDetection") { float threshold = n.has_attr("threshold") ? n.attr("threshold") : 0.01f;