Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Calex_1990
PaddleOCR
提交
f2a2f595
P
PaddleOCR
项目概览
Calex_1990
/
PaddleOCR
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleOCR
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleOCR
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
f2a2f595
编写于
7月 12, 2020
作者:
X
xiaoting
提交者:
GitHub
7月 12, 2020
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #304 from lyl120117/dev_vis
dump model structure
上级
3b60404e
d89d3384
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
13 addition
and
0 deletion
+13
-0
configs/rec/rec_icdar15_train.yml
configs/rec/rec_icdar15_train.yml
+1
-0
tools/program.py
tools/program.py
+3
-0
tools/train.py
tools/train.py
+9
-0
未找到文件。
configs/rec/rec_icdar15_train.yml
浏览文件 @
f2a2f595
...
...
@@ -14,6 +14,7 @@ Global:
character_type
:
en
loss_type
:
ctc
distort
:
true
debug
:
false
reader_yml
:
./configs/rec/rec_icdar15_reader.yml
pretrain_weights
:
./pretrain_models/rec_mv3_none_bilstm_ctc/best_accuracy
checkpoints
:
...
...
tools/program.py
浏览文件 @
f2a2f595
...
...
@@ -75,6 +75,8 @@ class AttrDict(dict):
global_config
=
AttrDict
()
default_config
=
{
'Global'
:
{
'debug'
:
False
,
}}
def
load_config
(
file_path
):
"""
...
...
@@ -85,6 +87,7 @@ def load_config(file_path):
Returns: global config
"""
merge_config
(
default_config
)
_
,
ext
=
os
.
path
.
splitext
(
file_path
)
assert
ext
in
[
'.yml'
,
'.yaml'
],
"only support yaml files for now"
merge_config
(
yaml
.
load
(
open
(
file_path
),
Loader
=
yaml
.
Loader
))
...
...
tools/train.py
浏览文件 @
f2a2f595
...
...
@@ -43,6 +43,7 @@ logger = initial_logger()
from
ppocr.data.reader_main
import
reader_main
from
ppocr.utils.save_load
import
init_model
from
ppocr.utils.character
import
CharacterOps
from
paddle.fluid.contrib.model_stat
import
summary
def
main
():
...
...
@@ -87,6 +88,14 @@ def main():
# compile program for multi-devices
train_compile_program
=
program
.
create_multi_devices_program
(
train_program
,
train_opt_loss_name
)
# dump mode structure
if
config
[
'Global'
][
'debug'
]:
if
'Attention'
in
config
[
'Head'
].
keys
():
logger
.
warning
(
'Does not suport dump attention...'
)
else
:
summary
(
train_program
)
init_model
(
config
,
train_program
,
exe
)
train_info_dict
=
{
'compile_program'
:
train_compile_program
,
\
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录