
    Z jTZ                        S SK r S SKrS SKrS SKJrJr  S SKrS SKJ	r	J
r
  SSKJr  SSKJr  SSKJrJr  SSKJrJrJrJrJrJr  SS	KJr  \" S
SS9r\R:                  " \5      r " S S\5      r " S S\5      r \" \ RB                  5      \ l!        \ RB                  RD                  b5  \ RB                  RD                  RG                  SSSS9\ RB                  l"        gg)    N)AnyTypeVar)create_repois_offline_mode   )custom_object_save)BatchFeature)is_valid_image
load_image)IMAGE_PROCESSOR_NAMEPROCESSOR_NAMEPushToHubMixin	copy_funcloggingsafe_load_json_file)cached_fileImageProcessorTypeImageProcessingMixin)boundc                       \ rS rSrSrSrg)r	   -   a  
Holds the output of the image processor specific `__call__` methods.

This class is derived from a python dictionary and can be used as a dictionary.

Args:
    data (`dict`):
        Dictionary of lists/arrays/tensors returned by the __call__ method ('pixel_values', etc.).
    tensor_type (`Union[None, str, TensorType]`, *optional*):
        You can give a tensor_type here to convert the lists of integers in PyTorch/Numpy Tensors at
        initialization.
 N)__name__
__module____qualname____firstlineno____doc____static_attributes__r       s/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/image_processing_base.pyr	   r	   -   s    r   r	   c                      \ rS rSrSrSrS r\     SS\\	   S\
\R                  -  S\
\R                  -  S-  S\S	\S
\
\-  S-  S\
S\	4S jj5       rS S\
\R                  -  S\4S jjr\S\
\R                  -  S\\\
\4   \\
\4   4   4S j5       r\S\\
\4   4S j5       rS\\
\4   4S jr\S\
\R                  -  4S j5       rS\
4S jrS\
\R                  -  4S jrS r\S!S j5       rS\
\\
   -  \\\
      -  4S jrSrg)"r   =   z|
This is an image processor mixin used to provide saving/loading functionality for sequential and image feature
extractors.
Nc           
          UR                  SS5        UR                  SS5        UR                  5        H  u  p# [        XU5        M     g! [         a%  n[        R                  SU SU SU  35        UeSnAff = f)z'Set elements of `kwargs` as attributes.feature_extractor_typeNprocessor_classz
Can't set z with value z for )popitemssetattrAttributeErrorloggererror)selfkwargskeyvalueerrs        r    __init__ImageProcessingMixin.__init__E   sy     	

+T2

$d+ ,,.JC5) ) " z#l5'tfMN	s   A
A; A66A;clspretrained_model_name_or_path	cache_dirforce_downloadlocal_files_onlytokenrevisionreturnc                     X'S'   X7S'   XGS'   XgS'   Ub  XWS'   U R                   " U40 UD6u  pU R                  " U40 UD6$ )a  
Instantiate a type of [`~image_processing_utils.ImageProcessingMixin`] from an image processor.

Args:
    pretrained_model_name_or_path (`str` or `os.PathLike`):
        This can be either:

        - a string, the *model id* of a pretrained image_processor hosted inside a model repo on
          huggingface.co.
        - a path to a *directory* containing a image processor file saved using the
          [`~image_processing_utils.ImageProcessingMixin.save_pretrained`] method, e.g.,
          `./my_model_directory/`.
        - a path to a saved image processor JSON *file*, e.g.,
          `./my_model_directory/preprocessor_config.json`.
    cache_dir (`str` or `os.PathLike`, *optional*):
        Path to a directory in which a downloaded pretrained model image processor should be cached if the
        standard cache should not be used.
    force_download (`bool`, *optional*, defaults to `False`):
        Whether or not to force to (re-)download the image processor files and override the cached versions if
        they exist.
    proxies (`dict[str, str]`, *optional*):
        A dictionary of proxy servers to use by protocol or endpoint, e.g., `{'http': 'foo.bar:3128',
        'http://hostname': 'foo.bar:4012'}.` The proxies are used on each request.
    token (`str` or `bool`, *optional*):
        The token to use as HTTP bearer authorization for remote files. If `True`, or not specified, will use
        the token generated when running `hf auth login` (stored in `~/.huggingface`).
    revision (`str`, *optional*, defaults to `"main"`):
        The specific model version to use. It can be a branch name, a tag name, or a commit id, since we use a
        git-based system for storing models and other artifacts on huggingface.co, so `revision` can be any
        identifier allowed by git.


        <Tip>

        To test a pull request you made on the Hub, you can pass `revision="refs/pr/<pr_number>"`.

        </Tip>

    return_unused_kwargs (`bool`, *optional*, defaults to `False`):
        If `False`, then this function returns just the final image processor object. If `True`, then this
        functions returns a `Tuple(image_processor, unused_kwargs)` where *unused_kwargs* is a dictionary
        consisting of the key/value pairs whose keys are not image processor attributes: i.e., the part of
        `kwargs` which has not been used to update `image_processor` and is otherwise ignored.
    subfolder (`str`, *optional*, defaults to `""`):
        In case the relevant files are located inside a subfolder of the model repo on huggingface.co, you can
        specify the folder name here.
    kwargs (`dict[str, Any]`, *optional*):
        The values in kwargs of any keys which are image processor attributes will be used to override the
        loaded values. Behavior concerning key/value pairs whose keys are *not* image processor attributes is
        controlled by the `return_unused_kwargs` keyword parameter.

