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

• pytorch_ctpn
• crnn.pytorch