提交 4e427410 编写于 作者: D dyning

add parameter enable_mkldnn

上级 814dc2fa
...@@ -125,7 +125,7 @@ def main(args): ...@@ -125,7 +125,7 @@ def main(args):
continue continue
starttime = time.time() starttime = time.time()
tackle_img_num += 1 tackle_img_num += 1
if not args.use_gpu and tackle_img_num % 30 == 0: if not args.use_gpu and args.enable_mkldnn and tackle_img_num % 30 == 0:
text_sys = TextSystem(args) text_sys = TextSystem(args)
dt_boxes, rec_res = text_sys(img) dt_boxes, rec_res = text_sys(img)
elapse = time.time() - starttime elapse = time.time() - starttime
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册