Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
天天代码码天天
PaddleOCR
提交
d671fefd
P
PaddleOCR
项目概览
天天代码码天天
/
PaddleOCR
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleOCR
通知
1
Star
1
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,发现更多精彩内容 >>
未验证
提交
d671fefd
编写于
7月 14, 2020
作者:
D
dyning
提交者:
GitHub
7月 14, 2020
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #327 from MissPenguin/develop
add parameter enable_mkldnn
上级
21c4934e
4e427410
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
20 addition
and
5 deletion
+20
-5
doc/doc_ch/benchmark.md
doc/doc_ch/benchmark.md
+11
-3
tools/infer/predict_system.py
tools/infer/predict_system.py
+4
-0
tools/infer/utility.py
tools/infer/utility.py
+5
-2
未找到文件。
doc/doc_ch/benchmark.md
浏览文件 @
d671fefd
...
...
@@ -16,6 +16,14 @@
|480|0.057|0.043|0.241| 0.175|
说明:
-
评估耗时阶段为图像输入到结果输出的完整阶段,包括了图像的预处理和后处理。
-
`Intel至强6148`
为服务器端CPU型号,目前正在调试使用Intel MKL-DNN 进一步加速CPU预测速度。
-
`骁龙855`
为移动端处理平台型号。
-
评估耗时阶段为图像输入到结果输出的完整阶段,包括了图像的预处理和后处理。
-
`Intel至强6148`
为服务器端CPU型号,测试中使用Intel MKL-DNN 加速CPU预测速度,使用该操作需要:
-
更新到飞桨latest版本:https://www.paddlepaddle.org.cn/documentation/docs/zh/install/Tables.html#whl-dev ,请根据自己环境的CUDA版本和Python版本选择相应的mkl版wheel包,如,CUDA10、Python3.7环境,应操作:
```
shell
# 获取安装包
wget https://paddle-wheel.bj.bcebos.com/0.0.0-gpu-cuda10-cudnn7-mkl/paddlepaddle_gpu-0.0.0-cp37-cp37m-linux_x86_64.whl
# 安装
pip3.7
install
paddlepaddle_gpu-0.0.0-cp37-cp37m-linux_x86_64.whl
```
-
预测时使用参数打开加速开关:
`--enable_mkldnn True`
-
`骁龙855`
为移动端处理平台型号。
tools/infer/predict_system.py
浏览文件 @
d671fefd
...
...
@@ -117,12 +117,16 @@ def main(args):
image_file_list
=
get_image_file_list
(
args
.
image_dir
)
text_sys
=
TextSystem
(
args
)
is_visualize
=
True
tackle_img_num
=
0
for
image_file
in
image_file_list
:
img
=
cv2
.
imread
(
image_file
)
if
img
is
None
:
logger
.
info
(
"error in loading image:{}"
.
format
(
image_file
))
continue
starttime
=
time
.
time
()
tackle_img_num
+=
1
if
not
args
.
use_gpu
and
args
.
enable_mkldnn
and
tackle_img_num
%
30
==
0
:
text_sys
=
TextSystem
(
args
)
dt_boxes
,
rec_res
=
text_sys
(
img
)
elapse
=
time
.
time
()
-
starttime
print
(
"Predict time of %s: %.3fs"
%
(
image_file
,
elapse
))
...
...
tools/infer/utility.py
浏览文件 @
d671fefd
...
...
@@ -64,6 +64,7 @@ def parse_args():
type
=
str
,
default
=
"./ppocr/utils/ppocr_keys_v1.txt"
)
parser
.
add_argument
(
"--use_space_char"
,
type
=
bool
,
default
=
True
)
parser
.
add_argument
(
"--enable_mkldnn"
,
type
=
bool
,
default
=
False
)
return
parser
.
parse_args
()
...
...
@@ -91,8 +92,10 @@ def create_predictor(args, mode):
config
.
enable_use_gpu
(
args
.
gpu_mem
,
0
)
else
:
config
.
disable_gpu
()
# config.enable_mkldnn()
config
.
set_cpu_math_library_num_threads
(
4
)
config
.
set_cpu_math_library_num_threads
(
6
)
if
args
.
enable_mkldnn
:
config
.
enable_mkldnn
()
#config.enable_memory_optim()
config
.
disable_glog_info
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录