Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

logging change

Browse Source
This commit is contained in:
Matthias Bisping 2022-04-04 23:37:49 +02:00
parent ce69f7d160
commit fdb7ebe618
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
image_prediction/utils/logger.py
View File

@ -12,8 +12,8 @@ def make_logger_getter():
handler = logging.StreamHandler() handler = logging.StreamHandler()
handler.setLevel(CONFIG.service.logging_level) handler.setLevel(CONFIG.service.logging_level)
log_format = "[%(levelname)s]: %(message)s" log_format = "%(asctime)s %(levelname)-8s %(message)s"
formatter = logging.Formatter(log_format) formatter = logging.Formatter(log_format, datefmt="%Y-%m-%d %H:%M:%S")
handler.setFormatter(formatter) handler.setFormatter(formatter)
logger.addHandler(handler) logger.addHandler(handler)

4
scripts/run_pipeline.py
View File

@ -4,8 +4,11 @@ import os
from glob import glob from glob import glob
from image_prediction.pipeline import load_pipeline from image_prediction.pipeline import load_pipeline
from image_prediction.utils import get_logger
from image_prediction.utils.pdf_annotation import annotate_pdf from image_prediction.utils.pdf_annotation import annotate_pdf
logger = get_logger()
def parse_args(): def parse_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
@ -21,6 +24,7 @@ def parse_args():
def process_pdf(pipeline, pdf_path): def process_pdf(pipeline, pdf_path):
with open(pdf_path, "rb") as f: with open(pdf_path, "rb") as f:
logger.info(f"Processing {pdf_path}")
predictions = list(pipeline(f.read())) predictions = list(pipeline(f.read()))
annotate_pdf( annotate_pdf(