Browse Source

add split op support for onnx2ncnn (#1299)

tags/20191113
bindog nihui 6 years ago
parent
commit
53d54115a8
1 changed files with 19 additions and 0 deletions
  1. +19
    -0
      tools/onnx/onnx2ncnn.cpp

+ 19
- 0
tools/onnx/onnx2ncnn.cpp View File

@@ -1213,6 +1213,10 @@ int main(int argc, char** argv)
{
fprintf(pp, "%-16s", "Softmax");
}
else if (op == "Split")
{
fprintf(pp, "%-16s", "Slice");
}
else if (op == "Sqrt")
{
fprintf(pp, "%-16s", "UnaryOp");
@@ -2023,6 +2027,21 @@ int main(int argc, char** argv)
fprintf(pp, " 0=%d", axis-1);
fprintf(pp, " 1=1");
}
else if (op == "Split")
{
int axis = get_node_attr_i(node, "axis", 0);
std::vector<int> split = get_node_attr_ai(node, "split");
if (axis < 1 || split.size() < 2)
fprintf(stderr, "Unsupported split attributes !\n");

fprintf(pp, " -23300=%d", output_size);
for (int i=0; i< split.size() - 1; i++)
{
fprintf(pp, ",%d", split[i]);
}
fprintf(pp, ",-233");
fprintf(pp, " 1=%d", axis - 1);
}
else if (op == "Sqrt")
{
int op_type = 5;


Loading…
Cancel
Save