/FOTS_two_stage

EAST + Rotate + CRNN

Primary LanguagePython

FOTS_two_stage

  • 中文OCR。
  • 检测分支为EAST，识别分支为CRNN。
  • 仅是测试代码
  • 训练数据集为 RCTW-17

Pre-requisite

  • python 3.6
  • torch
  • torchvision
  • pretrainedmodels
  • PIL
  • shapely
  • lmdb
  • 如果使用lanms，而不是locality_aware_nms，则需要python 3.5

Data

使用RCTW-17中文数据集。

数据集下载：

http://rctw.vlrlab.net/dataset/

How to use

  1. clone：
    • git clone https://github.com/MagicianQi/FOTS_two_stage
    • cd ./FOTS_two_stage/
    • 下载模型及测试图像 wget https://github.com/MagicianQi/FOTS_two_stage/releases/download/v0.1/models_and_imgs.zip
    • 解压 unzip models_and_imgs.zip
  2. 测试检测分支：
    • python testing_detection_branch.py
  3. 测试识别分支：
    • python testing_recognition_branch.py
  4. 测试OCR：
    • python testing_ocr.py