import logging from .detection_api import DetectionApi from .detector_config import ( PixelFormatEnum, InputTensorEnum, ModelConfig, ) from .detector_types import DetectorTypeEnum, api_types, DetectorConfig 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)