File size: 557 Bytes
3de7bf6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
"""Base logger for image logging consistency across all loggers used in anomalib."""
# Copyright (C) 2022-2024 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
from abc import abstractmethod
import numpy as np
from matplotlib.figure import Figure
class ImageLoggerBase:
"""Adds a common interface for logging the images."""
@abstractmethod
def add_image(self, image: np.ndarray | Figure, name: str | None = None, **kwargs) -> None:
"""Interface to log images in the respective loggers."""
raise NotImplementedError
|