__init__.pyi 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. __all__: list[str] = []
  2. import cv2
  3. import cv2.typing
  4. import os
  5. import typing as _typing
  6. # Enumerations
  7. FLANN_INDEX_TYPE_8U: int
  8. FLANN_INDEX_TYPE_8S: int
  9. FLANN_INDEX_TYPE_16U: int
  10. FLANN_INDEX_TYPE_16S: int
  11. FLANN_INDEX_TYPE_32S: int
  12. FLANN_INDEX_TYPE_32F: int
  13. FLANN_INDEX_TYPE_64F: int
  14. FLANN_INDEX_TYPE_STRING: int
  15. FLANN_INDEX_TYPE_BOOL: int
  16. FLANN_INDEX_TYPE_ALGORITHM: int
  17. LAST_VALUE_FLANN_INDEX_TYPE: int
  18. FlannIndexType = int
  19. """One of [FLANN_INDEX_TYPE_8U, FLANN_INDEX_TYPE_8S, FLANN_INDEX_TYPE_16U, FLANN_INDEX_TYPE_16S, FLANN_INDEX_TYPE_32S, FLANN_INDEX_TYPE_32F, FLANN_INDEX_TYPE_64F, FLANN_INDEX_TYPE_STRING, FLANN_INDEX_TYPE_BOOL, FLANN_INDEX_TYPE_ALGORITHM, LAST_VALUE_FLANN_INDEX_TYPE]"""
  20. # Classes
  21. class Index:
  22. # Functions
  23. @_typing.overload
  24. def __init__(self) -> None: ...
  25. @_typing.overload
  26. def __init__(self, features: cv2.typing.MatLike, params: cv2.typing.IndexParams, distType: int = ...) -> None: ...
  27. @_typing.overload
  28. def __init__(self, features: cv2.UMat, params: cv2.typing.IndexParams, distType: int = ...) -> None: ...
  29. @_typing.overload
  30. def build(self, features: cv2.typing.MatLike, params: cv2.typing.IndexParams, distType: int = ...) -> None: ...
  31. @_typing.overload
  32. def build(self, features: cv2.UMat, params: cv2.typing.IndexParams, distType: int = ...) -> None: ...
  33. @_typing.overload
  34. def knnSearch(self, query: cv2.typing.MatLike, knn: int, indices: cv2.typing.MatLike | None = ..., dists: cv2.typing.MatLike | None = ..., params: cv2.typing.SearchParams = ...) -> tuple[cv2.typing.MatLike, cv2.typing.MatLike]: ...
  35. @_typing.overload
  36. def knnSearch(self, query: cv2.UMat, knn: int, indices: cv2.UMat | None = ..., dists: cv2.UMat | None = ..., params: cv2.typing.SearchParams = ...) -> tuple[cv2.UMat, cv2.UMat]: ...
  37. @_typing.overload
  38. def radiusSearch(self, query: cv2.typing.MatLike, radius: float, maxResults: int, indices: cv2.typing.MatLike | None = ..., dists: cv2.typing.MatLike | None = ..., params: cv2.typing.SearchParams = ...) -> tuple[int, cv2.typing.MatLike, cv2.typing.MatLike]: ...
  39. @_typing.overload
  40. def radiusSearch(self, query: cv2.UMat, radius: float, maxResults: int, indices: cv2.UMat | None = ..., dists: cv2.UMat | None = ..., params: cv2.typing.SearchParams = ...) -> tuple[int, cv2.UMat, cv2.UMat]: ...
  41. def save(self, filename: str | os.PathLike[str]) -> None: ...
  42. @_typing.overload
  43. def load(self, features: cv2.typing.MatLike, filename: str | os.PathLike[str]) -> bool: ...
  44. @_typing.overload
  45. def load(self, features: cv2.UMat, filename: str | os.PathLike[str]) -> bool: ...
  46. def release(self) -> None: ...
  47. def getDistance(self) -> int: ...
  48. def getAlgorithm(self) -> int: ...