forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

use cls_batch_num replace rec_batch_num (#1298)

This commit is contained in:
zhoujun 2020-12-01 22:22:41 -06:00 committed by GitHub
parent 9b2c0e4838
commit 2985e7b897
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/infer/predict_cls.py
View File

@ -37,7 +37,7 @@ logger = get_logger()
class TextClassifier(object): class TextClassifier(object):
def __init__(self, args): def __init__(self, args):
self.cls_image_shape = [int(v) for v in args.cls_image_shape.split(",")] 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.cls_thresh = args.cls_thresh
self.use_zero_copy_run = args.use_zero_copy_run self.use_zero_copy_run = args.use_zero_copy_run
postprocess_params = { postprocess_params = {