diff --git a/doc/doc_ch/benchmark.md b/doc/doc_ch/benchmark.md index b562b53862c984d7f3b07c6face954276ca4d9ea..2b175950d6e424c56651819c3d6522f4530aad0e 100644 --- a/doc/doc_ch/benchmark.md +++ b/doc/doc_ch/benchmark.md @@ -18,4 +18,4 @@ 说明: - 评估耗时阶段为图像输入到结果输出的完整阶段,包括了图像的预处理和后处理。 - `Intel至强6148`为服务器端CPU型号,目前正在调试使用Intel MKL-DNN 进一步加速CPU预测速度。 -- `骁龙855`为移动端处理平台型号。 +- `骁龙855`为移动端处理平台型号。 \ No newline at end of file diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 0cf66d4c82c49e588eedd4855df4663040bcc4be..bab211a8dfa396b1aa8996cf7105452316441ed1 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -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()