use cls_batch_num replace rec_batch_num (#1298)

release/2.0-rc1-0
zhoujun 4 years ago committed by GitHub
parent 9b2c0e4838
commit 2985e7b897
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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 = {

Loading…
Cancel
Save