PytorchOCR
简介
PytorchOCR旨在打造一套训练，推理，部署一体的OCR引擎库
招募主程，有兴趣的可以邮箱联系nsnovio@gmail.com
更新日志
- 2020.07.01 添加 添加新算法文档
- 2020.06.29 添加检测的mb3和resnet50_vd预训练模型
- 2020.06.25 检测模块的训练和预测ok
- 2020.06.18 更新README
- 2020.06.17 识别模块的训练和预测ok
todo list
- crnn训练与python版预测
- DB训练与python版预测
- imagenet预训练模型
- 识别模型预训练模型
- DB通用模型
- 手机端部署
- With Triton
环境配置
需要的环境如下
- pytorch 1.4+
- torchvision 0.5+
- gcc 4.9+ (pse,pan会用到)
快速安装环境
pip3 install -r requirements.txt模型下载
链接：https://pan.baidu.com/s/1oCWJVyEpGAeagE4EwoV0kA 提取码：vvvx
文档教程
文本检测算法
PytorchOCR开源的文本检测算法列表：
- DB(paper)
在ICDAR2015文本检测公开数据集上，算法效果如下：
| 模型 | 骨干网络 | precision | recall | Hmean | 下载链接 |
|---|---|---|---|---|---|
| DB | MobileNetV3 | 85.09% | 67.98% | 75.58% | 见百度网盘 |
| DB | ResNet50_vd | 88.76% | 79.9% | 82.4% | 见百度网盘 |
文本识别算法
PytorchOCR开源的文本识别算法列表：
- CRNN(paper)
结果展示
贡献代码
我们非常欢迎你为PytorchOCR贡献代码，也十分感谢你的反馈。