PaddleX/latest/module_usage/tutorials/ocr_modules/formula_recognition

[giscus[bot]]

PaddleX/latest/module_usage/tutorials/ocr_modules/formula_recognition

https://paddlepaddle.github.io/PaddleX/latest/module_usage/tutorials/ocr_modules/formula_recognition.html

[Number18-tong]

from paddlex import create_model
model = create_model(model_name="PP-FormulaNet-L")
output = model.predict(input="layout6.png", batch_size=1)
for res in output:
res.print()
用上方代码测试后报错如下：
model = create_model(model_name="PP-FormulaNet-L")
File "/usr/local/lib/python3.10/dist-packages/paddlex/model.py", line 29, in create_model
return _ModelBasedInference(
File "/usr/local/lib/python3.10/dist-packages/paddlex/model.py", line 59, in init
self._predictor = create_predictor(*args, **kwargs)
File "/usr/local/lib/python3.10/dist-packages/paddlex/inference/models/init.py", line 112, in create_predictor
return BasicPredictor.get(model_name)(
File "/usr/local/lib/python3.10/dist-packages/paddlex/inference/models/formula_recognition/predictor.py", line 67, in init
self.pre_tfs, self.infer, self.post_op = self._build()
File "/usr/local/lib/python3.10/dist-packages/paddlex/inference/models/formula_recognition/predictor.py", line 96, in _build
infer = StaticInfer(
File "/usr/local/lib/python3.10/dist-packages/paddlex/inference/models/common/static_infer.py", line 125, in init
self._update_option(option)
File "/usr/local/lib/python3.10/dist-packages/paddlex/inference/models/common/static_infer.py", line 131, in _update_option
self._reset()
File "/usr/local/lib/python3.10/dist-packages/paddlex/inference/models/common/static_infer.py", line 148, in _reset
) = self._create()
File "/usr/local/lib/python3.10/dist-packages/paddlex/inference/models/common/static_infer.py", line 299, in _create
predictor = create_predictor(config)
ValueError: (InvalidArgument) The size of inputs must be equal to 2.
[Hint: Expected input_size == 2, but received input_size:1 != 2:2.] (at paddle/fluid/pir/dialect/operator/ir/pd_op3.cc:15851)

[liuhongen1234567]

您好，这是由于paddle和paddlex版本不匹配导致的，可以先wget https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-FormulaNet-L_pretrained.pdparams，然后在paddlex3.0正式版上执行下面的命令重新导出一下PP-FormulaNet-L模型 FLAGS_json_format_model=1 python main.py -c paddlex/configs/modules/formula_recognition/PP-FormulaNet-L.yaml
-o Global.mode=export
-o Export.weight_path='/paddle/code/paddlex/model_weight/PP-FormulaNet-L_pretrained.pdparams'，
之后，把 model = create_model(model_name="PP-FormulaNet-L") 改成 model = create_model(model_name="PP-FormulaNet-L"，model_dir ="之前导出的output")目录

[liuhongen1234567]

我们近期会上传paddle3.0正式版的模型