mangalametta/chinese-ocr

基于CTPN（tensorflow）+CRNN（pytorch）+CTC的不定长文本检测和识别

chinese-ocr

正體中文OCR系統

文字識別CTPN（tensorflow）
文字辨識CRNN（pytorch）+CTC
圖片生成Text Renderer

環境

python 3.6.7 + tensorflow 1.14.0 + pytorch 1.4.1

Demo

python demo.py    

下載 預訓練模型

還沒Train好

CRNN

將pytorch-crnn.pth放入/train/models中

CTPN

將checkpoints.zip解壓後的內容放入/ctpn/checkpoints中

模型訓練

CTPN訓練

詳見 tensorflow-ctpn

CRNN訓練

1.設定Text Renderer的字體、字庫、語料庫

修改train/render/config 在train/render/data下新增字體、字庫、語料庫

2.訓練

修改`train/config`
執行`train/train.py`

3.訓練結果

還沒Train好

效果展示

CTPN

還沒Train好

OCR

還沒Train好

參考

warp-ctc-pytorch chinese_ocr-(tensorflow+keras) CTPN-tensorflow crnn-pytorch