PytorchOCR

简介

PytorchOCR旨在打造一套训练，推理，部署一体的OCR引擎库

招募主程，有兴趣的可以邮箱联系[email protected]

更新日志

• 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贡献代码，也十分感谢你的反馈。

相关仓库

• https://github.com/WenmuZhou/OCR_DataSet