add parameter enable_mkldnn

This commit is contained in:
dyning 2020-07-14 14:39:00 +08:00
parent 814dc2fa37
commit 4e427410a0
1 changed files with 1 additions and 1 deletions

View File

@ -125,7 +125,7 @@ def main(args):
continue
starttime = time.time()
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)
dt_boxes, rec_res = text_sys(img)
elapse = time.time() - starttime