diff --git a/tools/onnx/onnx2ncnn.cpp b/tools/onnx/onnx2ncnn.cpp index 7c7342900..c966aefd3 100644 --- a/tools/onnx/onnx2ncnn.cpp +++ b/tools/onnx/onnx2ncnn.cpp @@ -2072,15 +2072,25 @@ int main(int argc, char** argv) { int axis = get_node_attr_i(node, "axis", 0); std::vector split = get_node_attr_ai(node, "split"); - if (axis < 1 || split.size() < 2) - fprintf(stderr, "Unsupported split attributes !\n"); + if (axis < 1) + fprintf(stderr, "Unsupported split axis !\n"); fprintf(pp, " -23300=%d", output_size); - for (int i=0; i< split.size() - 1; i++) + if (split.empty()) { - fprintf(pp, ",%d", split[i]); + for (int i=0; i