From d5d7886517e748e05da099cdbfd5784ebc1d8de1 Mon Sep 17 00:00:00 2001
From: nihuini <nihuini@tencent.com>
Date: Wed, 30 Oct 2019 10:31:24 +0800
Subject: [PATCH] resolve onnx Split conversion with empty split attribute

---
 tools/onnx/onnx2ncnn.cpp | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

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<int> 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<output_size; i++)
+                {
+                    fprintf(pp, ",-233");
+                }
+            }
+            else
+            {
+                for (int i=0; i<split.size() - 1; i++)
+                {
+                    fprintf(pp, ",%d", split[i]);
+                }
+                fprintf(pp, ",-233");
             }
-            fprintf(pp, ",-233");
             fprintf(pp, " 1=%d", axis - 1);
         }
         else if (op == "Sqrt")