diff --git a/paddleocr.py b/paddleocr.py index 958968174d9648e6b7cd9feeb4ae8e3a80b4095e..f2a3496897c07f8d969b198441076ef992774c19 100644 --- a/paddleocr.py +++ b/paddleocr.py @@ -151,8 +151,8 @@ class PaddleOCR(predict_system.TextSystem): """ params = parse_args(mMain=False) params.__dict__.update(**kwargs) - if params.show_log: - logger.setLevel(logging.DEBUG) + if not params.show_log: + logger.setLevel(logging.INFO) self.use_angle_cls = params.use_angle_cls lang = params.lang latin_lang = [ diff --git a/ppocr/utils/logging.py b/ppocr/utils/logging.py index 951141db8f39acac612029c8b69f4a29a0ab27ce..11896c37d9285e19a9526caa9c637d7eda7b1979 100644 --- a/ppocr/utils/logging.py +++ b/ppocr/utils/logging.py @@ -22,7 +22,7 @@ logger_initialized = {} @functools.lru_cache() -def get_logger(name='root', log_file=None, log_level=logging.INFO): +def get_logger(name='root', log_file=None, log_level=logging.DEBUG): """Initialize and get a logger by name. If the logger has not been initialized, this method will initialize the logger by adding one or two handlers, otherwise the initialized logger will diff --git a/ppstructure/paddlestructure.py b/ppstructure/paddlestructure.py index 57a53d6496f66771f1f6f7628751b4f0ac0fc3b5..f24234bb4831ced1c10a7bdfb4877f52f6932d1e 100644 --- a/ppstructure/paddlestructure.py +++ b/ppstructure/paddlestructure.py @@ -66,8 +66,8 @@ class PaddleStructure(OCRSystem): def __init__(self, **kwargs): params = parse_args(mMain=False) params.__dict__.update(**kwargs) - if params.show_log: - logger.setLevel(logging.DEBUG) + if not params.show_log: + logger.setLevel(logging.INFO) params.use_angle_cls = False # init model dir params.det_model_dir, det_url = confirm_model_dir_url(params.det_model_dir, diff --git a/ppstructure/predict_system.py b/ppstructure/predict_system.py index 9b305efee404b84a3a4ef8b2150e279860a1f19a..009c20521fc833d1da39699d6b39ba290cda81d0 100644 --- a/ppstructure/predict_system.py +++ b/ppstructure/predict_system.py @@ -24,6 +24,7 @@ os.environ["FLAGS_allocator_strategy"] = 'auto_growth' import cv2 import numpy as np import time +import logging import layoutparser as lp @@ -31,7 +32,7 @@ from ppocr.utils.utility import get_image_file_list, check_and_read_gif from ppocr.utils.logging import get_logger from tools.infer.predict_system import TextSystem from ppstructure.table.predict_table import TableSystem, to_excel -from ppstructure.utility import parse_args,draw_result +from ppstructure.utility import parse_args, draw_result logger = get_logger() @@ -40,6 +41,8 @@ class OCRSystem(object): def __init__(self, args): args.det_limit_type = 'resize_long' args.drop_score = 0 + if not args.show_log: + logger.setLevel(logging.INFO) self.text_system = TextSystem(args) self.table_system = TableSystem(args, self.text_system.text_detector, self.text_system.text_recognizer) self.table_layout = lp.PaddleDetectionLayoutModel("lp://PubLayNet/ppyolov2_r50vd_dcn_365e_publaynet/config", @@ -67,6 +70,7 @@ class OCRSystem(object): res_list.append({'type': region.type, 'bbox': [x1, y1, x2, y2], 'res': res}) return res_list + def save_res(res, save_folder, img_name): excel_save_folder = os.path.join(save_folder, img_name) os.makedirs(excel_save_folder, exist_ok=True) @@ -105,7 +109,7 @@ def main(args): starttime = time.time() res = structure_sys(img) save_res(res, save_folder, img_name) - draw_img = draw_result(img,res, args.vis_font_path) + draw_img = draw_result(img, res, args.vis_font_path) cv2.imwrite(os.path.join(save_folder, img_name, 'show.jpg'), draw_img) logger.info('result save to {}'.format(os.path.join(save_folder, img_name))) elapse = time.time() - starttime diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index ad1b7d4ef432924f1781a16eae07c171e571826b..7361d27f7eb05ce4644ebb52beda5c7e5d64a40d 100755 --- a/tools/infer/predict_system.py +++ b/tools/infer/predict_system.py @@ -24,6 +24,7 @@ import cv2 import copy import numpy as np import time +import logging from PIL import Image import tools.infer.utility as utility import tools.infer.predict_rec as predict_rec @@ -38,6 +39,9 @@ logger = get_logger() class TextSystem(object): 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 @@ -88,7 +92,7 @@ class TextSystem(object): ori_im = img.copy() dt_boxes, elapse = self.text_detector(img) - logger.info("dt_boxes num : {}, elapse : {}".format( + logger.debug("dt_boxes num : {}, elapse : {}".format( len(dt_boxes), elapse)) if dt_boxes is None: @@ -104,11 +108,11 @@ class TextSystem(object): if self.use_angle_cls and cls: img_crop_list, angle_list, elapse = self.text_classifier( img_crop_list) - logger.info("cls num : {}, elapse : {}".format( + logger.debug("cls num : {}, elapse : {}".format( len(img_crop_list), elapse)) rec_res, elapse = self.text_recognizer(img_crop_list) - logger.info("rec_res num : {}, elapse : {}".format( + logger.debug("rec_res num : {}, elapse : {}".format( len(rec_res), elapse)) # self.print_draw_crop_rec_res(img_crop_list, rec_res) filter_boxes, filter_rec_res = [], []