yolov5s.onnx转换tflite出错

[crash-A]

这是我的转换代码 from converter import onnx_converter onnx_converter( onnx_model_path = "./yolov5s.onnx", need_simplify = True, output_path = "./", target_formats = ['tflite'], # or ['keras'], ['keras', 'tflite'] weight_quant = False, int8_model = False, int8_mean = None, int8_std = None, image_root = None ) 这是报错内容 Checking 0/1... INFO:deformation_layers ::Reshape will process tensor after change back to NCHW format. Traceback (most recent call last): File "e:\nncase\onnx2tflite-main\onnx2tflite-main\onnx2tflite.py", line 2, in onnx_converter( File "e:\nncase\onnx2tflite-main\onnx2tflite-main\converter.py", line 21, in onnx_converter keras_model = keras_builder(model_proto, native_groupconv) File "e:\nncase\onnx2tflite-main\onnx2tflite-main\utils\builder.py", line 82, in keras_builder tf_tensor[node_outputs[index]] = tf_operator(tf_tensor, onnx_weights, node_inputs, op_attr, index=index)(_inputs) File "e:\nncase\onnx2tflite-main\onnx2tflite-main\layers\deformation_layers.py", line 140, in init end = start + node_attribute['split'][index] KeyError: 'split' 能麻烦看一下吗

[shichaoya]

这个问题有解决嘛，我也遇到了这个问题

[MPolaris]

额，你们可以debug一下这里，并自己修改一下。 如果还有问题，可以传一个onnx来我看看。