pytorch 转换时报错

也遇到了同样问题，楼主能加个ｗｘ聊聊吗？

--> Building model
E Catch exception when building RKNN model!
E Traceback (most recent call last):
E   File "/home/user/.local/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1567, in _create_c_op
E     c_op = c_api.TF_FinishOperation(op_desc)
E tensorflow.python.framework.errors_impl.InvalidArgumentError: Dimension 1 in both shapes must be equal, but are 6 and 26. Shapes are [1,6,6] and [1,26,26]. for 'cat_at_2304_1/cat_at_2304_1' (op: 'ConcatV2') with input shapes: [1,6,6,256], [1,26,26,512], [] and with computed input tensors: input[2] = <3>.
E During handling of the above exception, another exception occurred:
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 737, in rknn.api.rknn_base.RKNNBase.build
E   File "rknn/api/rknn_base.py", line 1644, in rknn.api.rknn_base.RKNNBase._quantize2
E   File "rknn/base/RKNNlib/app/medusa/quantization.py", line 105, in rknn.base.RKNNlib.app.medusa.quantization.Quantization.run
E   File "rknn/base/RKNNlib/app/medusa/quantization.py", line 44, in rknn.base.RKNNlib.app.medusa.quantization.Quantization._run_quantization
E   File "rknn/base/RKNNlib/app/medusa/workspace.py", line 129, in rknn.base.RKNNlib.app.medusa.workspace.Workspace.run
E   File "rknn/base/RKNNlib/app/medusa/workspace.py", line 99, in rknn.base.RKNNlib.app.medusa.workspace.Workspace._setup_graph
E   File "rknn/base/RKNNlib/app/medusa/workspace.py", line 100, in rknn.base.RKNNlib.app.medusa.workspace.Workspace._setup_graph
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 274, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 278, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 305, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 331, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 336, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/layer/RKNNlayer.py", line 287, in rknn.base.RKNNlib.layer.RKNNlayer.RKNNLayer.compute_tensor
E   File "rknn/base/RKNNlib/layer/concat.py", line 44, in rknn.base.RKNNlib.layer.concat.Concat.compute_out_tensor
E   File "/home/user/.local/lib/python3.6/site-packages/tensorflow/python/ops/array_ops.py", line 1189, in concat
E     return gen_array_ops.concat_v2(values=values, axis=axis, name=name)
E   File "/home/user/.local/lib/python3.6/site-packages/tensorflow/python/ops/gen_array_ops.py", line 953, in concat_v2
E     "ConcatV2", values=values, axis=axis, name=name)
E   File "/home/user/.local/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
E     op_def=op_def)
E   File "/home/user/.local/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 3392, in create_op
E     op_def=op_def)
E   File "/home/user/.local/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1734, in __init__
E     control_input_ops)
E   File "/home/user/.local/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1570, in _create_c_op
E     raise ValueError(str(e))
E ValueError: Dimension 1 in both shapes must be equal, but are 6 and 26. Shapes are [1,6,6] and [1,26,26]. for 'cat_at_2304_1/cat_at_2304_1' (op: 'ConcatV2') with input shapes: [1,6,6,256], [1,26,26,512], [] and with computed input tensors: input[2] = <3>.
Build pytorch failed!

Sean 发表于 2020-3-25 09:13
也遇到了同样问题，楼主能加个ｗｘ聊聊吗？

--> Building model

用的是torch的yolo3代码