Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleOCR
提交
72ebbf2d
P
PaddleOCR
项目概览
PaddlePaddle
/
PaddleOCR
大约 1 年 前同步成功
通知
1528
Star
32962
Fork
6643
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
108
列表
看板
标记
里程碑
合并请求
7
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleOCR
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
108
Issue
108
列表
看板
标记
里程碑
合并请求
7
合并请求
7
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
72ebbf2d
编写于
5月 15, 2020
作者:
L
LDOUBLEV
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
sorted outputs when export model
上级
d539508e
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
7 addition
and
15 deletion
+7
-15
tools/infer/predict_det.py
tools/infer/predict_det.py
+5
-12
tools/infer/predict_rec.py
tools/infer/predict_rec.py
+0
-1
tools/program.py
tools/program.py
+2
-2
未找到文件。
tools/infer/predict_det.py
浏览文件 @
72ebbf2d
...
...
@@ -142,8 +142,8 @@ class TextDetector(object):
outputs
.
append
(
output
)
outs_dict
=
{}
if
self
.
det_algorithm
==
"EAST"
:
outs_dict
[
'f_
score
'
]
=
outputs
[
0
]
outs_dict
[
'f_
geo
'
]
=
outputs
[
1
]
outs_dict
[
'f_
geo
'
]
=
outputs
[
0
]
outs_dict
[
'f_
score
'
]
=
outputs
[
1
]
else
:
outs_dict
[
'maps'
]
=
outputs
[
0
]
dt_boxes_list
=
self
.
postprocess_op
(
outs_dict
,
[
ratio_list
])
...
...
@@ -169,14 +169,7 @@ if __name__ == "__main__":
total_time
+=
elapse
count
+=
1
print
(
"Predict time of %s:"
%
image_file
,
elapse
)
img
=
cv2
.
cvtColor
(
img
,
cv2
.
COLOR_BGR2RGB
)
draw_img
=
draw_ocr
(
img
,
dt_boxes
,
None
,
None
,
False
)
draw_img_save
=
"./inference_results/"
if
not
os
.
path
.
exists
(
draw_img_save
):
os
.
makedirs
(
draw_img_save
)
cv2
.
imwrite
(
os
.
path
.
join
(
draw_img_save
,
os
.
path
.
basename
(
image_file
)),
draw_img
[:,
:,
::
-
1
])
print
(
"The visualized image saved in {}"
.
format
(
os
.
path
.
join
(
draw_img_save
,
os
.
path
.
basename
(
image_file
))))
"""
add visualized code
"""
print
(
"Avg Time:"
,
total_time
/
(
count
-
1
))
tools/infer/predict_rec.py
浏览文件 @
72ebbf2d
...
...
@@ -114,7 +114,6 @@ if __name__ == "__main__":
valid_image_file_list
.
append
(
image_file
)
img_list
.
append
(
img
)
rec_res
,
predict_time
=
text_recognizer
(
img_list
)
rec_res
,
predict_time
=
text_recognizer
(
img_list
)
for
ino
in
range
(
len
(
img_list
)):
print
(
"Predicts of %s:%s"
%
(
valid_image_file_list
[
ino
],
rec_res
[
ino
]))
print
(
"Total predict time for %d images:%.3f"
%
...
...
tools/program.py
浏览文件 @
72ebbf2d
...
...
@@ -191,8 +191,8 @@ def build_export(config, main_prog, startup_prog):
func_infor
=
config
[
'Architecture'
][
'function'
]
model
=
create_module
(
func_infor
)(
params
=
config
)
image
,
outputs
=
model
(
mode
=
'export'
)
fetches_var
=
[
outputs
[
name
]
for
name
in
outputs
]
fetches_var_name
=
[
name
for
name
in
outputs
]
fetches_var
=
sorted
([
outputs
[
name
]
for
name
in
outputs
])
fetches_var_name
=
[
name
for
name
in
fetches_var
]
feeded_var_names
=
[
image
.
name
]
target_vars
=
fetches_var
return
feeded_var_names
,
target_vars
,
fetches_var_name
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录