From 65b4cc0cf336efb1e6d1ed9ad2f91bd3fbbd1866 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 17 Dec 2020 20:58:23 +0800 Subject: [PATCH] adjust rec_batch_num as 1 --- tools/infer/utility.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 4b06b60b..1b5e6b52 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -62,7 +62,7 @@ def parse_args(): parser.add_argument("--rec_model_dir", type=str) parser.add_argument("--rec_image_shape", type=str, default="3, 32, 320") parser.add_argument("--rec_char_type", type=str, default='ch') - parser.add_argument("--rec_batch_num", type=int, default=6) + parser.add_argument("--rec_batch_num", type=int, default=1) parser.add_argument("--max_text_length", type=int, default=25) parser.add_argument( "--rec_char_dict_path", -- GitLab