1 Star 0 Fork 68

Jackey870 / document-ocr

forked from 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/Jakey870_admin/document-ocr.git
git@gitee.com:Jakey870_admin/document-ocr.git
Jakey870_admin
document-ocr
document-ocr
master

搜索帮助

53164aa7 5694891 3bd8fe86 5694891