diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py index 97dfa5214628123d0c9b7edd7d94060a2bfd2a1e..4c16f929352640f2b770fe2ff94fd4497f1308f5 100755 --- 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)