S^(h cddlZddlZddlZddlZddlmZddlmZmZm Z m Z ddl Z ddl Z ddlmZddlmZddlmZmZmZmZmZmZmZmZmZmZmZerddlm Z e d d Z!ejDe#Z$Gd d eZGdd eZ%ee%jLe%_&e%jLjN8e%jLjNjQddde%jL_'yy)N)BytesIO)AnyOptionalTypeVarUnion)custom_object_save) BatchFeature) IMAGE_PROCESSOR_NAMEPushToHubMixinadd_model_info_to_auto_map"add_model_info_to_custom_pipelines cached_file copy_func download_urlis_offline_mode is_remote_urlis_vision_availablelogging)ImageImageProcessorTypeImageProcessingMixin)boundceZdZdZy)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/TensorFlow/Numpy Tensors at initialization. N)__name__ __module__ __qualname____doc__`/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/transformers/image_processing_base.pyr r 7s r r ceZdZdZdZdZdefdZe dde e de ee jfdee ee jfd ed ed ee eefd ed e fdZd de ee jfdefdZede ee jfd eeeefeeefffdZedeeeffdZd eeeffdZede ee jffdZd efdZde ee jffdZdZed!dZde eeeffdZy)"rz This is an image processor mixin used to provide saving/loading functionality for sequential and image feature extractors. Nc |jdd|jdd|_|jD]\}} t|||y#t$r%}t j d|d|d||d}~wwxYw)z'Set elements of `kwargs` as attributes.feature_extractor_typeNprocessor_classz Can't set z with value z for )pop_processor_classitemssetattrAttributeErrorloggererror)selfkwargskeyvalueerrs r!__init__zImageProcessingMixin.__init__Os  +T2 & + To test a pull request you made on the Hub, you can pass `revision="refs/pr/"`. 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} ```r7r8r9r;use_auth_tokenNrThe `use_auth_token` argument is deprecated and will be removed in v5 of Transformers. Please use `token` instead.V`token` and `use_auth_token` are both specified. Please set only the argument `token`.r:)r&warningswarn FutureWarning ValueErrorget_image_processor_dict from_dict) r5r6r7r8r9r:r;r.r>image_processor_dicts r!from_pretrainedz$ImageProcessingMixin.from_pretrainedbst({#1 %5!"%z$4d;  % MME   l#E  #F7O'Cs'C'CDa'lek'l$fs}}1Nr?r:r@zProvided path (z#) should be a directory, not a fileT)exist_okcommit_messagerepo_id)configzImage processor saved in )rMr:)r&rArBrCgetrDospathisfileAssertionErrormakedirssplitsep _create_repo_get_files_timestamps _auto_classr joinr to_json_filer+info_upload_modified_files) r-rIrJr.r>rMrNfiles_timestampsoutput_image_processor_files r!save_pretrainedz$ImageProcessingMixin.save_pretraineds $4d;  % MME zz'4(4 l-F7O 77>>. ) ?>2BBe!fg g NT2 #ZZ(8$?NjjN,@,@,Mb,QRG'd'':6:G#99.I     ' t^D A')ggll>CW&X# 56 /0K/LMN   ' ' -jj) ( ,,,r c |jdd}|jdd}|jdd}|jdd}|jdd}|jdd}|jd d} |jd d} |jd d } |jd t} |jdd} |jdd}|)tjdt| t d|}d|d}| | |d<t r| stjdd} t|}tjj|}tjj|r tjj|| }tjj|r|}d}n2t|r|}t!|}n| } t#||||||| ||| |  } t)|d5}|j+}dddt-j.}|rtjd |||fStjd d!|d"|vrt3|d"||d"<d#|vrt5|d#||d#<||fS#t$$rt&$rt%d|d|d| dwxYw#1swYxYw#t,j0$rt%d|dwxYw)$a 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. r7Nr8Fresume_downloadproxiesr:r>r9r; subfolderimage_processor_filename_from_pipeline _from_autor?r@image processor) file_typefrom_auto_classusing_pipelinez+Offline mode: forcing local_files_only=TrueT) r7r8rerdr9r: user_agentr;rfz 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 fileutf-8encodingz"It looks like the config file at 'z' is not a valid JSON file.zloading configuration file z from cache at auto_mapcustom_pipelines)r&r rArBrCrDrr+r^strrRrSisdirr\rTrrrOSError ExceptionopenreadjsonloadsJSONDecodeErrorr r)r5r6r.r7r8rdrer:r>r9r;rfrh from_pipelinermrois_localimage_processor_fileresolved_image_processor_filereadertextrGs r!rEz-ImageProcessingMixin.get_image_processor_dicts`(JJ{D1 $4e< **%6=**Y- 7D)$4d;!::&8%@::j$/JJ{B/ #)::.HJ^#_  #3T:  **\59  % MME   l#E#4Y  $+8J' (  %5 KKE F# (+,I(J%77==!>? 77==6 7#%77<<0MOg#h 77>>7 8,I )H 8 9#@ ,89V,W )#;  0;1('#1#$3%5)%' 1-4 3gF %&{{} %#'::d#3   KK56S5TU V$V++ KK-.B-C?SpRqr 113M(46S4$Z0"%99;]();<>[<$%78$V++O  67T6UV99V8WX//G.HO  % %## 45R4SSno  s0 J7 KKK(KK K#K5rGc |j}|jdd}d|vrd|vr|jd|d<d|vrd|vr|jd|d<|di|}g}|jD]0\}}t||st ||||j |2|D]}|j|dt jd||r||fS|S)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_kwargsFsize crop_sizeNzImage processor r)copyr&r(hasattrr)appendr+r^)r5rGr.rimage_processor to_remover/r0s r!rFzImageProcessingMixin.from_dicts " 488:%zz*@%H V *> >+1::f+=  ( & [4H%H06 ;0G  -5 45  ,,. &JC,e4  % & "C JJsD ! "  &&789 "F* *" "r cvtj|j}|jj|d<|S)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)rdeepcopy__dict__ __class__r)r-outputs r!to_dictzImageProcessingMixin.to_dicts0t}}-)-)@)@%& r json_filect|d5}|j}dddtj}|di|S#1swY&xYw)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. rprqNr)ryrzr{r|)r5rrrrGs r!from_json_filez#ImageProcessingMixin.from_json_filesN)g . !&;;=D !#zz$/*)** ! !s AA c|j}|jD]3\}}t|tjs!|j ||<5|j dd}|||d<tj|dddzS)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. r'Nr%T)indent sort_keys ) rr( isinstancenpndarraytolistr&r{dumps)r- dictionaryr/r0r's r!to_json_stringz#ImageProcessingMixin.to_json_strings\\^ $**, 1JC%,"',,. 3 1 &>>* This API is experimental and may have some slight breaking changes in the next releases. Args: auto_class (`str` or `type`, *optional*, defaults to `"AutoImageProcessor "`): The auto class to register this new image processor with. rNz is not a valid auto class.) rrurtransformers.models.automodelsautorrDr[)r5 auto_class auto_modules r!register_for_auto_classz,ImageProcessingMixin.register_for_auto_classsC *c*#,,J66{J/ |+FGH H$r image_url_or_urlsc`ddi}t|tr|Dcgc]}|j|c}St|trPt j |d|}|j tjt|jStdt|cc}w)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 User-AgentzuMozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36T)streamheadersz=only a single or a list of entries is supported but got type=) rlist fetch_imagesrurequestsrQraise_for_statusrryrcontent TypeErrortype)r-rrxresponses r!rz!ImageProcessingMixin.fetch_imagess !  ' .2CDQD%%a(D D )3 /||$5dGTH  % % '::gh&6&678 8[\`ar\s[tuv v EsB+)NFFNmain)F)AutoImageProcessor)rrrrr[r2rur4 classmethodrrrrRPathLikerboolrHrbtupledictrrErFrrrr]rrrrrr r!rrGs K 0C08<$!&,0o= $ %o=',S"++-='>o=E#r{{"234o= o=  o= c4i() o=o= o=o=b;-eC4D.E;-TX;-zv,,1#r{{2B,Cv, tCH~tCH~- .v,v,p*#T#s(^*#*#X c3h +uS"++-='>++$GG* 05bkk1A+B 0D%%2weCcN.Cwr rkrzimage processor file)object object_class object_files))rr{rRrAiortypingrrrrnumpyrrdynamic_module_utilsr feature_extraction_utilsr BaseBatchFeatureutilsr r r rrrrrrrrPILrr get_loggerrr+rrJrformatrr r!rs  004F    19OP   H % #  `w>`wF$--A-M-M#N ##++7/C/O/O/W/W/^/^ /CRh0_0$$,8r