From 7eede6b44bc828acae0e87fb382e8b906a3c882e Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Tue, 1 Dec 2020 15:44:04 +0800 Subject: [PATCH] use cls_batch_num replace rec_batch_num --- tools/infer/predict_cls.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/predict_cls.py b/tools/infer/predict_cls.py index 7d7e4720..9ec03396 100755 --- a/tools/infer/predict_cls.py +++ b/tools/infer/predict_cls.py @@ -37,7 +37,7 @@ logger = get_logger() class TextClassifier(object): def __init__(self, args): self.cls_image_shape = [int(v) for v in args.cls_image_shape.split(",")] - self.cls_batch_num = args.rec_batch_num + self.cls_batch_num = args.cls_batch_num self.cls_thresh = args.cls_thresh self.use_zero_copy_run = args.use_zero_copy_run postprocess_params = { -- GitLab