wehub-resource-sync
19 lines
605 B
Python
19 lines
605 B
Python
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)
|