__init__.pyi 2.6 KB

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