__init__.pyi 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. __all__: list[str] = []
  2. import cv2
  3. import cv2.typing
  4. import typing as _typing
  5. # Classes
  6. class GraphSegmentation(cv2.Algorithm):
  7. # Functions
  8. @_typing.overload
  9. def processImage(self, src: cv2.typing.MatLike, dst: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  10. @_typing.overload
  11. def processImage(self, src: cv2.UMat, dst: cv2.UMat | None = ...) -> cv2.UMat: ...
  12. def setSigma(self, sigma: float) -> None: ...
  13. def getSigma(self) -> float: ...
  14. def setK(self, k: float) -> None: ...
  15. def getK(self) -> float: ...
  16. def setMinSize(self, min_size: int) -> None: ...
  17. def getMinSize(self) -> int: ...
  18. class SelectiveSearchSegmentationStrategy(cv2.Algorithm):
  19. # Functions
  20. @_typing.overload
  21. def setImage(self, img: cv2.typing.MatLike, regions: cv2.typing.MatLike, sizes: cv2.typing.MatLike, image_id: int = ...) -> None: ...
  22. @_typing.overload
  23. def setImage(self, img: cv2.UMat, regions: cv2.UMat, sizes: cv2.UMat, image_id: int = ...) -> None: ...
  24. def get(self, r1: int, r2: int) -> float: ...
  25. def merge(self, r1: int, r2: int) -> None: ...
  26. class SelectiveSearchSegmentationStrategyColor(SelectiveSearchSegmentationStrategy):
  27. ...
  28. class SelectiveSearchSegmentationStrategySize(SelectiveSearchSegmentationStrategy):
  29. ...
  30. class SelectiveSearchSegmentationStrategyTexture(SelectiveSearchSegmentationStrategy):
  31. ...
  32. class SelectiveSearchSegmentationStrategyFill(SelectiveSearchSegmentationStrategy):
  33. ...
  34. class SelectiveSearchSegmentationStrategyMultiple(SelectiveSearchSegmentationStrategy):
  35. # Functions
  36. def addStrategy(self, g: SelectiveSearchSegmentationStrategy, weight: float) -> None: ...
  37. def clearStrategies(self) -> None: ...
  38. class SelectiveSearchSegmentation(cv2.Algorithm):
  39. # Functions
  40. @_typing.overload
  41. def setBaseImage(self, img: cv2.typing.MatLike) -> None: ...
  42. @_typing.overload
  43. def setBaseImage(self, img: cv2.UMat) -> None: ...
  44. def switchToSingleStrategy(self, k: int = ..., sigma: float = ...) -> None: ...
  45. def switchToSelectiveSearchFast(self, base_k: int = ..., inc_k: int = ..., sigma: float = ...) -> None: ...
  46. def switchToSelectiveSearchQuality(self, base_k: int = ..., inc_k: int = ..., sigma: float = ...) -> None: ...
  47. @_typing.overload
  48. def addImage(self, img: cv2.typing.MatLike) -> None: ...
  49. @_typing.overload
  50. def addImage(self, img: cv2.UMat) -> None: ...
  51. def clearImages(self) -> None: ...
  52. def addGraphSegmentation(self, g: GraphSegmentation) -> None: ...
  53. def clearGraphSegmentations(self) -> None: ...
  54. def addStrategy(self, s: SelectiveSearchSegmentationStrategy) -> None: ...
  55. def clearStrategies(self) -> None: ...
  56. def process(self) -> _typing.Sequence[cv2.typing.Rect]: ...
  57. # Functions
  58. def createGraphSegmentation(sigma: float = ..., k: float = ..., min_size: int = ...) -> GraphSegmentation: ...
  59. def createSelectiveSearchSegmentation() -> SelectiveSearchSegmentation: ...
  60. def createSelectiveSearchSegmentationStrategyColor() -> SelectiveSearchSegmentationStrategyColor: ...
  61. def createSelectiveSearchSegmentationStrategyFill() -> SelectiveSearchSegmentationStrategyFill: ...
  62. @_typing.overload
  63. def createSelectiveSearchSegmentationStrategyMultiple() -> SelectiveSearchSegmentationStrategyMultiple: ...
  64. @_typing.overload
  65. def createSelectiveSearchSegmentationStrategyMultiple(s1: SelectiveSearchSegmentationStrategy) -> SelectiveSearchSegmentationStrategyMultiple: ...
  66. @_typing.overload
  67. def createSelectiveSearchSegmentationStrategyMultiple(s1: SelectiveSearchSegmentationStrategy, s2: SelectiveSearchSegmentationStrategy) -> SelectiveSearchSegmentationStrategyMultiple: ...
  68. @_typing.overload
  69. def createSelectiveSearchSegmentationStrategyMultiple(s1: SelectiveSearchSegmentationStrategy, s2: SelectiveSearchSegmentationStrategy, s3: SelectiveSearchSegmentationStrategy) -> SelectiveSearchSegmentationStrategyMultiple: ...
  70. @_typing.overload
  71. def createSelectiveSearchSegmentationStrategyMultiple(s1: SelectiveSearchSegmentationStrategy, s2: SelectiveSearchSegmentationStrategy, s3: SelectiveSearchSegmentationStrategy, s4: SelectiveSearchSegmentationStrategy) -> SelectiveSearchSegmentationStrategyMultiple: ...
  72. def createSelectiveSearchSegmentationStrategySize() -> SelectiveSearchSegmentationStrategySize: ...
  73. def createSelectiveSearchSegmentationStrategyTexture() -> SelectiveSearchSegmentationStrategyTexture: ...