__init__.pyi 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210
  1. __all__: list[str] = []
  2. import cv2
  3. import cv2.typing
  4. import typing as _typing
  5. # Classes
  6. class Map:
  7. # Functions
  8. @_typing.overload
  9. def warp(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  10. @_typing.overload
  11. def warp(self, img1: cv2.UMat, img2: cv2.UMat | None = ...) -> cv2.UMat: ...
  12. @_typing.overload
  13. def inverseWarp(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  14. @_typing.overload
  15. def inverseWarp(self, img1: cv2.UMat, img2: cv2.UMat | None = ...) -> cv2.UMat: ...
  16. def inverseMap(self) -> Map: ...
  17. def compose(self, map: Map) -> None: ...
  18. def scale(self, factor: float) -> None: ...
  19. class MapAffine(Map):
  20. # Functions
  21. @_typing.overload
  22. def __init__(self) -> None: ...
  23. @_typing.overload
  24. def __init__(self, linTr: cv2.typing.MatLike, shift: cv2.typing.MatLike) -> None: ...
  25. @_typing.overload
  26. def __init__(self, linTr: cv2.UMat, shift: cv2.UMat) -> None: ...
  27. @_typing.overload
  28. def inverseWarp(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  29. @_typing.overload
  30. def inverseWarp(self, img1: cv2.UMat, img2: cv2.UMat | None = ...) -> cv2.UMat: ...
  31. def inverseMap(self) -> Map: ...
  32. def compose(self, map: Map) -> None: ...
  33. def scale(self, factor: float) -> None: ...
  34. @_typing.overload
  35. def getLinTr(self, linTr: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  36. @_typing.overload
  37. def getLinTr(self, linTr: cv2.UMat | None = ...) -> cv2.UMat: ...
  38. @_typing.overload
  39. def getShift(self, shift: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  40. @_typing.overload
  41. def getShift(self, shift: cv2.UMat | None = ...) -> cv2.UMat: ...
  42. class Mapper:
  43. # Functions
  44. @_typing.overload
  45. def calculate(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike, init: Map = ...) -> Map: ...
  46. @_typing.overload
  47. def calculate(self, img1: cv2.UMat, img2: cv2.UMat, init: Map = ...) -> Map: ...
  48. def getMap(self) -> Map: ...
  49. class MapperGradAffine(Mapper):
  50. # Functions
  51. def __init__(self) -> None: ...
  52. @_typing.overload
  53. def calculate(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike, init: Map = ...) -> Map: ...
  54. @_typing.overload
  55. def calculate(self, img1: cv2.UMat, img2: cv2.UMat, init: Map = ...) -> Map: ...
  56. def getMap(self) -> Map: ...
  57. class MapperGradEuclid(Mapper):
  58. # Functions
  59. def __init__(self) -> None: ...
  60. @_typing.overload
  61. def calculate(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike, init: Map = ...) -> Map: ...
  62. @_typing.overload
  63. def calculate(self, img1: cv2.UMat, img2: cv2.UMat, init: Map = ...) -> Map: ...
  64. def getMap(self) -> Map: ...
  65. class MapperGradProj(Mapper):
  66. # Functions
  67. def __init__(self) -> None: ...
  68. @_typing.overload
  69. def calculate(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike, init: Map = ...) -> Map: ...
  70. @_typing.overload
  71. def calculate(self, img1: cv2.UMat, img2: cv2.UMat, init: Map = ...) -> Map: ...
  72. def getMap(self) -> Map: ...
  73. class MapperGradShift(Mapper):
  74. # Functions
  75. def __init__(self) -> None: ...
  76. @_typing.overload
  77. def calculate(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike, init: Map = ...) -> Map: ...
  78. @_typing.overload
  79. def calculate(self, img1: cv2.UMat, img2: cv2.UMat, init: Map = ...) -> Map: ...
  80. def getMap(self) -> Map: ...
  81. class MapperGradSimilar(Mapper):
  82. # Functions
  83. def __init__(self) -> None: ...
  84. @_typing.overload
  85. def calculate(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike, init: Map = ...) -> Map: ...
  86. @_typing.overload
  87. def calculate(self, img1: cv2.UMat, img2: cv2.UMat, init: Map = ...) -> Map: ...
  88. def getMap(self) -> Map: ...
  89. class MapperPyramid(Mapper):
  90. numLev_: int
  91. numIterPerScale_: int
  92. # Functions
  93. def __init__(self, baseMapper: Mapper) -> None: ...
  94. @_typing.overload
  95. def calculate(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike, init: Map = ...) -> Map: ...
  96. @_typing.overload
  97. def calculate(self, img1: cv2.UMat, img2: cv2.UMat, init: Map = ...) -> Map: ...
  98. def getMap(self) -> Map: ...
  99. class MapTypeCaster:
  100. # Functions
  101. @staticmethod
  102. def toAffine(sourceMap: Map) -> MapAffine: ...
  103. @staticmethod
  104. def toShift(sourceMap: Map) -> MapShift: ...
  105. @staticmethod
  106. def toProjec(sourceMap: Map) -> MapProjec: ...
  107. class MapProjec(Map):
  108. # Functions
  109. @_typing.overload
  110. def __init__(self) -> None: ...
  111. @_typing.overload
  112. def __init__(self, projTr: cv2.typing.MatLike) -> None: ...
  113. @_typing.overload
  114. def __init__(self, projTr: cv2.UMat) -> None: ...
  115. @_typing.overload
  116. def inverseWarp(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  117. @_typing.overload
  118. def inverseWarp(self, img1: cv2.UMat, img2: cv2.UMat | None = ...) -> cv2.UMat: ...
  119. def inverseMap(self) -> Map: ...
  120. def compose(self, map: Map) -> None: ...
  121. def scale(self, factor: float) -> None: ...
  122. @_typing.overload
  123. def getProjTr(self, projTr: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  124. @_typing.overload
  125. def getProjTr(self, projTr: cv2.UMat | None = ...) -> cv2.UMat: ...
  126. def normalize(self) -> None: ...
  127. class MapShift(Map):
  128. # Functions
  129. @_typing.overload
  130. def __init__(self) -> None: ...
  131. @_typing.overload
  132. def __init__(self, shift: cv2.typing.MatLike) -> None: ...
  133. @_typing.overload
  134. def __init__(self, shift: cv2.UMat) -> None: ...
  135. @_typing.overload
  136. def inverseWarp(self, img1: cv2.typing.MatLike, img2: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  137. @_typing.overload
  138. def inverseWarp(self, img1: cv2.UMat, img2: cv2.UMat | None = ...) -> cv2.UMat: ...
  139. def inverseMap(self) -> Map: ...
  140. def compose(self, map: Map) -> None: ...
  141. def scale(self, factor: float) -> None: ...
  142. @_typing.overload
  143. def getShift(self, shift: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  144. @_typing.overload
  145. def getShift(self, shift: cv2.UMat | None = ...) -> cv2.UMat: ...