mmpl/evaluation/metrics/builder.py

import copy
import inspect
from typing import List, Union

import torch
import torch.nn as nn
import lightning
import torchmetrics
import torchmetrics.detection

from mmengine.config import Config, ConfigDict
from mmpl.registry import METRICS


def register_pl_metrics() -> List[str]:
    """Register loggers in ``lightning.pytorch.loggers`` to the ``LOGGERS`` registry.

    Returns:
        List[str]: A list of registered optimizers' name.
    """
    pl_metrics = []
    for modules in [torchmetrics, torchmetrics.detection]:
        for module_name in dir(modules):
            if module_name.startswith('__'):
                continue
            _metric = getattr(modules, module_name)
            if inspect.isclass(_metric):
                METRICS.register_module(module=_metric)
                pl_metrics.append(module_name)
    return pl_metrics


PL_METRICS = register_pl_metrics()