/text-detect-recognition-hub

a project for text detection and recognition using PyTorch 1.3 include train code test code infer code and dev in flask code

Primary LanguagePython

text-detect-recognition-hub

text-detect-recognition is an open source project text detection and text recognition toolbox base on the pytorch.

It work's on pytorch 1.3-1.5

ops:nativectc requires pytorch 1.7

features:

  • split module design
  • Integrated distributed training

安装

第一次安装该代码要安装cpp拓展部分.

sudo python3 setup.py build_ext --inplace

训练

训练代码:

python3 -m torch.distributed.launch  --nproc_per_node=2  tools/train_*.py  config.py   --gpus 0

训练的参数有保存工作目录, 是否从断点开始恢复等,详情看tools/train.sh

使用DataParallel进行训练：

 tools/train.py  config.py  --distributed 0 --gpus 2

eval

python3 tools/eval.py config.py *.pth --eval detect
python3 tools/eval.py config.py *.pth --eval reco

部署 推理：

推理 的代码: 看tests/ test_inferrenceapi.py test_detetcorapi.py 有如何得到文本框跟预测

end2end api: tools/end2endflaskapi.py 提供了利用flask 部署ocr服务的方式，其中提供了文字检测跟文字识别结合的形式 其只要修改det_config_file,det_checkpoint rec_config_file,rec_checkpoint所在路径即可

##启动命令：
python3 tools/end2endflaskapi.py


##另开终端测试命令：
curl -F 'image=@/data/zhb/data/receipt/end2end/receipt_2nd_icdr15val/imgs/img_10000.jpg' 127.0.0.1:5000/ocr