1_quantize_model.sh support batchsize in source-file

我有一个pytorch模型(包含shufflenetV2), 转换为onnx。input shape=1,3,320,320
使用 0_import_model.sh时,会提示 Do not support shape > 4。但不会出错,能成功生成json, data文件。
使用1_quantize_model.sh出错,日志如下。我的source-file里面有20张图片。我尝试添加–batch-size 20到命令行中,会产生一样的错误。请问如何解决?

I Fitting image with scale.
I Channel mean value [127.5, 127.5, 127.5, 127.5]
I [TRAINER]Quantization start…
[TRAINER]Quantization start…
I Init validate tensor provider.
I Enqueue samples 20
I Init provider with 20 samples.
D set up a quantize net
D Process input_256 …
D Acuity output shape(input): (20 320 320 3)
D Real output shape: (20, 320, 320, 3)
D Process Conv_475_215 …
D Acuity output shape(convolution): (20 160 160 24)
D Real output shape: (20, 160, 160, 24)
D Process Relu_477_212 …
D Acuity output shape(relu): (20 160 160 24)
D Real output shape: (20, 160, 160, 24)
D Process MaxPool_478_209 …
D Acuity output shape(pooling): (20 80 80 24)
D Real output shape: (20, 80, 80, 24)
D Process Conv_479_206 …
D Acuity output shape(convolution): (20 40 40 24)
D Real output shape: (20, 40, 40, 24)
D Process Conv_481_202 …
D Acuity output shape(convolution): (20 40 40 24)
D Real output shape: (20, 40, 40, 24)
D Process Relu_483_198 …
D Acuity output shape(relu): (20 40 40 24)
D Real output shape: (20, 40, 40, 24)
D Process Conv_484_213 …
D Acuity output shape(convolution): (20 80 80 24)
D Real output shape: (20, 80, 80, 24)
D Process Relu_486_210 …
D Acuity output shape(relu): (20 80 80 24)
D Real output shape: (20, 80, 80, 24)
D Process Conv_487_207 …
D Acuity output shape(convolution): (20 40 40 24)
D Real output shape: (20, 40, 40, 24)
D Process Conv_489_203 …
D Acuity output shape(convolution): (20 40 40 24)
D Real output shape: (20, 40, 40, 24)
D Process Relu_491_199 …
D Acuity output shape(relu): (20 40 40 24)
D Real output shape: (20, 40, 40, 24)
D Process Concat_492_194 …
D Acuity output shape(concat): (20 40 40 48)
D Real output shape: (20, 40, 40, 48)
D Process Reshape_494_190_acuity_mark_perm_286 …
D Acuity output shape(permute): (20 48 40 40)
D Real output shape: (20, 48, 40, 40)
D Process Reshape_494_190 …
W Do not support shape > 4.
D Acuity output shape(reshape): (1 2 24 40 40)
Traceback (most recent call last):
File “tensorflow/python/framework/ops.py”, line 1659, in _create_c_op
tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot reshape a tensor with 1536000 elements to shape [1,2,24,40,40] (76800 elements) for ‘Reshape_494_190/Reshape_494_190’ (op: ‘Reshape’) with input shapes: [20,48,40,40], [5] and with input tensors computed as partial shapes: input[1] = [1,2,24,40,40].

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File “tensorzonex.py”, line 446, in
File “tensorzonex.py”, line 383, in main
File “acuitylib/app/tensorzone/quantization.py”, line 156, in run
File “acuitylib/app/tensorzone/quantization.py”, line 103, in _run_quantization
File “acuitylib/app/tensorzone/workspace.py”, line 172, in _setup_graph
File “acuitylib/app/tensorzone/graph.py”, line 59, in generate
File “acuitylib/acuitynetbuilder.py”, line 274, in build
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 301, in build_layer
File “acuitylib/acuitynetbuilder.py”, line 331, in build_layer
File “acuitylib/layer/acuitylayer.py”, line 280, in compute_tensor
File “acuitylib/layer/reshapelayer.py”, line 108, in compute_out_tensor
File “tensorflow/python/ops/gen_array_ops.py”, line 7179, in reshape
File “tensorflow/python/framework/op_def_library.py”, line 788, in _apply_op_helper
File “tensorflow/python/util/deprecation.py”, line 507, in new_func
File “tensorflow/python/framework/ops.py”, line 3300, in create_op
File “tensorflow/python/framework/ops.py”, line 1823, in init
File “tensorflow/python/framework/ops.py”, line 1662, in _create_c_op
ValueError: Cannot reshape a tensor with 1536000 elements to shape [1,2,24,40,40] (76800 elements) for ‘Reshape_494_190/Reshape_494_190’ (op: ‘Reshape’) with input shapes: [20,48,40,40], [5] and with input tensors computed as partial shapes: input[1] = [1,2,24,40,40].
[4705] Failed to execute script tensorzonex