__init__.pyi 657 B

1234567891011121314151617181920212223242526272829303132
  1. import cv2.typing
  2. import typing
  3. # Enumerations
  4. TraitAs_TENSOR: int
  5. TRAIT_AS_TENSOR: int
  6. TraitAs_IMAGE: int
  7. TRAIT_AS_IMAGE: int
  8. TraitAs = int
  9. """One of [TraitAs_TENSOR, TRAIT_AS_TENSOR, TraitAs_IMAGE, TRAIT_AS_IMAGE]"""
  10. # Classes
  11. class PyParams:
  12. # Functions
  13. @typing.overload
  14. def __init__(self) -> None: ...
  15. @typing.overload
  16. def __init__(self, tag: str, model_path: str) -> None: ...
  17. def cfgMeanStd(self, layer_name: str, m: cv2.typing.Scalar, s: cv2.typing.Scalar) -> PyParams: ...
  18. def cfgNormalize(self, layer_name: str, flag: bool) -> PyParams: ...
  19. # Functions
  20. def params(tag: str, model_path: str) -> PyParams: ...