34 Star 162 Fork 68

Rocky / document-ocr

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
README.md 790 Bytes
一键复制 编辑 原始数据 按行查看 历史
rockyzhengwu 提交于 2019-10-27 00:10 . fix read

版面分析

简介

模型使用 UNet 结构在训练时用一个decoder 作为辅助,整体架构如下图

用法

  1. 数据准备  准备数据如样例 生成label_list 文件如下
/data/1.xml
/data/2.xml

data_factory.py 中配置相关参数

可自行生成其他格式数据实现对应的 data_generator 即可

  1. 训练
python train.py --name <task_name>
  1. 测试 & 导出模型
python --eval.py --name <task_name>
python --eval.py --export 
  1. 加载导出的模型 在 ```load_saved_mode.py`` 中添加模型位置然后运行就能运行测试导出的模型或者对外提供API

Python
1
https://gitee.com/rockyzheng/document-ocr.git
git@gitee.com:rockyzheng/document-ocr.git
rockyzheng
document-ocr
document-ocr
master

搜索帮助