Awesome
<div align="center"> <div align="center"> <h1><b>Rapid ⚡︎ LaTeX OCR</b></h1> </div> <div> </div><a href="https://huggingface.co/spaces/SWHL/RapidLaTeXOCRDemo" target="_blank"><img src="https://img.shields.io/badge/%F0%9F%A4%97 Hugging Face-Demo-blue"></a> <a href="https://www.modelscope.cn/studios/liekkas/RapidLatexOCRDemo/summary" target="_blank"><img src="https://img.shields.io/badge/魔搭-Demo-blue"></a> <a href="https://aistudio.baidu.com/application/detail/23590" target="_blank"><img src="https://img.shields.io/badge/百度AI%20Studio-Demo-blue"></a> <a href=""><img src="https://img.shields.io/badge/Python->=3.6,<3.13-aff.svg"></a> <a href=""><img src="https://img.shields.io/badge/OS-Linux%2C%20Win%2C%20Mac-pink.svg"></a> <a href="https://pepy.tech/project/rapid_latex_ocr"><img src="https://static.pepy.tech/personalized-badge/rapid_latex_ocr?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=Downloads"></a> <a href="https://pypi.org/project/rapid_latex_ocr/"><img alt="PyPI" src="https://img.shields.io/pypi/v/rapid_latex_ocr"></a> <a href="https://semver.org/"><img alt="SemVer2.0" src="https://img.shields.io/badge/SemVer-2.0-brightgreen"></a> <a href="https://github.com/psf/black"><img src="https://img.shields.io/badge/code%20style-black-000000.svg"></a>
</div>Introduction
rapid_latex_ocr
is a tool to convert formula images to latex format.
The reasoning code in the repo is modified from LaTeX-OCR, the model has all been converted to ONNX format, and the reasoning code has been simplified, Inference is faster and easier to deploy.
The repo only has codes based on ONNXRuntime
or OpenVINO
inference in onnx format, and does not contain training model codes. If you want to train your own model, please move to LaTeX-OCR.
If it helps you, please give a little star ⭐ or sponsor a cup of coffee (click the link in Sponsor at the top of the page)
🔥 Model Conversion Notes 👉 ConvertLaTeXOCRToONNX
Online Demo
<div align="center"> <img src="https://github.com/RapidAI/RapidLatexOCR/releases/download/v0.0.0/demo.gif" width="100%" height="100%"> </div>Framework
flowchart LR
A(Preprocess Formula\n ProcessLaTeXFormulaTools) --> B(Train\n LaTeX-OCR) --> C(Convert \n ConvertLaTeXOCRToONNX) --> D(Deploy\n RapidLaTeXOCR)
click A "https://github.com/SWHL/ProcessLaTeXFormulaTools" _blank
click B "https://github.com/lukas-blecher/LaTeX-OCR" _blank
click C "https://github.com/SWHL/ConvertLaTeXOCRToONNX" _blank
click D "https://github.com/RapidAI/RapidLaTeXOCR" _blank
Installation
[!NOTE] When installing the package through pip, the model file will be automatically downloaded and placed under models in the installation directory.
If the Internet speed is slow, you can download it separately through Google Drive or Baidu NetDisk.
pip install rapid_latex_ocr
Usage
Used by python script
from rapid_latex_ocr import LaTeXOCR
model = LaTeXOCR()
img_path = "tests/test_files/6.png"
with open(img_path, "rb") as f:
data = f.read()
res, elapse = model(data)
print(res)
print(elapse)
Used by command line
$ rapid_latex_ocr tests/test_files/6.png
# {\\frac{x^{2}}{a^{2}}}-{\\frac{y^{2}}{b^{2}}}=1
# 0.47902780000000034
Code Contributors
<p align="left"> <a href="https://github.com/RapidAI/RapidLatexOCR/graphs/contributors"> <img src="https://contrib.rocks/image?repo=RapidAI/RapidLatexOCR" width="20%"/> </a> </p>Contributing
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Please make sure to update tests as appropriate.
Sponsor
If you want to sponsor the project, you can directly click the Buy me a coffee image, please write a note (e.g. your github account name) to facilitate adding to the sponsorship list below.
<div align="left"> <a href="https://www.buymeacoffee.com/SWHL"><img src="https://raw.githubusercontent.com/RapidAI/.github/main/assets/buymeacoffe.png" width="30%" height="30%"></a> </div>License
This project is released under the MIT license.