/OCR-net

PyTorch based OCR involving CTPN (detection) and CRNN (recognition)

Primary LanguagePython

OCR-net

PyTorch based OCR involving CTPN (detection) and CRNN (recognition)

Requirements

  • Python-3.6
  • pytorch-0.4.1+
  • torchvision-0.2.1
  • opencv-3.4.0.14
  • numpy-1.14.3

Detection

Detection is based on CTPN

Recognition

Recognition is based on CRNN

Test

Download pretrained models from gdrive link into checkpoints folder

python3 demo.py

Implementation References