Merge pull request #3779 from LDOUBLEV/fix_trt_res

Fix trt rec warmup
2021-08-24 17:23:20 +08:00 · 2021-08-24 17:23:20 +08:00 · 45dff759f8
parent c16b6ccb73 3691291ae1
commit 45dff759f8
1 changed files with 1 additions and 1 deletions
--- a/tools/infer/predict_rec.py
+++ b/tools/infer/predict_rec.py
@ -278,7 +278,7 @@ def main(args):
    if args.warmup:
        img = np.random.uniform(0, 255, [32, 320, 3]).astype(np.uint8)
        for i in range(2):
-            res = text_recognizer([img])
+            res = text_recognizer([img] * int(args.rec_batch_num))

    for image_file in image_file_list:
        img, flag = check_and_read_gif(image_file)