From 2985e7b89747b635c41bf2e2fcf7c99d82682bc0 Mon Sep 17 00:00:00 2001
From: zhoujun <zjwenmu@gmail.com>
Date: Tue, 1 Dec 2020 22:22:41 -0600
Subject: [PATCH] use cls_batch_num replace rec_batch_num (#1298)

---
 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