__init__.pyi 2.5 KB

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