__init__.pyi 2.6 KB

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