diff --git a/tools/onnx/onnx2ncnn.cpp b/tools/onnx/onnx2ncnn.cpp index abe8f176d..fdda157f2 100644 --- a/tools/onnx/onnx2ncnn.cpp +++ b/tools/onnx/onnx2ncnn.cpp @@ -556,6 +556,10 @@ int main(int argc, char** argv) { fprintf(pp, "%-16s", "UnaryOp"); } + else if (op == "Flatten") + { + fprintf(pp, "%-16s", "Flatten"); + } else if (op == "Floor") { fprintf(pp, "%-16s", "UnaryOp"); @@ -1081,6 +1085,14 @@ int main(int argc, char** argv) int op_type = 7; fprintf(pp, " 0=%d", op_type); } + else if (op == "Flatten") + { + int axis = get_node_attr_i(node, "axis", 1); + if (axis != 1) + { + fprintf(stderr, "Unsupported Flatten axis %d!\n", axis); + } + } else if (op == "Floor") { int op_type = 2;