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)