Felix92's picture
Update README.md
8bde31c verified
metadata
language:
  - en
  - da
  - fr
  - it
  - es
  - de
  - pt
  - cs
  - pl
  - nl
  - 'no'
  - fi
license: apache-2.0

Optical Character Recognition made seamless & accessible to anyone, powered by TensorFlow 2 & PyTorch

Task: recognition

https://github.com/mindee/doctr

Example usage:

>>> from doctr.io import DocumentFile
>>> from doctr.models import ocr_predictor, from_hub

>>> img = DocumentFile.from_images(['<image_path>'])
>>> # Load your model from the hub
>>> model = from_hub('mindee/my-model')

>>> # Pass it to the predictor
>>> # If your model is a recognition model:
>>> predictor = ocr_predictor(det_arch='db_mobilenet_v3_large',
>>>                           reco_arch=model,
>>>                           pretrained=True)

>>> # If your model is a detection model:
>>> predictor = ocr_predictor(det_arch=model,
>>>                           reco_arch='crnn_mobilenet_v3_small',
>>>                           pretrained=True)

>>> # Get your predictions
>>> res = predictor(img)