基于pytorch的OCR识别库
对一年来学习的算法做了一个整理,同时也方便使用(项目持续更新中)
识别库的地址: github地址
这里支持的算法包括:
DBnet 论文链接
PSEnet 论文链接
PANnet 论文链接
SASTnet 论文链接
CRNN 论文链接
这里做了什么:
- 对上面算法进行了实现和实验
- 提供各个算法的训练和测试接口
- 提供各个算法的预训练模型
- 支持pytorch转成onnx,及onnx调用
接下来会做什么:
- 提供onnx转tensorrt的python接口,及调用
- 训练通用ocr的检测和识别模型,提供预训练模型和inference模型
- 模型压缩,进一步缩小模型大小
- 模型蒸馏,提升模型准确率
- tensorrt部署
- 手机端部署
效果:
最后,感谢各位论文作者的辛勤付出