Returns:
    A image processor of type [`~image_processing_utils.ImageProcessingMixin`].

Examples:

```python
# We can't instantiate directly the base class *ImageProcessingMixin* so let's show the examples on a
# derived class: *CLIPImageProcessor*
image_processor = CLIPImageProcessor.from_pretrained(
    "openai/clip-vit-base-patch32"
)  # Download image_processing_config from huggingface.co and cache.
image_processor = CLIPImageProcessor.from_pretrained(
    "./test/saved_model/"
)  # E.g. image processor (or model) was saved using *save_pretrained('./test/saved_model/')*
image_processor = CLIPImageProcessor.from_pretrained("./test/saved_model/preprocessor_config.json")
image_processor = CLIPImageProcessor.from_pretrained(
    "openai/clip-vit-base-patch32", do_normalize=False, foo=False
)
assert image_processor.do_normalize is False
image_processor, unused_kwargs = CLIPImageProcessor.from_pretrained(
    "openai/clip-vit-base-patch32", do_normalize=False, foo=False, return_unused_kwargs=True
)
assert image_processor.do_normalize is False
assert unused_kwargs == {"foo": False}
```r5   r6   r7   r9   r8   )get_image_processor_dict	from_dict)	r3   r4   r5   r6   r7   r8   r9   r-   image_processor_dicts	            r    from_pretrained$ImageProcessingMixin.from_pretrainedT   sd    n ({#1 %5!"%z#7O'*'C'CDa'lek'l$}}1<V<<r   save_directorypush_to_hubc           	         [         R                  R                  U5      (       a  [        SU S35      e[         R                  " USS9  U(       aw  UR                  SS5      nUR                  SUR                  [         R                  R                  5      S   5      n[        U4S	S0UD6R                  nU R                  U5      nU R                  b
  [        XU S
9  [         R                  R                  U[        5      nU R                  U5        [         R#                  SU 35        U(       a"  U R%                  UWWWUR'                  S5      S9  U/$ )a  
Save an image processor object to the directory `save_directory`, so that it can be re-loaded using the
[`~image_processing_utils.ImageProcessingMixin.from_pretrained`] class method.

Args:
    save_directory (`str` or `os.PathLike`):
        Directory where the image processor JSON file will be saved (will be created if it does not exist).
    push_to_hub (`bool`, *optional*, defaults to `False`):
        Whether or not to push your model to the Hugging Face model hub after saving it. You can specify the
        repository you want to push to with `repo_id` (will default to the name of `save_directory` in your
        namespace).
    kwargs (`dict[str, Any]`, *optional*):
        Additional key word arguments passed along to the [`~utils.PushToHubMixin.push_to_hub`] method.
zProvided path (z#) should be a directory, not a fileT)exist_okcommit_messageNrepo_idrD   )configzImage processor saved in r8   )rE   r8   )ospathisfileAssertionErrormakedirsr&   splitsepr   rF   _get_files_timestamps_auto_classr   joinr   to_json_filer*   info_upload_modified_filesget)r,   rA   rB   r-   rE   rF   files_timestampsoutput_image_processor_files           r    save_pretrained$ImageProcessingMixin.save_pretrained   s0    77>>.)) ?>2BBe!fgg
NT2#ZZ(8$?NjjN,@,@,Mb,QRG!'CDCFCKKG#99.I 'tDA ')ggll>CW&X#56/0K/LMN'' -jj) (  ,,,r   c                    UR                  SS5      nUR                  SS5      nUR                  SS5      nUR                  SS5      nUR                  SS5      nUR                  SS5      nUR                  S	S
5      n	UR                  S[        5      n
UR                  SS5      nUR                  SS5      nSUS.nUb  XS'   [        5       (       a  U(       d  [        R	                  S5        Sn[        U5      n[        R                  R                  U5      n[        R                  R                  U5      (       a  [        R                  R                  X5      n[        R                  R                  U5      (       a  UnSnSnO-U
n [        U[        UUUUUUUU	SS9n[        UUUUUUUUUU	SS9nSnUb  [        U5      nSU;   a  US   nUb  Uc  [        U5      nUc  [        SU SU SU
 S35      eU(       a  [        R	                  SU 35        UU4$ [        R	                  SW SU 35        UU4$ ! [         a    e [         a    [        SU SU SU
 S35      ef = f)aa  
From a `pretrained_model_name_or_path`, resolve to a dictionary of parameters, to be used for instantiating a
image processor of type [`~image_processor_utils.ImageProcessingMixin`] using `from_dict`.

Parameters:
    pretrained_model_name_or_path (`str` or `os.PathLike`):
        The identifier of the pre-trained checkpoint from which we want the dictionary of parameters.
    subfolder (`str`, *optional*, defaults to `""`):
        In case the relevant files are located inside a subfolder of the model repo on huggingface.co, you can
        specify the folder name here.
    image_processor_filename (`str`, *optional*, defaults to `"config.json"`):
        The name of the file in the model directory to use for the image processor config.

Returns:
    `tuple[Dict, Dict]`: The dictionary(ies) that will be used to instantiate the image processor object.
r5   Nr6   Fproxiesr8   r7   r9   	subfolder image_processor_filename_from_pipeline
_from_autoimage processor)	file_typefrom_auto_classusing_pipelinez+Offline mode: forcing local_files_only=TrueT)
filenamer5   r6   r\   r7   r8   
user_agentr9   r]   %_raise_exceptions_for_missing_entriesz Can't load image processor for 'z'. If you were trying to load it from 'https://huggingface.co/models', make sure you don't have a local directory with the same name. Otherwise, make sure 'z2' is the correct path to a directory containing a z fileimage_processorzloading configuration file z from cache at )r&   r   r   r*   rT   strrI   rJ   isdirrR   rK   r   r   OSError	Exceptionr   )r3   r4   r-   r5   r6   r\   r8   r7   r9   r]   r_   from_pipelinerd   rg   is_localimage_processor_fileresolved_image_processor_fileresolved_processor_filer>   processor_dicts                       r    r<   -ImageProcessingMixin.get_image_processor_dict   s   ( JJ{D1	$4e<**Y-

7D)!::&8%@::j$/JJ{B/	#)::.HJ^#_ 

#3T: **\59#4Y
$+8'(%5KKEF#(+,I(J%77==!>?77==677#%77<<0M#h 77>>788,I)&*#H#; &*51+'#1#%5)%':?+' 1<11'#1#%5)%':?1-:  $".01HIN N2'56G'H$(49M9U#67T#U '23P2Q R55R4S T++C*DEK  KK56S5TUV $V++	 KK-.B-C?SpRqr $V++Q    67T6U V99V8W X//G.HO s   *H= =*I'r>   c           	         UR                  5       nUR                  SS5      nUR                  UR                  5        VVs0 s H"  u  pEX@R                  R
                  ;   d  M   XE_M$     snn5        U " S0 UD6n/ n[        [        UR                  5       5      5       H]  n[        Xh5      (       d  M  XR                  R
                  ;  d  M0  [        XhUR                  US5      5        UR                  U5        M_     U(       a&  [        R                  SU R                   SU S35        [        R                  SU 35        U(       a  Xb4$ U$ s  snnf )a  
Instantiates a type of [`~image_processing_utils.ImageProcessingMixin`] from a Python dictionary of parameters.

Args:
    image_processor_dict (`dict[str, Any]`):
        Dictionary that will be used to instantiate the image processor object. Such a dictionary can be
        retrieved from a pretrained checkpoint by leveraging the
        [`~image_processing_utils.ImageProcessingMixin.to_dict`] method.
    kwargs (`dict[str, Any]`):
        Additional parameters from which to initialize the image processor object.

Returns:
    [`~image_processing_utils.ImageProcessingMixin`]: The image processor object instantiated from those
    parameters.
return_unused_kwargsFNzImage processor z	: kwargs z were applied for backward compatibility. To avoid this warning, add them to valid_kwargs: create a custom TypedDict extending ImagesKwargs with these keys and set it as the `valid_kwargs` class attribute.r   )copyr&   updater'   valid_kwargs__annotations__reversedlistkeyshasattrr(   appendr*   warning_oncer   rT   )	r3   r>   r-   rv   kvri   
extra_keysr.   s	            r    r=   ImageProcessingMixin.from_dict^  s-   "  488:%zz*@%H##flln$nndaM]M]MmMmHmTQTn$no5 45 
D/0C,,<L<L<\<\1\fjjd.CD!!#& 1 "3<<.	* Fa b 	&&789"**"") %os   E
$E
c                 x    [         R                  " U R                  5      nU R                  R                  US'   U$ )z
Serializes this instance to a Python dictionary.

Returns:
    `dict[str, Any]`: Dictionary of all the attributes that make up this image processor instance.
image_processor_type)rw   deepcopy__dict__	__class__r   )r,   outputs     r    to_dictImageProcessingMixin.to_dict  s0     t}}-)-)@)@%&r   	json_filec                     [        USS9 nUR                  5       nSSS5        [        R                  " W5      nU " S0 UD6$ ! , (       d  f       N,= f)a  
Instantiates a image processor of type [`~image_processing_utils.ImageProcessingMixin`] from the path to a JSON
file of parameters.

Args:
    json_file (`str` or `os.PathLike`):
        Path to the JSON file containing the parameters.

Returns:
    A image processor of type [`~image_processing_utils.ImageProcessingMixin`]: The image_processor object
    instantiated from that JSON file.
utf-8encodingNr   )openreadjsonloads)r3   r   readertextr>   s        r    from_json_file#ImageProcessingMixin.from_json_file  sG     )g.&;;=D /#zz$/*)** /.s   A
Ac                     U R                  5       nUR                  5        H8  u  p#[        U[        R                  5      (       d  M&  UR                  5       X'   M:     [        R                  " USSS9S-   $ )z
Serializes this instance to a JSON string.

Returns:
    `str`: String containing all the attributes that make up this feature_extractor instance in JSON format.
   T)indent	sort_keys
)r   r'   
isinstancenpndarraytolistr   dumps)r,   
dictionaryr.   r/   s       r    to_json_string#ImageProcessingMixin.to_json_string  s\     \\^
$**,JC%,,"',,.
 - zz*Q$?$FFr   json_file_pathc                     [        USSS9 nUR                  U R                  5       5        SSS5        g! , (       d  f       g= f)z
Save this instance to a JSON file.

Args:
    json_file_path (`str` or `os.PathLike`):
        Path to the JSON file in which this image_processor instance's parameters will be saved.
wr   r   N)r   writer   )r,   r   writers      r    rS   !ImageProcessingMixin.to_json_file  s3     .#8FLL,,./ 988s	    5
Ac                 T    U R                   R                   SU R                  5        3$ )N )r   r   r   )r,   s    r    __repr__ImageProcessingMixin.__repr__  s(    ..))*!D,?,?,A+BCCr   c                     [        U[        5      (       d  UR                  nSSKJs  Jn  [        X!5      (       d  [        U S35      eXl        g)aK  
Register this class with a given auto class. This should only be used for custom image processors as the ones
in the library are already mapped with `AutoImageProcessor `.



Args:
    auto_class (`str` or `type`, *optional*, defaults to `"AutoImageProcessor "`):
        The auto class to register this new image processor with.
r   Nz is not a valid auto class.)	r   rj   r   transformers.models.automodelsautor~   
ValueErrorrQ   )r3   
auto_classauto_modules      r    register_for_auto_class,ImageProcessingMixin.register_for_auto_class  sE     *c**#,,J66{//
|+FGHH$r   image_url_or_urlsc                    [        U[        [        45      (       a!  U Vs/ s H  o R                  U5      PM     sn$ [        U[        5      (       a  [        U5      $ [        U5      (       a  U$ [        S[        U5       35      es  snf )z
Convert a single or a list of urls into the corresponding `PIL.Image` objects.

If a single url is passed, the return value will be a single object. If a list is passed a list of objects is
returned.
z=only a single or a list of entries is supported but got type=)	r   r|   tuplefetch_imagesrj   r   r
   	TypeErrortype)r,   r   xs      r    r   !ImageProcessingMixin.fetch_images  s     '$772CD2CQ%%a(2CDD)3///00-..$$[\`ar\s[tuvv Es   Br   )NFFNmain)F)AutoImageProcessor)r   r   r   r   r   rQ   r1   classmethodr   r   rj   rI   PathLikeboolr?   rY   r   dictr   r<   r=   r   r   r   rS   r   r   r|   r   r   r   r   r    r   r   =   s   
 K  /3$!&#'`=$%`='*R[['8`= $t+`= 	`=
 `= TzD `= `= 
`= `=D.-cBKK.? .-d .-` t,,/"++,=t,	tCH~tCH~-	.t, t,l &#T#s(^ &# &#P
c3h 
 +sR[['8 + +$G G	03+< 	0D % %*wcDIoT#Y.O wr   rb   r   zimage processor file)objectobject_classobject_files)$rw   r   rI   typingr   r   numpyr   huggingface_hubr   r   dynamic_module_utilsr   feature_extraction_utilsr	   BaseBatchFeatureimage_utilsr
   r   utilsr   r   r   r   r   r   	utils.hubr   r   
get_loggerr   r*   r   rB   r   formatr   r   r    <module>r      s      	   8 4 F 3  # 19OP  
		H	%
#  jw> jwZ $--A-M-M#N   ##++7/C/O/O/W/W/^/^ /CRh 0_ 0$$, 8r   