__init__.pyi 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. __all__: list[str] = []
  2. import cv2.typing
  3. import typing as _typing
  4. # Classes
  5. class PyParams:
  6. # Functions
  7. @_typing.overload
  8. def __init__(self) -> None: ...
  9. @_typing.overload
  10. def __init__(self, tag: str, model_path: str, bin_path: str, device: str) -> None: ...
  11. @_typing.overload
  12. def __init__(self, tag: str, blob_path: str, device: str) -> None: ...
  13. def cfgPluginConfig(self, config: cv2.typing.map_string_and_string) -> PyParams: ...
  14. @_typing.overload
  15. def cfgInputTensorLayout(self, tensor_layout: str) -> PyParams: ...
  16. @_typing.overload
  17. def cfgInputTensorLayout(self, layout_map: cv2.typing.map_string_and_string) -> PyParams: ...
  18. @_typing.overload
  19. def cfgInputModelLayout(self, tensor_layout: str) -> PyParams: ...
  20. @_typing.overload
  21. def cfgInputModelLayout(self, layout_map: cv2.typing.map_string_and_string) -> PyParams: ...
  22. @_typing.overload
  23. def cfgOutputTensorLayout(self, tensor_layout: str) -> PyParams: ...
  24. @_typing.overload
  25. def cfgOutputTensorLayout(self, layout_map: cv2.typing.map_string_and_string) -> PyParams: ...
  26. @_typing.overload
  27. def cfgOutputModelLayout(self, tensor_layout: str) -> PyParams: ...
  28. @_typing.overload
  29. def cfgOutputModelLayout(self, layout_map: cv2.typing.map_string_and_string) -> PyParams: ...
  30. @_typing.overload
  31. def cfgOutputTensorPrecision(self, precision: int) -> PyParams: ...
  32. @_typing.overload
  33. def cfgOutputTensorPrecision(self, precision_map: cv2.typing.map_string_and_int) -> PyParams: ...
  34. @_typing.overload
  35. def cfgReshape(self, new_shape: _typing.Sequence[int]) -> PyParams: ...
  36. @_typing.overload
  37. def cfgReshape(self, new_shape_map: cv2.typing.map_string_and_vector_size_t) -> PyParams: ...
  38. def cfgNumRequests(self, nireq: int) -> PyParams: ...
  39. @_typing.overload
  40. def cfgMean(self, mean_values: _typing.Sequence[float]) -> PyParams: ...
  41. @_typing.overload
  42. def cfgMean(self, mean_map: cv2.typing.map_string_and_vector_float) -> PyParams: ...
  43. @_typing.overload
  44. def cfgScale(self, scale_values: _typing.Sequence[float]) -> PyParams: ...
  45. @_typing.overload
  46. def cfgScale(self, scale_map: cv2.typing.map_string_and_vector_float) -> PyParams: ...
  47. @_typing.overload
  48. def cfgResize(self, interpolation: int) -> PyParams: ...
  49. @_typing.overload
  50. def cfgResize(self, interpolation: cv2.typing.map_string_and_int) -> PyParams: ...
  51. # Functions
  52. @_typing.overload
  53. def params(tag: str, model_path: str, weights: str, device: str) -> PyParams: ...
  54. @_typing.overload
  55. def params(tag: str, bin_path: str, device: str) -> PyParams: ...