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