Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleOCR
提交
b8b5faaf
P
PaddleOCR
项目概览
PaddlePaddle
/
PaddleOCR
接近 1 年 前同步成功
通知
1518
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,发现更多精彩内容 >>
未验证
提交
b8b5faaf
编写于
12月 19, 2021
作者:
D
Double_V
提交者:
GitHub
12月 19, 2021
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #4968 from LDOUBLEV/sdmgr
add ips
上级
1b3cf0da
9b8c0ef3
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
20 addition
and
6 deletion
+20
-6
configs/kie/kie_unet_sdmgr.yml
configs/kie/kie_unet_sdmgr.yml
+3
-3
tools/infer_kie.py
tools/infer_kie.py
+17
-3
未找到文件。
configs/kie/kie_unet_sdmgr.yml
浏览文件 @
b8b5faaf
Global
:
use_gpu
:
True
epoch_num
:
30
0
epoch_num
:
6
0
log_smooth_window
:
20
print_batch_step
:
50
save_model_dir
:
./output/kie_5/
...
...
@@ -13,7 +13,7 @@ Global:
# you should set load_static_weights as False.
load_static_weights
:
False
cal_metric_during_train
:
False
pretrained_model
:
./output/kie_4/best_accuracy
pretrained_model
:
checkpoints
:
save_inference_dir
:
use_visualdl
:
False
...
...
@@ -108,4 +108,4 @@ Eval:
shuffle
:
False
drop_last
:
False
batch_size_per_card
:
1
# must be 1
num_workers
:
4
\ No newline at end of file
num_workers
:
4
tools/infer_kie.py
浏览文件 @
b8b5faaf
...
...
@@ -33,8 +33,9 @@ import paddle
from
ppocr.data
import
create_operators
,
transform
from
ppocr.modeling.architectures
import
build_model
from
ppocr.utils.save_load
import
init
_model
from
ppocr.utils.save_load
import
load
_model
import
tools.program
as
program
import
time
def
read_class_list
(
filepath
):
...
...
@@ -80,7 +81,8 @@ def draw_kie_result(batch, node, idx_to_cls, count):
vis_img
=
np
.
ones
((
h
,
w
*
3
,
3
),
dtype
=
np
.
uint8
)
*
255
vis_img
[:,
:
w
]
=
img
vis_img
[:,
w
:]
=
pred_img
save_kie_path
=
os
.
path
.
dirname
(
config
[
'Global'
][
'save_res_path'
])
+
"/kie_results/"
save_kie_path
=
os
.
path
.
dirname
(
config
[
'Global'
][
'save_res_path'
])
+
"/kie_results/"
if
not
os
.
path
.
exists
(
save_kie_path
):
os
.
makedirs
(
save_kie_path
)
save_path
=
os
.
path
.
join
(
save_kie_path
,
str
(
count
)
+
".png"
)
...
...
@@ -93,7 +95,7 @@ def main():
# build model
model
=
build_model
(
config
[
'Architecture'
])
init_model
(
config
,
model
,
logger
)
load_model
(
config
,
model
)
# create data ops
transforms
=
[]
...
...
@@ -111,10 +113,15 @@ def main():
os
.
makedirs
(
os
.
path
.
dirname
(
save_res_path
))
model
.
eval
()
warmup_times
=
0
count_t
=
[]
with
open
(
save_res_path
,
"wb"
)
as
fout
:
with
open
(
config
[
'Global'
][
'infer_img'
],
"rb"
)
as
f
:
lines
=
f
.
readlines
()
for
index
,
data_line
in
enumerate
(
lines
):
if
index
==
10
:
warmup_t
=
time
.
time
()
data_line
=
data_line
.
decode
(
'utf-8'
)
substr
=
data_line
.
strip
(
"
\n
"
).
split
(
"
\t
"
)
img_path
,
label
=
data_dir
+
"/"
+
substr
[
0
],
substr
[
1
]
...
...
@@ -122,16 +129,23 @@ def main():
with
open
(
data
[
'img_path'
],
'rb'
)
as
f
:
img
=
f
.
read
()
data
[
'image'
]
=
img
st
=
time
.
time
()
batch
=
transform
(
data
,
ops
)
batch_pred
=
[
0
]
*
len
(
batch
)
for
i
in
range
(
len
(
batch
)):
batch_pred
[
i
]
=
paddle
.
to_tensor
(
np
.
expand_dims
(
batch
[
i
],
axis
=
0
))
st
=
time
.
time
()
node
,
edge
=
model
(
batch_pred
)
node
=
F
.
softmax
(
node
,
-
1
)
count_t
.
append
(
time
.
time
()
-
st
)
draw_kie_result
(
batch
,
node
,
idx_to_cls
,
index
)
logger
.
info
(
"success!"
)
logger
.
info
(
"It took {} s for predict {} images."
.
format
(
np
.
sum
(
count_t
),
len
(
count_t
)))
ips
=
len
(
count_t
[
warmup_times
:])
/
np
.
sum
(
count_t
[
warmup_times
:])
logger
.
info
(
"The ips is {} images/s"
.
format
(
ips
))
if
__name__
==
'__main__'
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录