Skip to content
This repository has been archived by the owner on Jun 28, 2024. It is now read-only.

Latest commit

 

History

History
38 lines (25 loc) · 790 Bytes

README.md

File metadata and controls

38 lines (25 loc) · 790 Bytes

版面分析

简介

模型使用 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