Skip to content
Snippets Groups Projects
Commit d11befff authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[engine] Set logger on module level

parent e4695064
No related branches found
No related tags found
1 merge request!12Streamlining
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# only use to evaluate 2nd human annotator
# # only used to evaluate 2nd human annotator
import os import os
import logging
import numpy as np import numpy as np
import torch import torch
import pandas as pd import pandas as pd
...@@ -17,6 +17,9 @@ from bob.ip.binseg.utils.plot import ( ...@@ -17,6 +17,9 @@ from bob.ip.binseg.utils.plot import (
from PIL import Image from PIL import Image
from torchvision.transforms.functional import to_tensor from torchvision.transforms.functional import to_tensor
import logging
logger = logging.getLogger(__name__)
def batch_metrics(predictions, ground_truths, names, output_folder, logger): def batch_metrics(predictions, ground_truths, names, output_folder, logger):
""" """
...@@ -116,7 +119,6 @@ def do_eval( ...@@ -116,7 +119,6 @@ def do_eval(
device to use ``'cpu'`` or ``'cuda'`` device to use ``'cpu'`` or ``'cuda'``
output_folder : str output_folder : str
""" """
logger = logging.getLogger("bob.ip.binseg.engine.evaluate")
logger.info("Start evaluation") logger.info("Start evaluation")
logger.info("Prediction folder {}".format(prediction_folder)) logger.info("Prediction folder {}".format(prediction_folder))
results_subfolder = os.path.join(output_folder, "results") results_subfolder = os.path.join(output_folder, "results")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment