[Task]: MlFlow support

[Ravio1i]

What is the motivation for this task?

We are using MlFlow. Is it planned to support sending logs to mlflow?

Describe the solution you'd like

  logger: mlflow

Additional context

No response

[samet-akcay]

@Ravio1i , to do this, we just need to implement the AnomalibMLFlow logger class that is similar to following logger implementations:

• AnomalibWandbLogger
• AnomalibTensorBoardLogger
• AnomalibCometLogger

It would inherit Anomalib's ImageLoggerBase and Lightning's Mlflow logger, and implement add_image(self, image: np.ndarray | Figure, name: str | None = None, **kwargs: Any) method.

from pytorch_lightning.loggers import Mlflow
from anomalib.utils.loggers.base import ImageLoggerBase


class AnomalibMlflow(ImageLoggerBase, Mlflow):
    
    def __init__():
        ...

    @rank_zero_only
    def add_image(self, image: np.ndarray | Figure, name: str | None = None, **kwargs: Any):
        ...

[ninafiona]

Can you create a merge request? I'm also using it. 🙏

[ggiret-thinkdeep]

Hello
Any news about this ? I'm interested too ! :)

[SilverStarCoder]

@samet-akcay we are also using MlFlow, this would be a great addition 🙏

[samet-akcay]

yeah, we would like to add MLFlow support

[samet-akcay]

Closing this as it is being addressed here #1759