[PaddlePaddle/PaddleOCR]表格识别自定义模型问题

系统环境/System Environment：Ubuntu 16

版本号/Version：Paddle：2.3.2-GPU PaddleOCR：2.6 问题相关组件/Related components：

运行指令/Command Code：

!python ppstructure/table/predict_table.py \ --det_model_dir=/data/lcpang/lc/project_table/model/infer_model/det_sast_ic15 \ --rec_model_dir=/data/lcpang/lc/project_table/model/infer_model/rec_r31_sar_infer \ --table_model_dir=/data/lcpang/lc/project_table/model/infer_model/table_master \ --rec_char_dict_path=ppocr/utils/dict90.txt \ --table_char_dict_path=ppocr/utils/dict/table_master_structure_dict.txt \ --image_dir=/data/lcpang/lc/project_table/dataset/pubtabnet_icdar_data/imgs/val/PMC1501036_002_00.png \ --output=output/table

完整报错/Complete Error Message： [2022/09/26 21:57:33] ppocr INFO: [0/1] /data/lcpang/lc/project_table/dataset/pubtabnet_icdar_data/imgs/val/PMC1501036_002_00.png Traceback (most recent call last): File "ppstructure/table/predict_table.py", line 254, in main(args) File "ppstructure/table/predict_table.py", line 206, in main predres, = table_sys(img) File "ppstructure/table/predict_table.py", line 103, in call structure_res, elapse = self._structure(copy.deepcopy(img)) File "ppstructure/table/predict_table.py", line 132, in _structure structure_res, elapse = self.table_structurer(copy.deepcopy(img)) File "/data/lcpang/lc/PaddleOCR-2.6.0/ppstructure/table/predict_structure.py", line 103, in call self.predictor.run() ValueError: In user code:

File "tools/export_model.py", line 255, in main() File "tools/export_model.py", line 251, in main model, arch_config, save_path, logger, input_shape=input_shape) File "tools/export_model.py", line 171, in export_single_model paddle.jit.save(model, save_path) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/jit.py", line 631, in wrapper func(layer, path, input_spec, configs) File "/data/lcpang/.local/lib/python3.7/site-packages/decorator.py", line 232, in fun return caller(func, *(extras + args), *kw) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/wrapped_decorator.py", line 25, in impl return wrapped_func(args, kwargs) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/base.py", line 51, in impl return func(*args, kwargs) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/jit.py", line 861, in save inner_input_spec, with_hook=with_hook) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 528, in concrete_program_specify_input_spec *desired_input_spec, with_hook=with_hook) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 436, in get_concrete_program concrete_program, partial_program_layer = self._program_cache[cache_key] File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 801, in getitem self._caches[item_id] = self._build_once(item) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 790, in _build_once *cache_key.kwargs) File "/data/lcpang/.local/lib/python3.7/site-packages/decorator.py", line 232, in fun return caller(func, (extras + args), kw) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/wrapped_decorator.py", line 25, in impl return wrapped_func(*args, kwargs) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/base.py", line 51, in impl return func(*args, kwargs) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/program_translator.py", line 733, in from_func_spec outputs = static_func(inputs) File "/tmp/tmpsjo5nkzi.py", line 28, in forward false_fn_1, (x,), (x,), (x,)) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/convert_operators.py", line 211, in convert_ifelse out = _run_py_ifelse(pred, true_fn, false_fn, true_args, false_args) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/convert_operators.py", line 257, in _run_py_ifelse return true_fn(true_args) if pred else false_fn(false_args) File "/data/lcpang/lc/PaddleOCR-2.6.0/ppocr/modeling/architectures/base_model.py", line 86, in forward x = self.backbone(x) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 930, in call return self._dygraph_call_func(inputs, kwargs) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 915, in _dygraph_call_func outputs = self.forward(*inputs, kwargs) File "/data/lcpang/lc/PaddleOCR-2.6.0/ppocr/modeling/backbones/table_master_resnet.py", line 214, in forward x = self.layer2(x) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 930, in call return self._dygraph_call_func(*inputs, kwargs) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 915, in _dygraph_call_func outputs = self.forward(*inputs, *kwargs) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/container.py", line 98, in forward input = layer(input) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 930, in call return self._dygraph_call_func(inputs, kwargs) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 915, in _dygraph_call_func outputs = self.forward(*inputs, kwargs) File "/tmp/tmprh5qbiop.py", line 21, in forward true_fn_11, false_fn_11, (out,), (out,), (out,)) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/convert_operators.py", line 211, in convert_ifelse out = _run_py_ifelse(pred, true_fn, false_fn, true_args, false_args) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/convert_operators.py", line 257, in _run_py_ifelse return true_fn(true_args) if pred else false_fn(false_args) File "/data/lcpang/lc/PaddleOCR-2.6.0/ppocr/modeling/backbones/table_master_resnet.py", line 73, in forward out = self.context_block(out) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 930, in call return self._dygraph_call_func(*inputs, *kwargs) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 915, in _dygraph_call_func outputs = self.forward(inputs, kwargs) File "/data/lcpang/lc/PaddleOCR-2.6.0/ppocr/modeling/backbones/table_master_resnet.py", line 343, in forward context = self.spatial_pool(x) File "/tmp/tmp7fpv9x8x.py", line 59, in spatial_pool ,), (context, x)) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/convert_operators.py", line 211, in convert_ifelse out = _run_py_ifelse(pred, true_fn, false_fn, true_args, false_args) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/dygraph/dygraph_to_static/convert_operators.py", line 257, in _run_py_ifelse return true_fn(true_args) if pred else false_fn(false_args) File "/data/lcpang/lc/PaddleOCR-2.6.0/ppocr/modeling/backbones/table_master_resnet.py", line 300, in spatial_pool batch self.headers, self.single_header_inplanes, height, width File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/tensor/manipulation.py", line 2139, in reshape return paddle.fluid.layers.reshape(x=x, shape=shape, name=name) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/layers/nn.py", line 6450, in reshape "XShape": x_shape}) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/layer_helper.py", line 44, in append_op return self.main_program.current_block().append_op(args, **kwargs) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/framework.py", line 3621, in append_op attrs=kwargs.get("attrs", None)) File "/data/lcpang/anaconda3/envs/ppstr/lib/python3.7/site-packages/paddle/fluid/framework.py", line 2635, in init for frame in traceback.extract_stack():

InvalidArgumentError: The 'shape' attribute in ReshapeOp is invalid. The input tensor X'size must be divisible by known capacity of 'shape'. But received X's shape = [1, 256, 122, 122], X's size = 3810304, 'shape' is [-1, 256, 120, 120], known capacity of 'shape' is -3686400. [Hint: Expected output_shape[unk_dim_idx] capacity == -in_size, but received output_shape[unk_dim_idx] capacity:-3686400 != -in_size:-3810304.] (at /paddle/paddle/fluid/operators/reshape_op.cc:190) [operator < reshape2 > error]