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, 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): 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) # 文本识别器 self.use_angle_cls = args.use_angle_cls if self.use_angle_cls: # 是否使用分类 self.text_classifier = predict_cls.TextClassifier(args) # 使用分类器