grFdZddlZddlmZddlmZGddeZdS)z Processor class for TrOCR. N)contextmanager)ProcessorMixinceZdZdZddgZdZdZdfd ZdZd Z d Z e d Z e d Ze d ZxZS)TrOCRProcessora Constructs a TrOCR processor which wraps a vision image processor and a TrOCR tokenizer into a single processor. [`TrOCRProcessor`] offers all the functionalities of [`ViTImageProcessor`/`DeiTImageProcessor`] and [`RobertaTokenizer`/`XLMRobertaTokenizer`]. See the [`~TrOCRProcessor.__call__`] and [`~TrOCRProcessor.decode`] for more information. Args: image_processor ([`ViTImageProcessor`/`DeiTImageProcessor`], *optional*): An instance of [`ViTImageProcessor`/`DeiTImageProcessor`]. The image processor is a required input. tokenizer ([`RobertaTokenizer`/`XLMRobertaTokenizer`], *optional*): An instance of [`RobertaTokenizer`/`XLMRobertaTokenizer`]. The tokenizer is a required input. image_processor tokenizerAutoImageProcessor AutoTokenizerNc ,d}d|vr/tjdt|d}||n|}|t d|t dt |||j|_d|_ dS)Nfeature_extractorzhThe `feature_extractor` argument is deprecated and will be removed in v5, use `image_processor` instead.z)You need to specify an `image_processor`.z"You need to specify a `tokenizer`.F) warningswarn FutureWarningpop ValueErrorsuper__init__rcurrent_processor_in_target_context_manager)selfrr kwargsr __class__s f/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/models/trocr/processing_trocr.pyrzTrOCRProcessor.__init__,s  & ( ( M    !' +> ? ? -<-H//N_  "HII I  ABB B )444!%!5*/'''cb|jr |j|i|S|dd}|dd}t|dkr|d}|dd}||t d||j|g|Ri|}||j|fi|}||S||S|d|d<|S) a When used in normal mode, this method forwards all its arguments to AutoImageProcessor's [`~AutoImageProcessor.__call__`] and returns its output. If used in the context [`~TrOCRProcessor.as_target_processor`] this method forwards all its arguments to TrOCRTokenizer's [`~TrOCRTokenizer.__call__`]. Please refer to the doctsring of the above two methods for more information. imagesNtextrzBYou need to specify either an `images` or `text` input to process. input_idslabels)rrrlenrrr )rargsrrrinputs encodingss r__call__zTrOCRProcessor.__call__@s  * ;)4)4:6:: :Hd++zz&$'' t99q==!WF8D >dlabb b  )T)&B4BBB6BBF  &t66v66I <M ^ (5F8 Mrc&|jj|i|S)z This method forwards all its arguments to TrOCRTokenizer's [`~PreTrainedTokenizer.batch_decode`]. Please refer to the docstring of this method for more information. )r batch_decoderr#rs rr(zTrOCRProcessor.batch_decodeas +t~*D;F;;;rc&|jj|i|S)z This method forwards all its arguments to TrOCRTokenizer's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )r decoder)s rr+zTrOCRProcessor.decodehs %t~$d5f555rc#Ktjdd|_|j|_dV|j|_d|_dS)z Temporarily sets the tokenizer for processing the input. Useful for encoding the labels when fine-tuning TrOCR. z`as_target_processor` is deprecated and will be removed in v5 of Transformers. You can process your labels by using the argument `text` of the regular `__call__` method (either in the same call as your images inputs, or in a separate call.TNF)rrrr rrrs ras_target_processorz"TrOCRProcessor.as_target_processorosW   9   +/'!% !%!5*/'''rcDtjdt|jS)Nzg`feature_extractor_class` is deprecated and will be removed in v5. Use `image_processor_class` instead.)rrrimage_processor_classr-s rfeature_extractor_classz&TrOCRProcessor.feature_extractor_classs' u    ))rcDtjdt|jS)Nz[`feature_extractor` is deprecated and will be removed in v5. Use `image_processor` instead.)rrrrr-s rr z TrOCRProcessor.feature_extractors' i    ##r)NN)__name__ __module__ __qualname____doc__ attributesr0tokenizer_classrr&r(r+rr.propertyr1r __classcell__)rs@rrrs  $[1J0%O000000(B<<<666 0 0^ 0**X*$$X$$$$$rr)r6r contextlibrprocessing_utilsrrrrr>s|%%%%%%......t$t$t$t$t$^t$t$t$t$t$r