3jSSKJr SSKJr SSKJr SSKrSSKJr SSKJ r SSK J r J r "SS \5r S \ R4SS jjr"S S \ 5rg)) annotations)IntEnum)UnionN)Tensor)Module) KORNIA_CHECKKORNIA_CHECK_IS_TENSORc0\rSrSrSrSrSrSrSrSr Sr S r g ) ThresholdTypezThreshold types compatible with OpenCV fixed thresholding types. Note: THRESH_OTSU is intentionally not supported in this PR. rN) __name__ __module__ __qualname____firstlineno____doc__ THRESH_BINARYTHRESH_BINARY_INV THRESH_TRUNC THRESH_TOZEROTHRESH_TOZERO_INV THRESH_OTSU__static_attributes__rR/home/wildlama/miniconda3/lib/python3.13/site-packages/kornia/enhance/threshold.pyr r s+ MLMKrr o@c [U5 [U5nU[[R5-(a [ S5e[ U[Vs1sH#oU[R:wdM[U5iM% sn;SUS35 Un[ U[5(d*[R"X`RURS9nO$URURURS9nUn[ U[5(d*[R"XpRURS9nO$URURURS9nX:n[R"U5n U[[R5:Xa[R"XU 5$U[[R 5:Xa[R"XU5$U[[R"5:Xa[R$"X5$U[[R&5:Xa[R"XU 5$U[[R(5:Xa[R"XU5$[+SU35es snf)aApply a fixed-level threshold to each element in the input tensor. Implements OpenCV-like behavior for the following threshold types: - THRESH_BINARY - THRESH_BINARY_INV - THRESH_TRUNC - THRESH_TOZERO - THRESH_TOZERO_INV Args: input: Image tensor of shape (..., H, W). Typically (B, C, H, W). thresh: Threshold value (scalar or tensor broadcastable to input). maxval: Maximum value used with binary thresholding types. type: Threshold type. Returns: Thresholded tensor with same shape/dtype/device as `input`. Raises: NotImplementedError: if THRESH_OTSU flag is passed. ValueError: if type is not supported. zFTHRESH_OTSU is not implemented yet. Please use a fixed threshold type.zUnsupported threshold type: z;. Supported: BINARY, BINARY_INV, TRUNC, TOZERO, TOZERO_INV.)devicedtype)r intr rNotImplementedErrorr isinstancertorchtensorr#r$to zeros_likerwhererrminimumrr ValueError) inputthreshmaxvaltypetxthresh_tmaxval_tmaskzeross r thresholdr9.s85! D A 3}(( ))!"jkk m NmM4M4M/Mfc!fm NN &tf,gh H h ' '<<U[[Q;;ell%++;FH h ' '<<U[[Q;;ell%++;F  D   U #EC ++ ,,{{4511C // 00{{411C ** ++}}U--C ++ ,,{{4..C // 00{{4.. 3D6: ;;IOs I> ,I> cd^\rSrSrSrS\R 4SU4SjjjrSSjrSr U=r $) Thresholdzz.Module wrapper for `kornia.enhance.threshold`.r!c>[TU]5 [U5Ul[U5Ul[ U5UlgN)super__init__floatr0r1r%r2)selfr0r1r2 __class__s r r@Threshold.__init__}s1 Fm Fm I rcX[XRURUR5$r>)r9r0r1r2)rBr/s r forwardThreshold.forwards T[[$))DDr)r1r0r2)r0rAr1rAr2Union[int, ThresholdType]returnNone)r/rrIr) rrrrrr rr@rFr __classcell__)rCs@r r;r;zsP8 *7*E*E   (    EErr;) r/rr0Union[float, Tensor]r1rLr2rHrIr) __future__renumrtypingrr(rtorch.nnrkornia.core.checkrr r rr9r;rrr rRs~$# B G &$)&3&A&A I< I< I< !I< $ I<  I