
    Z jW
                         S r SSKrSSKJr  SSKJr  SSKJr  SSK	J
r
JrJrJrJr  SSKJrJr  SS	KJr  \ " S
 S\5      5       rS/rg)z Image processor class for LeViT.    N)
functional   )TorchvisionBackend)get_resize_output_image_size)IMAGENET_DEFAULT_MEANIMAGENET_DEFAULT_STDChannelDimensionPILImageResamplingSizeDict)ImagesKwargsUnpack)auto_docstringc            	          ^  \ rS rSrSr\R                  r\r	\
rSS0rSrSSS.rSrSrSrSrSrS	\\   4U 4S
 jjr SSSS\SSSS4U 4S jjjrSrU =r$ )LevitImageProcessor    zKTorchvision backend for LeViT with custom resize (shortest_edge * 256/224).shortest_edge   FheightwidthTNkwargsc                 &   > [         TU ]  " S0 UD6  g )N )super__init__)selfr   	__class__s     ځ/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/models/levit/image_processing_levit.pyr   LevitImageProcessor.__init__0   s    "6"    imageztorch.Tensorsizeresamplez7PILImageResampling | tvF.InterpolationMode | int | Nonereturnc                 V  > UR                   (       a=  [        SUR                   -  5      n[        XS[        R                  S9u  pg[        XgS9nOHUR                  (       a  UR                  (       d&  [        S[        UR                  5       5       S35      e[        TU ]0  " U4X#S.UD6$ )zDResize: shortest_edge is rescaled to int((256/224) * shortest_edge).g$I$I?F)r"   default_to_squareinput_data_formatr   zFSize dict must have keys 'height' and 'width' or 'shortest_edge'. Got .)r"   r#   )r   intr   r	   FIRSTr   r   r   
ValueErrorlistkeysr   resize)	r   r!   r"   r#   r   r   new_size_heightnew_size_widthr   s	           r   r.   LevitImageProcessor.resize3   s     d.@.@ @AM.JUVfVlVl/+O ?IDDJJXY]^b^g^g^iYjXkklm  w~eL$LVLLr    r   )N)__name__
__module____qualname____firstlineno____doc__r
   BICUBICr#   r   
image_meanr   	image_stdr"   r&   	crop_size	do_resizedo_center_crop
do_rescaledo_normalizedo_convert_rgbr   r   r   r   r.   __static_attributes____classcell__)r   s   @r   r   r       s    U!))H&J$IS!D-IINJLN#!5 # OS	MM M L	M 
M Mr    r   )r6   torchtorchvision.transforms.v2r   tvFimage_processing_backendsr   image_transformsr   image_utilsr   r   r	   r
   r   processing_utilsr   r   utilsr   r   __all__r   r    r   <module>rK      sR    '  7 ; <  5 # $M, $M $MN !
!r    