Step-by-step guide to converting PP-FormulaNet_plus-M from PaddlePaddle to ONNX, fixing the model, and running predictions.
-
Clone the repository
git clone https://github.com/FahNos/pp_formula_to_onnx.git cd pp_formula_to_onnx -
Install dependencies
pip install -r requirements_onnx.txt
-
Download pre-trained model
wget -P ./pretrained_model https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-FormulaNet_plus-M_pretrained.pdparams
python ./tools/export_onnx.py --config ./configs/rec/PP-FormuaNet/PP-FormulaNet_plus-M_ONNX.yamlpython ./tools/fix_head_onnx.pypython ./tools/onnx_predict_pp_formualnet_plus_M.py --config ./configs/rec/PP-FormuaNet/PP-FormulaNet_plus-M_ONNX.yaml
\zeta_{0}(\nu)=-\frac{\nu\varrho^{-2\nu}}{\pi}\int_{\mu}^{\infty}d\omega\int_{C_{+}}d z\frac{2z^{2}}{(z^{2}+\omega^{2})^{\nu+1}}\breve{\Psi}(\omega;z)e^{i\epsilon z}\quad,
If this project helps you, please give it a ⭐ Star on GitHub!