blakeblackshear.frigate/frigate/detectors/__init__.py

import logging

from .detector_config import InputTensorEnum, ModelConfig, PixelFormatEnum  # noqa: F401
from .detector_types import DetectorConfig, DetectorTypeEnum, api_types  # noqa: F401

logger = logging.getLogger(__name__)


def create_detector(detector_config):
    if detector_config.type == DetectorTypeEnum.cpu:
        logger.warning(
            "CPU detectors are not recommended and should only be used for testing or for trial purposes."
        )

    api = api_types.get(detector_config.type)
    if not api:
        raise ValueError(detector_config.type)
    return api(detector_config)