__init__.pyi 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121
  1. __all__: list[str] = []
  2. import cv2
  3. import cv2.typing
  4. import typing as _typing
  5. # Enumerations
  6. RETINA_COLOR_RANDOM: int
  7. RETINA_COLOR_DIAGONAL: int
  8. RETINA_COLOR_BAYER: int
  9. # Classes
  10. class Retina(cv2.Algorithm):
  11. # Functions
  12. def getInputSize(self) -> cv2.typing.Size: ...
  13. def getOutputSize(self) -> cv2.typing.Size: ...
  14. def setup(self, retinaParameterFile: str = ..., applyDefaultSetupOnFailure: bool = ...) -> None: ...
  15. def printSetup(self) -> str: ...
  16. def write(self, fs: str) -> None: ...
  17. def setupOPLandIPLParvoChannel(self, colorMode: bool = ..., normaliseOutput: bool = ..., photoreceptorsLocalAdaptationSensitivity: float = ..., photoreceptorsTemporalConstant: float = ..., photoreceptorsSpatialConstant: float = ..., horizontalCellsGain: float = ..., HcellsTemporalConstant: float = ..., HcellsSpatialConstant: float = ..., ganglionCellsSensitivity: float = ...) -> None: ...
  18. def setupIPLMagnoChannel(self, normaliseOutput: bool = ..., parasolCells_beta: float = ..., parasolCells_tau: float = ..., parasolCells_k: float = ..., amacrinCellsTemporalCutFrequency: float = ..., V0CompressionParameter: float = ..., localAdaptintegration_tau: float = ..., localAdaptintegration_k: float = ...) -> None: ...
  19. @_typing.overload
  20. def run(self, inputImage: cv2.typing.MatLike) -> None: ...
  21. @_typing.overload
  22. def run(self, inputImage: cv2.UMat) -> None: ...
  23. @_typing.overload
  24. def applyFastToneMapping(self, inputImage: cv2.typing.MatLike, outputToneMappedImage: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  25. @_typing.overload
  26. def applyFastToneMapping(self, inputImage: cv2.UMat, outputToneMappedImage: cv2.UMat | None = ...) -> cv2.UMat: ...
  27. @_typing.overload
  28. def getParvo(self, retinaOutput_parvo: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  29. @_typing.overload
  30. def getParvo(self, retinaOutput_parvo: cv2.UMat | None = ...) -> cv2.UMat: ...
  31. @_typing.overload
  32. def getParvoRAW(self, retinaOutput_parvo: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  33. @_typing.overload
  34. def getParvoRAW(self, retinaOutput_parvo: cv2.UMat | None = ...) -> cv2.UMat: ...
  35. @_typing.overload
  36. def getParvoRAW(self) -> cv2.typing.MatLike: ...
  37. @_typing.overload
  38. def getMagno(self, retinaOutput_magno: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  39. @_typing.overload
  40. def getMagno(self, retinaOutput_magno: cv2.UMat | None = ...) -> cv2.UMat: ...
  41. @_typing.overload
  42. def getMagnoRAW(self, retinaOutput_magno: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  43. @_typing.overload
  44. def getMagnoRAW(self, retinaOutput_magno: cv2.UMat | None = ...) -> cv2.UMat: ...
  45. @_typing.overload
  46. def getMagnoRAW(self) -> cv2.typing.MatLike: ...
  47. def setColorSaturation(self, saturateColors: bool = ..., colorSaturationValue: float = ...) -> None: ...
  48. def clearBuffers(self) -> None: ...
  49. def activateMovingContoursProcessing(self, activate: bool) -> None: ...
  50. def activateContoursProcessing(self, activate: bool) -> None: ...
  51. @classmethod
  52. @_typing.overload
  53. def create(cls, inputSize: cv2.typing.Size) -> Retina: ...
  54. @classmethod
  55. @_typing.overload
  56. def create(cls, inputSize: cv2.typing.Size, colorMode: bool, colorSamplingMethod: int = ..., useRetinaLogSampling: bool = ..., reductionFactor: float = ..., samplingStrength: float = ...) -> Retina: ...
  57. class RetinaFastToneMapping(cv2.Algorithm):
  58. # Functions
  59. @_typing.overload
  60. def applyFastToneMapping(self, inputImage: cv2.typing.MatLike, outputToneMappedImage: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  61. @_typing.overload
  62. def applyFastToneMapping(self, inputImage: cv2.UMat, outputToneMappedImage: cv2.UMat | None = ...) -> cv2.UMat: ...
  63. def setup(self, photoreceptorsNeighborhoodRadius: float = ..., ganglioncellsNeighborhoodRadius: float = ..., meanLuminanceModulatorK: float = ...) -> None: ...
  64. @classmethod
  65. def create(cls, inputSize: cv2.typing.Size) -> RetinaFastToneMapping: ...
  66. class TransientAreasSegmentationModule(cv2.Algorithm):
  67. # Functions
  68. def getSize(self) -> cv2.typing.Size: ...
  69. def setup(self, segmentationParameterFile: str = ..., applyDefaultSetupOnFailure: bool = ...) -> None: ...
  70. def printSetup(self) -> str: ...
  71. def write(self, fs: str) -> None: ...
  72. @_typing.overload
  73. def run(self, inputToSegment: cv2.typing.MatLike, channelIndex: int = ...) -> None: ...
  74. @_typing.overload
  75. def run(self, inputToSegment: cv2.UMat, channelIndex: int = ...) -> None: ...
  76. @_typing.overload
  77. def getSegmentationPicture(self, transientAreas: cv2.typing.MatLike | None = ...) -> cv2.typing.MatLike: ...
  78. @_typing.overload
  79. def getSegmentationPicture(self, transientAreas: cv2.UMat | None = ...) -> cv2.UMat: ...
  80. def clearAllBuffers(self) -> None: ...
  81. @classmethod
  82. def create(cls, inputSize: cv2.typing.Size) -> TransientAreasSegmentationModule: ...