提交 df79294c 编写于 作者: D dyning

add parameter enable_mkldnn

...@@ -18,4 +18,4 @@ ...@@ -18,4 +18,4 @@
说明: 说明:
- 评估耗时阶段为图像输入到结果输出的完整阶段,包括了图像的预处理和后处理。 - 评估耗时阶段为图像输入到结果输出的完整阶段,包括了图像的预处理和后处理。
- `Intel至强6148`为服务器端CPU型号,目前正在调试使用Intel MKL-DNN 进一步加速CPU预测速度。 - `Intel至强6148`为服务器端CPU型号,目前正在调试使用Intel MKL-DNN 进一步加速CPU预测速度。
- `骁龙855`为移动端处理平台型号。 - `骁龙855`为移动端处理平台型号。
\ No newline at end of file
...@@ -64,6 +64,7 @@ def parse_args(): ...@@ -64,6 +64,7 @@ def parse_args():
type=str, type=str,
default="./ppocr/utils/ppocr_keys_v1.txt") default="./ppocr/utils/ppocr_keys_v1.txt")
parser.add_argument("--use_space_char", type=bool, default=True) parser.add_argument("--use_space_char", type=bool, default=True)
parser.add_argument("--enable_mkldnn", type=bool, default=False)
return parser.parse_args() return parser.parse_args()
...@@ -91,8 +92,10 @@ def create_predictor(args, mode): ...@@ -91,8 +92,10 @@ def create_predictor(args, mode):
config.enable_use_gpu(args.gpu_mem, 0) config.enable_use_gpu(args.gpu_mem, 0)
else: else:
config.disable_gpu() config.disable_gpu()
# config.enable_mkldnn() config.set_cpu_math_library_num_threads(6)
config.set_cpu_math_library_num_threads(4) if args.enable_mkldnn:
config.enable_mkldnn()
#config.enable_memory_optim() #config.enable_memory_optim()
config.disable_glog_info() config.disable_glog_info()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册