E Catch exception when building RKNN model!E Traceback (most recent call last):E File "/home/bowie/miniconda3/envs/rknn/lib/python3.6/site-packages/tensorflow_core/python/framework/ops.py", line 1607, in _create_c_opE c_op = c_api.TF_FinishOperation(op_desc)E tensorflow.python.framework.errors_impl.InvalidArgumentError: Dimensions must be equal, but are 512 and 256 for 'ConvTranspose_251_31_2/Add' (op: 'Add') with input shapes: [100,16,16,512], [256].E During handling of the above exception, another exception occurred:......E File "rknn/base/RKNNlib/layer/RKNNlayer.py", line 287, in rknn.base.RKNNlib.layer.RKNNlayer.RKNNLayer.compute_tensorE File "rknn/base/RKNNlib/layer/deconvolution.py", line 121, in rknn.base.RKNNlib.layer.deconvolution.Deconvolution.compute_out_tensor......E File "/home/bowie/miniconda3/envs/rknn/lib/python3.6/site-packages/tensorflow_core/python/framework/ops.py", line 1770, in __init__E control_input_ops)E File "/home/bowie/miniconda3/envs/rknn/lib/python3.6/site-packages/tensorflow_core/python/framework/ops.py", line 1610, in _create_c_opE raise ValueError(str(e))
E ValueError: Dimensions must be equal, but are 512 and 256 for 'ConvTranspose_251_31_2/Add' (op: 'Add') with input shapes: [100,16,16,512], [256].
rknn = RKNN()
rknn.config(channel_mean_value='128 128 128 128', target_platform=['rk3399pro'])
rknn.load_onnx('model.onnx')
rknn.build(do_quantization=True)
leok 发表于 2020-5-26 08:36
试看看把相关toolkit装以下版本:
tensorflow-gpu: 1.13.1
onnx: 1.4.1
- ...
- E File "/home/bowie/miniconda3/envs/rknn/lib/python3.6/site-packages/tensorflow/python/ops/nn_ops.py", line 1641, in conv2d_transpose
- E filter.get_shape()[2]))
- E ValueError: output_shape does not match filter's output channels, 512 != 256
复制代码
jefferyzhang 发表于 2020-5-26 08:37
使用反卷积需要手动固定卷积核size大小,参看该贴:http://t.rock-chips.com/forum.php?mod=redirect&goto= ...
欢迎光临 Toybrick (https://t.rock-chips.com/) | Powered by Discuz! X3.3 |