def __init__(self, args, language_list, logger):
self.logger = logger
self.text_detector = predict_det.TextDetector(args)
self.logger.info("初始化检测模型:{} 最大边:{} 路径:{}".format(
args.det_algorithm, args.det_max_side_len, args.det_model_dir))
self.language_map = args.language_map
self.text_recognizer_classifier = {}
for idx, language_m in enumerate(language_list):
self.logger.info("---------- {} -----------".format(language_m))
args.update(language_m)
self.logger.info("{} 更新识别语言{}的配置文件".format(idx, language_m))
text_recognizer = predict_rec.TextRecognizer(args)
self.logger.info(
"{} 初始化识别模型:{} 语言:{} 词典:{} 模型路径:{} 输入大小:{}, rec_batch_num:{}, max_text_length:{}"
.format(idx, args.rec_algorithm, args.rec_char_type,
args.rec_char_dict_path, args.rec_model_dir,
args.rec_image_shape, args.rec_batch_num,
args.max_text_length))
if args.use_angle_cls:
text_classifier = predict_cls.TextClassifier(args)
self.logger.info(
"{} 初始化方向分类模型路径:{}, 输入大小:{} batch_num:{}, cls_thresh:{}".
format(idx, args.cls_model_dir, args.cls_image_shape,
args.cls_batch_num, args.cls_thresh))
else:
text_classifier = None
self.logger.info("{} 未设置方向分类模型 use_angle_cls={}".format(
idx, args.use_angle_cls))
self.text_recognizer_classifier[language_m] = [
text_recognizer, text_classifier
]
def __init__(self, args):
self.text_detector = predict_det.TextDetector(args)
self.text_recognizer = predict_rec.TextRecognizer(args)
self.use_angle_cls = args.use_angle_cls
self.drop_score = args.drop_score
if self.use_angle_cls:
self.text_classifier = predict_cls.TextClassifier(args)
def __init__(self, args):
print(222)
self.text_detector = predict_det.TextDetector(args)
print(222)
self.text_recognizer = predict_rec.TextRecognizer(args)
print(222)
self.use_angle_cls = args.use_angle_cls
if self.use_angle_cls:
self.text_classifier = predict_cls.TextClassifier(args)
def __init__(self, args):
if not args.show_log:
logger.setLevel(logging.INFO)
self.text_detector = predict_det.TextDetector(args)
self.text_recognizer = predict_rec.TextRecognizer(args)
self.use_angle_cls = args.use_angle_cls
self.drop_score = args.drop_score
if self.use_angle_cls:
self.text_classifier = predict_cls.TextClassifier(args)
def __init__(self, config):
OCR_CFG = utility.get_default_config()
OCR_CFG.update(config)
self.cfg = OCR_CFG
self.text_detector = predict_det.TextDetector(OCR_CFG)
self.text_recognizer = predict_rec.TextRecognizer(OCR_CFG)
self.use_angle_cls = OCR_CFG['use_angle_cls']
self.drop_score = OCR_CFG['drop_score']
if self.use_angle_cls:
self.text_classifier = predict_cls.TextClassifier(OCR_CFG)
def __init__(self, args):
self.text_detector = predict_det.TextDetector(args)
self.text_recognizer = predict_rec.TextRecognizer(args)
def __init__(self, args, text_detector=None, text_recognizer=None):
self.text_detector = predict_det.TextDetector(
args) if text_detector is None else text_detector
self.text_recognizer = predict_rec.TextRecognizer(
args) if text_recognizer is None else text_recognizer
self.table_structurer = predict_strture.TableStructurer(args)
def __init__(self, args):
self.text_detector = predict_det.TextDetector(args) # 文本检测器
self.text_recognizer = predict_rec.TextRecognizer(args) # 文本识别器
self.use_angle_cls = args.use_angle_cls
if self.use_angle_cls: # 是否使用分类
self.text_classifier = predict_cls.TextClassifier(args) # 使用分类器
