Merge pull request #3779 from LDOUBLEV/fix_trt_res

Fix trt rec warmup

Merge pull request #3779 from LDOUBLEV/fix_trt_res
Fix trt rec warmup
45dff759 · xiaoting · GitHub · c16b6ccb · 3691291a · 45dff759
Unverified Commit 45dff759 authored Aug 24, 2021 by xiaoting Committed by GitHub Aug 24, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tools/infer/predict_rec.py tools/infer/predict_rec.py +1 -1

No files found.
--- 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)