__init__.py 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180
  1. __all__ = [
  2. "IntPointer",
  3. "MatLike",
  4. "MatShape",
  5. "Size",
  6. "Size2f",
  7. "Scalar",
  8. "Point",
  9. "Point2i",
  10. "Point2f",
  11. "Point2d",
  12. "Point3i",
  13. "Point3f",
  14. "Point3d",
  15. "Range",
  16. "Rect",
  17. "Rect2i",
  18. "Rect2f",
  19. "Rect2d",
  20. "Moments",
  21. "RotatedRect",
  22. "TermCriteria",
  23. "Vec2i",
  24. "Vec2f",
  25. "Vec2d",
  26. "Vec3i",
  27. "Vec3f",
  28. "Vec3d",
  29. "Vec4i",
  30. "Vec4f",
  31. "Vec4d",
  32. "Vec6f",
  33. "FeatureDetector",
  34. "DescriptorExtractor",
  35. "FeatureExtractor",
  36. "Matx33f",
  37. "Matx33d",
  38. "Matx44f",
  39. "Matx44d",
  40. "LayerId",
  41. "LayerParams",
  42. "IndexParams",
  43. "SearchParams",
  44. "map_string_and_string",
  45. "map_string_and_int",
  46. "map_string_and_vector_size_t",
  47. "map_string_and_vector_float",
  48. "map_int_and_double",
  49. "GProtoArg",
  50. "GProtoInputArgs",
  51. "GProtoOutputArgs",
  52. "GRunArg",
  53. "GOptRunArg",
  54. "GMetaArg",
  55. "Prim",
  56. "GTypeInfo",
  57. "ExtractArgsCallback",
  58. "ExtractMetaCallback",
  59. ]
  60. import cv2.gapi.wip.draw
  61. import cv2.dnn
  62. import numpy
  63. import cv2.mat_wrapper
  64. import cv2
  65. import typing as _typing
  66. if _typing.TYPE_CHECKING:
  67. NumPyArrayNumeric = numpy.ndarray[_typing.Any, numpy.dtype[numpy.integer[_typing.Any] | numpy.floating[_typing.Any]]]
  68. else:
  69. NumPyArrayNumeric = numpy.ndarray
  70. if _typing.TYPE_CHECKING:
  71. NumPyArrayFloat32 = numpy.ndarray[_typing.Any, numpy.dtype[numpy.float32]]
  72. else:
  73. NumPyArrayFloat32 = numpy.ndarray
  74. if _typing.TYPE_CHECKING:
  75. NumPyArrayFloat64 = numpy.ndarray[_typing.Any, numpy.dtype[numpy.float64]]
  76. else:
  77. NumPyArrayFloat64 = numpy.ndarray
  78. if _typing.TYPE_CHECKING:
  79. TermCriteria_Type = cv2.TermCriteria_Type
  80. else:
  81. TermCriteria_Type = int
  82. IntPointer = int
  83. """Represents an arbitrary pointer"""
  84. MatLike = _typing.Union[cv2.mat_wrapper.Mat, NumPyArrayNumeric]
  85. MatShape = _typing.Sequence[int]
  86. Size = _typing.Sequence[int]
  87. """Required length is 2"""
  88. Size2f = _typing.Sequence[float]
  89. """Required length is 2"""
  90. Scalar = _typing.Union[_typing.Sequence[float], float]
  91. """Max sequence length is at most 4"""
  92. Point = _typing.Sequence[int]
  93. """Required length is 2"""
  94. Point2i = Point
  95. Point2f = _typing.Sequence[float]
  96. """Required length is 2"""
  97. Point2d = _typing.Sequence[float]
  98. """Required length is 2"""
  99. Point3i = _typing.Sequence[int]
  100. """Required length is 3"""
  101. Point3f = _typing.Sequence[float]
  102. """Required length is 3"""
  103. Point3d = _typing.Sequence[float]
  104. """Required length is 3"""
  105. Range = _typing.Sequence[int]
  106. """Required length is 2"""
  107. Rect = _typing.Sequence[int]
  108. """Required length is 4"""
  109. Rect2i = _typing.Sequence[int]
  110. """Required length is 4"""
  111. Rect2f = _typing.Sequence[float]
  112. """Required length is 4"""
  113. Rect2d = _typing.Sequence[float]
  114. """Required length is 4"""
  115. Moments = _typing.Dict[str, float]
  116. RotatedRect = _typing.Tuple[Point2f, Size2f, float]
  117. """Any type providing sequence protocol is supported"""
  118. TermCriteria = _typing.Tuple[TermCriteria_Type, int, float]
  119. """Any type providing sequence protocol is supported"""
  120. Vec2i = _typing.Sequence[int]
  121. """Required length is 2"""
  122. Vec2f = _typing.Sequence[float]
  123. """Required length is 2"""
  124. Vec2d = _typing.Sequence[float]
  125. """Required length is 2"""
  126. Vec3i = _typing.Sequence[int]
  127. """Required length is 3"""
  128. Vec3f = _typing.Sequence[float]
  129. """Required length is 3"""
  130. Vec3d = _typing.Sequence[float]
  131. """Required length is 3"""
  132. Vec4i = _typing.Sequence[int]
  133. """Required length is 4"""
  134. Vec4f = _typing.Sequence[float]
  135. """Required length is 4"""
  136. Vec4d = _typing.Sequence[float]
  137. """Required length is 4"""
  138. Vec6f = _typing.Sequence[float]
  139. """Required length is 6"""
  140. FeatureDetector = cv2.Feature2D
  141. DescriptorExtractor = cv2.Feature2D
  142. FeatureExtractor = cv2.Feature2D
  143. Matx33f = NumPyArrayFloat32
  144. """NDArray(shape=(3, 3), dtype=numpy.float32)"""
  145. Matx33d = NumPyArrayFloat64
  146. """NDArray(shape=(3, 3), dtype=numpy.float64)"""
  147. Matx44f = NumPyArrayFloat32
  148. """NDArray(shape=(4, 4), dtype=numpy.float32)"""
  149. Matx44d = NumPyArrayFloat64
  150. """NDArray(shape=(4, 4), dtype=numpy.float64)"""
  151. LayerId = cv2.dnn.DictValue
  152. LayerParams = _typing.Dict[str, _typing.Union[int, float, str]]
  153. IndexParams = _typing.Dict[str, _typing.Union[bool, int, float, str]]
  154. SearchParams = _typing.Dict[str, _typing.Union[bool, int, float, str]]
  155. map_string_and_string = _typing.Dict[str, str]
  156. map_string_and_int = _typing.Dict[str, int]
  157. map_string_and_vector_size_t = _typing.Dict[str, _typing.Sequence[int]]
  158. map_string_and_vector_float = _typing.Dict[str, _typing.Sequence[float]]
  159. map_int_and_double = _typing.Dict[int, float]
  160. GProtoArg = _typing.Union[Scalar, cv2.GMat, cv2.GOpaqueT, cv2.GArrayT]
  161. GProtoInputArgs = _typing.Sequence[GProtoArg]
  162. GProtoOutputArgs = _typing.Sequence[GProtoArg]
  163. GRunArg = _typing.Union[MatLike, Scalar, cv2.GOpaqueT, cv2.GArrayT, _typing.Sequence[_typing.Any], None]
  164. GOptRunArg = _typing.Optional[GRunArg]
  165. GMetaArg = _typing.Union[cv2.GMat, Scalar, cv2.GOpaqueT, cv2.GArrayT]
  166. Prim = _typing.Union[cv2.gapi.wip.draw.Text, cv2.gapi.wip.draw.Circle, cv2.gapi.wip.draw.Image, cv2.gapi.wip.draw.Line, cv2.gapi.wip.draw.Rect, cv2.gapi.wip.draw.Mosaic, cv2.gapi.wip.draw.Poly]
  167. GTypeInfo = _typing.Union[cv2.GMat, Scalar, cv2.GOpaqueT, cv2.GArrayT]
  168. ExtractArgsCallback = _typing.Callable[[_typing.Sequence[GTypeInfo]], _typing.Sequence[GRunArg]]
  169. ExtractMetaCallback = _typing.Callable[[_typing.Sequence[GTypeInfo]], _typing.Sequence[GMetaArg]]