Vpf5ddlmZddlmZmZddlmZddlZddlZ ddl m Z ddl m Z ddl mZ ddlmZdd lmZdd lmZd Zd Zd Zd)dZd*dZGddejZejeejdejdejeiZde j j!fd+d#Z"d,d$Z#ee d%&d-d'Z$de j j!fd-d(Z%dS).) annotations)CallableSequence)partialN)jit)lax)numpy)core)canonicalize_axis)promote_dtypes_inexactc x|tjtj|zztjtj|z|z z}tj|dktj|tj|dktjdz|dzzdd}tj||kd|S)NgMbP?r)jnpsinnppiwheredivide)radiusxyouts T/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/image/scale.py_fill_lanczos_kernelrs swruqy!!!CGBEAI,>$?$??! !d(CJq#)AFBE1Hq!tOQ*O*OPPRSTT# 1v:r3 ' ''cd|zdz |z|zdz}tj|dkd|zdz|zdz |zdz|}tj|dkd|S)Ng?g@?g@g@r)rr)rrs r_fill_keys_cubic_kernelr!&sk 'C-1 !B&# !r'TAX^q025:R?EE# 17B $ $$rcVtjddtj|z S)Nrr)rmaximumabsrs r_fill_triangle_kernelr&0s QCGAJJ ' ''r input_size core.DimSize output_sizekernelr antialiasboolc jtj||}d|z }|rtj|dnd}tj||dz|z||zz dz } tj| tjddftj||ddtjfz |z } || } tj| dd} tjtj| dttj tj j zktj | tj| dk| dd} tj|dz } tjtj| d k| | ktjddf| dS) Nrdtype?rT)axiskeepdimsg@@rr )r result_typer#aranger$newaxissumrfloatrfinfofloat32epsrr dimension_as_value logical_and)r'r)scale translationr*r+r/ inv_scale kernel_scalesample_frweightstotal_weight_suminput_size_minus_0_5s rcompute_weight_matrE4s /% - -%5j)09@Y+++b,z+U333c9YFI%&(+,( ghs{AAA~&j5111!!!S[.ABCC F1II'WW1t<<< I g%%0D0D0H*I*I"II j#)$4$9r*r+rK input_shape contractions in_indices out_indicesidmnws r_scale_and_translater^Vs+ [  S.. . . . . \  c%jj ( ( ( ( \  c+.. . . . .! H,E#l++,,--*U3|,,--..+ %%++da!QV$$AAAQA1aq;q>!9 . ..4fQWooC --12333&&*KNNk""" Az FL F F FI F FFrc>eZdZdZdZdZdZdZdZe d d Z d S) ResizeMethoda>Image resize method. Possible values are: NEAREST: Nearest-neighbor interpolation. LINEAR: `Linear interpolation`_. LANCZOS3: `Lanczos resampling`_, using a kernel of radius 3. LANCZOS5: `Lanczos resampling`_, using a kernel of radius 5. CUBIC: `Cubic interpolation`_, using the Keys cubic kernel. .. _Linear interpolation: https://en.wikipedia.org/wiki/Bilinear_interpolation .. _Cubic interpolation: https://en.wikipedia.org/wiki/Bicubic_interpolation .. _Lanczos resampling: https://en.wikipedia.org/wiki/Lanczos_resampling rrrsstrc|dkr tjS|dvr tjS|dkr tjS|dkr tjS|dvr tjSt d|d)Nnearest)linearbilinear trilineartrianglelanczos3lanczos5)cubicbicubictricubiczUnknown resize method "")r`NEARESTLINEARLANCZOS3LANCZOS5CUBIC ValueError)rcs r from_stringzResizeMethod.from_strings|I~~  !! ;;;   j  "" j  "" . . .   5555 6 66rN)rcrd) __name__ __module__ __qualname____doc__rqrrrsrtru staticmethodrwrrr`r`osY0 ' & ( ( % 7 7 7< 7 7 7rr`c"td|S)Ng@rr%s rr%9"a%@%@rc"td|S)Ng@rr%s rrrrrTrLmethodstr | ResizeMethodc tj|}t||jkrd|d|j}t |t |trt |}|tj krt dt |tsJt|} t|\}t||\}}t|||||| ||S)a Apply a scale and translation to an image. Generates a new image of shape 'shape' by resampling from the input image using the sampling method corresponding to method. For 2D images, this operation transforms a location in the input images, (x, y), to a location in the output image according to:: (x * scale[1] + translation[1], y * scale[0] + translation[0]) (Note the *inverse* warp is used to generate the sample locations.) Assumes half-centered pixels, i.e the pixel at integer location ``row, col`` has coordinates ``y, x = row + 0.5, col + 0.5``, and similarly for other input image dimensions. If an output location(pixel) maps to an input sample location that is outside the input boundaries then the value for the output location will be set to zero. The ``method`` argument expects one of the following resize methods: ``ResizeMethod.LINEAR``, ``"linear"``, ``"bilinear"``, ``"trilinear"``, ``"triangle"`` `Linear interpolation`_. If ``antialias`` is ``True``, uses a triangular filter when downsampling. ``ResizeMethod.CUBIC``, ``"cubic"``, ``"bicubic"``, ``"tricubic"`` `Cubic interpolation`_, using the Keys cubic kernel. ``ResizeMethod.LANCZOS3``, ``"lanczos3"`` `Lanczos resampling`_, using a kernel of radius 3. ``ResizeMethod.LANCZOS5``, ``"lanczos5"`` `Lanczos resampling`_, using a kernel of radius 5. .. _Linear interpolation: https://en.wikipedia.org/wiki/Bilinear_interpolation .. _Cubic interpolation: https://en.wikipedia.org/wiki/Bicubic_interpolation .. _Lanczos resampling: https://en.wikipedia.org/wiki/Lanczos_resampling Args: image: a JAX array. shape: the output shape, as a sequence of integers with length equal to the number of dimensions of `image`. spatial_dims: A length K tuple specifying the spatial dimensions that the passed scale and translation should be applied to. scale: A [K] array with the same number of dimensions as image, containing the scale to apply in each dimension. translation: A [K] array with the same number of dimensions as image, containing the translation to apply in each dimension. method: the resizing method to use; either a ``ResizeMethod`` instance or a string. Available methods are: LINEAR, LANCZOS3, LANCZOS5, CUBIC. antialias: Should an antialiasing filter be used when downsampling? Defaults to ``True``. Has no effect when upsampling. Returns: The scale and translated image. @shape must have length equal to the number of dimensions of x;  vs zONearest neighbor resampling is not currently supported for scale_and_translate.)r canonicalize_shaperMrQrLrv isinstancerdr`rwrq_kernelsr r^) imagerLrHr=r>rr+rKmsgr*s rscale_and_translaters z  !% ( (%ZZ5: ( ( (+ ( (C S//.  % %f - -F |### 0 1 11 FL ) )))) F & !% ( (&%-e[AA% eUL%$i < <z"_resize_nearest..sZVVQ"3KNLQROTTVqVVVVVVrr.r0)rLrMtuplerOrr4rr9r r;floorrRint32slice) rrFrHrZr[r\offsetsindicesrUs ` @r_resize_nearestrsG+ [  S.. . . . .VVVVV%K(8(8"9"9VVVVV,   aAAQAz!2:...48OPQ8R8RRUYUlmnUoUooGirz2233::28DDGT{{mc+...GGAJ %..AA (r)rrrarb)static_argnumsc Ntjkrddj}t|t |t rt |}|t jkrtSt |t sJt|}t\tfdttD}fd|D}t||dgt|z|||S)Nrrc3hK|],}tjj||(|V-dSr)r rrL)rrYrrLs rrz_resize.."s[OOQ"3EKNE!HMMOqOOOOOOrcg|]Y}tj|drdn9tj|tjj|z ZS)rr)r rr;rL)rrZrrLs r z_resize..$s " " "'a!44 F33$:QRWXYRZ:[:[^b^uv{wBCDwE_F_F;F " " "rr)rMrQrLrvrrdr`rwrqrrr rrOr^) rrLrr+rKrr*rHr=s `` r_resizersm ZZ5: ( ( (+ ( (C S//.  % %f - -F |### 5% ( (( FL ) )))) F & !% ( (&%OOOOO%E "3"3OOOOO, " " " " "  " " "% eUL#bTC ,=,=%=v' 4 44rcLt|tj||||S)aImage resize. The ``method`` argument expects one of the following resize methods: ``ResizeMethod.NEAREST``, ``"nearest"`` `Nearest neighbor interpolation`_. The values of ``antialias`` and ``precision`` are ignored. ``ResizeMethod.LINEAR``, ``"linear"``, ``"bilinear"``, ``"trilinear"``, ``"triangle"`` `Linear interpolation`_. If ``antialias`` is ``True``, uses a triangular filter when downsampling. ``ResizeMethod.CUBIC``, ``"cubic"``, ``"bicubic"``, ``"tricubic"`` `Cubic interpolation`_, using the Keys cubic kernel. ``ResizeMethod.LANCZOS3``, ``"lanczos3"`` `Lanczos resampling`_, using a kernel of radius 3. ``ResizeMethod.LANCZOS5``, ``"lanczos5"`` `Lanczos resampling`_, using a kernel of radius 5. .. _Nearest neighbor interpolation: https://en.wikipedia.org/wiki/Nearest-neighbor_interpolation .. _Linear interpolation: https://en.wikipedia.org/wiki/Bilinear_interpolation .. _Cubic interpolation: https://en.wikipedia.org/wiki/Bicubic_interpolation .. _Lanczos resampling: https://en.wikipedia.org/wiki/Lanczos_resampling Args: image: a JAX array. shape: the output shape, as a sequence of integers with length equal to the number of dimensions of `image`. Note that :func:`resize` does not distinguish spatial dimensions from batch or channel dimensions, so this includes all dimensions of the image. To represent a batch or a channel dimension, simply leave that element of the shape unchanged. method: the resizing method to use; either a ``ResizeMethod`` instance or a string. Available methods are: LINEAR, LANCZOS3, LANCZOS5, CUBIC. antialias: should an antialiasing filter be used when downsampling? Defaults to ``True``. Has no effect when upsampling. Returns: The resized image. )rr r)rrLrr+rKs rresizer+s-V /66   r)r'r(r)r(r*rr+r,)rFrGrHrIr+r,)rLrGrHrIrrr+r,)rFrG)rLrGrrr+r,)& __future__rcollections.abcrr functoolsrenumr rjaxrrrjax._srcr jax._src.utilr jax._src.numpy.utilr rr!r&rEr^Enumr`rrrsrtrur PrecisionHIGHESTrrrrr}rrrs"#"""""........ ++++++666666(((%%%(((UUUUDGGGG2171717171749171717h.@@@@/  +/"%-"7 O<O<O<O<O