use cls_batch_num replace rec_batch_num (#1298)
This commit is contained in:
parent
9b2c0e4838
commit
2985e7b897
|
@ -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…
Reference in New Issue