A simple LaTeX OCR package
Project description
Simple-LaTeX-OCR
The encoder is resnetv2+simplevit, the decoder is transformer, the attention mechanism uses flashattention2.0 from pytorch2.0, and the dataset size is 112 million
Online experience
https://huggingface.co/spaces/chaodreaming/Simple-LaTeX-OCR
https://www.modelscope.cn/studios/chaodreaming/Simple-LaTeX-OCR/summary
Install the package simple_latex_ocr
:
pip install simple_latex_ocr
Use from within Python
from simple_latex_ocr.models import Latex_OCR
model = Latex_OCR()
img_path = "tests/test_files/5.png"
result = model.predict(img_path)
print(result['formula'])
print(result['confidence'])
print(result['elapse'])
Used by command line
$ simple_latex_ocr tests/test_files/2.png
Used by api
$ python -m simple_latex_ocr.api.run
#You can use test.py to initiate a request for verification
Contribution
Contributions of any kind are welcome.
Acknowledgment
Code taken and modified from lukas-blecher, RapidAI,ultralytics
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
simple_latex_ocr-0.0.8.tar.gz
(16.1 kB
view hashes)
Built Distribution
Close
Hashes for simple_latex_ocr-0.0.8-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 578f47db1b54634583896201dc509f87a23efb8ed2849862ac1a2ffdaa38b251 |
|
MD5 | ebc4231619d8a96f8f25d92e34b3f8c0 |
|
BLAKE2b-256 | 0c8c55b4045a8871078a3f39423a3a207a9cc29e91e589ad1f154510fb2b1bdf |