__init__.pyi 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. __all__: list[str] = []
  2. import cv2.typing
  3. import typing as _typing
  4. # Classes
  5. class CoreML:
  6. # Functions
  7. def __init__(self) -> None: ...
  8. def cfgUseCPUOnly(self) -> CoreML: ...
  9. def cfgEnableOnSubgraph(self) -> CoreML: ...
  10. def cfgEnableOnlyNeuralEngine(self) -> CoreML: ...
  11. class CUDA:
  12. # Functions
  13. @_typing.overload
  14. def __init__(self) -> None: ...
  15. @_typing.overload
  16. def __init__(self, dev_id: int) -> None: ...
  17. class TensorRT:
  18. # Functions
  19. @_typing.overload
  20. def __init__(self) -> None: ...
  21. @_typing.overload
  22. def __init__(self, dev_id: int) -> None: ...
  23. class OpenVINO:
  24. # Functions
  25. @_typing.overload
  26. def __init__(self) -> None: ...
  27. @_typing.overload
  28. def __init__(self, dev_type: str) -> None: ...
  29. @_typing.overload
  30. def __init__(self, params: cv2.typing.map_string_and_string) -> None: ...
  31. def cfgCacheDir(self, dir: str) -> OpenVINO: ...
  32. def cfgNumThreads(self, nthreads: int) -> OpenVINO: ...
  33. def cfgEnableOpenCLThrottling(self) -> OpenVINO: ...
  34. def cfgEnableDynamicShapes(self) -> OpenVINO: ...
  35. class DirectML:
  36. # Functions
  37. @_typing.overload
  38. def __init__(self) -> None: ...
  39. @_typing.overload
  40. def __init__(self, device_id: int) -> None: ...
  41. @_typing.overload
  42. def __init__(self, adapter_name: str) -> None: ...