Ë
    ºrœhH  ã                   ó°   — d Z ddlmZmZ ddlmZmZ ddlmZ ddl	m
Z
 ddlmZ  e«       rddlmZ  G d„ d	e«      Z e
d
¬«       G d„ de«      «       ZdgZy)z&Fast Video processor class for VJEPA2.é   )ÚIMAGENET_DEFAULT_MEANÚIMAGENET_DEFAULT_STD)ÚUnpackÚVideosKwargs)Úis_vision_available)Úrequires)ÚBaseVideoProcessor)ÚPILImageResamplingc                   ó   — e Zd Zy)ÚVJEPA2VideoProcessorInitKwargsN)Ú__name__Ú
__module__Ú__qualname__© ó    ú…/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/transformers/models/vjepa2/video_processing_vjepa2.pyr   r      s   … r   r   )Útorchvision)Úbackendsc                   ó|   ‡ — e Zd Zej                  ZeZeZ	d e
d«      iZdZdZdZdZdZeZdgZdee   fˆ fd„Zˆ xZS )ÚVJEPA2VideoProcessorÚshortest_edgeg’$I’$Ir@é   TÚpixel_values_videosÚkwargsc                 óà   •— |j                  dd«      }t        |t        «      s$t        |t        «      rd|vrt	        d«      ‚|d   }t        |dz  dz  «      }d|i|d<   t        ‰|   di |¤Ž y )	NÚ	crop_sizer   Úheightz@crop_size must be an integer or a dictionary with a 'height' keyéà   r   Úsizer   )ÚgetÚ
isinstanceÚintÚdictÚ
ValueErrorÚsuperÚ__init__)Úselfr   r   Úresize_sizeÚ	__class__s       €r   r&   zVJEPA2VideoProcessor.__init__0   su   ø€ Ø—J‘J˜{¨CÓ0ˆ	Ü˜)¤SÔ)Ü˜i¬Ô.°(À)Ñ2KÜ Ð!cÓdÐdØ! (Ñ+ˆIÜ˜) c™/¨CÑ/Ó0ˆØ)¨;Ð7ˆˆv‰Ü‰ÑÑ"˜6Ó"r   )r   r   r   r
   ÚBILINEARÚresampler   Ú
image_meanr   Ú	image_stdr"   r   r   Ú	do_resizeÚ
do_rescaleÚdo_center_cropÚdo_normalizer   Úvalid_kwargsÚmodel_input_namesr   r&   Ú__classcell__)r)   s   @r   r   r   "   se   ø„ à!×*Ñ*€HØ&€JØ$€IØ™S Ó1Ð2€DØ€IØ€IØ€JØ€NØ€LØ1€LØ.Ð/Ðð# Ð(FÑ!G÷ #ñ #r   r   N)Ú__doc__Úimage_utilsr   r   Úprocessing_utilsr   r   Úutilsr   Úutils.import_utilsr   Úvideo_processing_utilsr	   r
   r   r   Ú__all__r   r   r   ú<module>r<      s^   ðñ -÷÷ 5Ý (Ý *Ý 8ñ ÔÝ1ô 8 \Ô 7ñ 
Ð#Ô$ô#Ð-ó #ó %ð#ð0 "Ð
"r   