T^(h4 XdZddlmZmZmZddlZddlmZm Z m Z ddl m Z m Z mZmZddlmZmZmZmZmZmZmZmZddlmZmZmZmZmZmZerddl Z erddl!Z!ejDe#Z$d Z% dd ejLd ee'd ee'd eee'effdZ(GddeZ)dgZ*y)z%Image processor class for LayoutLMv2.)DictOptionalUnionN)BaseImageProcessor BatchFeature get_size_dict)flip_channel_orderresizeto_channel_dimension_format to_pil_image)ChannelDimension ImageInputPILImageResamplinginfer_channel_dimension_formatmake_list_of_imagesto_numpy_array valid_imagesvalidate_preprocess_arguments) TensorTypefilter_out_non_signature_kwargsis_pytesseract_availableis_vision_availableloggingrequires_backendsctd|d|z ztd|d|z ztd|d|z ztd|d|z zgS)Nirr)int)boxwidthheights x/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/transformers/models/layoutlmv2/image_processing_layoutlmv2.py normalize_boxr$5s` DCFUN #$ DCFVO $% DCFUN #$ DCFVO $%  imagelangtesseract_configinput_data_formatc||nd}t||}|j\}}tj||d|}|d|d|d|d|d f\}} } } } t |D cgc]\} }|j r| }} }t |D cgc] \} }| |vs |}} }t | D cgc] \} }| |vs |} } }t | D cgc] \} }| |vs |} } }t | D cgc] \} }| |vs |} } }t | D cgc] \} }| |vs |} } }g}t | | | | D]$\}}}}||||z||zg}|j|&g}|D]}|jt||| t|t|k(sJd ||fScc}} wcc}} wcc}} wcc}} wcc}} wcc}} w) zdApplies Tesseract OCR on a document image, and returns recognized words + normalized bounding boxes.r)dict)r' output_typeconfigtextlefttopr!r"z-Not as many words as there are bounding boxes) r size pytesseract image_to_data enumeratestripzipappendr$len)r&r'r(r) pil_image image_width image_heightdatawordsr1r2r!r"idxwordirrelevant_indicescoord actual_boxesxywh actual_boxnormalized_boxesr s r#apply_tesseractrK>s,<+G'RU6GHI )K  $ $YTvVf gD&*6lDL$u+tT[}^bck^l&l#E4eV09/?T)#ttzz|#TT#,U#3 Uic4sBT7TT UE U$-dO Ujc5sBT7TE UD U#,S> SZS%S@R5R5 SC S%.u%5 WzsEDV9VU WE W&/&7 Y U3FX;Xe YF YL$UF3( 1aAE1q5) J'( O c; MNO u:-. ._0__ . " "")U U U S W YsH&F.?F. F4!F46 F:F: G%G: GG G )G ceZdZdZdgZddej dddfdedee e fded ed e e d e e d dffd Z ej ddfde jdee e fdede ee efde ee efd e jf dZedddddddej&df dede edee e fded e ed e e d e e de ee efdede ee efd ej.j.fdZxZS)LayoutLMv2ImageProcessora Constructs a LayoutLMv2 image processor. Args: do_resize (`bool`, *optional*, defaults to `True`): Whether to resize the image's (height, width) dimensions to `(size["height"], size["width"])`. Can be overridden by `do_resize` in `preprocess`. size (`Dict[str, int]` *optional*, defaults to `{"height": 224, "width": 224}`): Size of the image after resizing. Can be overridden by `size` in `preprocess`. resample (`PILImageResampling`, *optional*, defaults to `Resampling.BILINEAR`): Resampling filter to use if resizing the image. Can be overridden by the `resample` parameter in the `preprocess` method. apply_ocr (`bool`, *optional*, defaults to `True`): Whether to apply the Tesseract OCR engine to get words + normalized bounding boxes. Can be overridden by `apply_ocr` in `preprocess`. ocr_lang (`str`, *optional*): The language, specified by its ISO code, to be used by the Tesseract OCR engine. By default, English is used. Can be overridden by `ocr_lang` in `preprocess`. tesseract_config (`str`, *optional*, defaults to `""`): Any additional custom configuration flags that are forwarded to the `config` parameter when calling Tesseract. For example: '--psm 6'. Can be overridden by `tesseract_config` in `preprocess`. pixel_valuesTNr+ do_resizer3resample apply_ocrocr_langr(returnc t|di|||nddd}t|}||_||_||_||_||_||_y)N)r"r!) super__init__r rOr3rPrQrRr() selfrOr3rPrQrRr(kwargs __class__s r#rXz!LayoutLMv2ImageProcessor.__init__s[ "6"'tc-JT""   "  0r%r& data_formatr)c t|}d|vsd|vrtd|j|d|df}t|f||||d|S)a Resize an image to `(size["height"], size["width"])`. Args: image (`np.ndarray`): Image to resize. size (`Dict[str, int]`): Dictionary in the format `{"height": int, "width": int}` specifying the size of the output image. resample (`PILImageResampling`, *optional*, defaults to `PILImageResampling.BILINEAR`): `PILImageResampling` filter to use when resizing the image e.g. `PILImageResampling.BILINEAR`. data_format (`ChannelDimension` or `str`, *optional*): The channel dimension format for the output image. If unset, the channel dimension format of the input image is used. Can be one of: - `"channels_first"` or `ChannelDimension.FIRST`: image in (num_channels, height, width) format. - `"channels_last"` or `ChannelDimension.LAST`: image in (height, width, num_channels) format. - `"none"` or `ChannelDimension.NONE`: image in (height, width) format. input_data_format (`ChannelDimension` or `str`, *optional*): The channel dimension format for the input image. If unset, the channel dimension format is inferred from the input image. Can be one of: - `"channels_first"` or `ChannelDimension.FIRST`: image in (num_channels, height, width) format. - `"channels_last"` or `ChannelDimension.LAST`: image in (height, width, num_channels) format. - `"none"` or `ChannelDimension.NONE`: image in (height, width) format. Returns: `np.ndarray`: The resized image. r"r!zFThe `size` dictionary must contain the keys `height` and `width`. Got )r3rPr\r))r ValueErrorkeysr )rYr&r3rPr\r)rZ output_sizes r#r zLayoutLMv2ImageProcessor.resizesyFT" 4 7$#6efjfofofqerst tH~tG}5   #/     r%imagesreturn_tensorsc 4||n |j}||n |j}t|}||n |j}||n |j}||n |j }||n |j }t|}t|s tdt||||D cgc] } t| }} | t|d} |rKt|dg} g} |D]6} t| ||| \}}| j|| j|8|r"|D cgc]} |j!| ||| }} |D cgc]} t#| | }} |D cgc]} t%| | | }} t'd|i| }|r  |d < |d <|Scc} wcc} wcc} wcc} w) a Preprocess an image or batch of images. Args: images (`ImageInput`): Image to preprocess. do_resize (`bool`, *optional*, defaults to `self.do_resize`): Whether to resize the image. size (`Dict[str, int]`, *optional*, defaults to `self.size`): Desired size of the output image after resizing. resample (`PILImageResampling`, *optional*, defaults to `self.resample`): Resampling filter to use if resizing the image. This can be one of the enum `PIL.Image` resampling filter. Only has an effect if `do_resize` is set to `True`. apply_ocr (`bool`, *optional*, defaults to `self.apply_ocr`): Whether to apply the Tesseract OCR engine to get words + normalized bounding boxes. ocr_lang (`str`, *optional*, defaults to `self.ocr_lang`): The language, specified by its ISO code, to be used by the Tesseract OCR engine. By default, English is used. tesseract_config (`str`, *optional*, defaults to `self.tesseract_config`): Any additional custom configuration flags that are forwarded to the `config` parameter when calling Tesseract. return_tensors (`str` or `TensorType`, *optional*): The type of tensors to return. Can be one of: - Unset: Return a list of `np.ndarray`. - `TensorType.TENSORFLOW` or `'tf'`: Return a batch of type `tf.Tensor`. - `TensorType.PYTORCH` or `'pt'`: Return a batch of type `torch.Tensor`. - `TensorType.NUMPY` or `'np'`: Return a batch of type `np.ndarray`. - `TensorType.JAX` or `'jax'`: Return a batch of type `jax.numpy.ndarray`. data_format (`ChannelDimension` or `str`, *optional*, defaults to `ChannelDimension.FIRST`): The channel dimension format for the output image. Can be one of: - `ChannelDimension.FIRST`: image in (num_channels, height, width) format. - `ChannelDimension.LAST`: image in (height, width, num_channels) format. zkInvalid image type. Must be of type PIL.Image.Image, numpy.ndarray, torch.Tensor, tf.Tensor or jax.ndarray.)rOr3rPrr4r,)r&r3rPr))input_channel_dimrN)r> tensor_typer?boxes)rOr3r rPrQrRr(rrr^rrrrrKr9r r r r)rYrarOr3rPrQrRr(rbr\r)r& words_batch boxes_batchr?rfr>s r# preprocessz#LayoutLMv2ImageProcessor.preprocesss^"+!6IDNN 'tTYYT"'38!*!6IDNN '38/?/K+QUQfQf$V,F#:  & 6<vay I   dM 2KK *.uh@Pduv u""5)""5) * $ %dXYj kF _eeUZ$U>OPeent ej '{N_ `  .&!9~V 'DM'DM A=  f sFF 8FF)__name__ __module__ __qualname____doc__model_input_namesrBILINEARboolrstrrrrXnpndarrayrrr rFIRSTrrPILImageri __classcell__)r[s@r#rMrMes.((#'9'B'B"&*,1138n1% 1  1 3- 1#3-1 14(:'B'B>BDH . zz. 38n. % . eC)9$9:; . $E#/?*?$@A .  . `%&%)#'+$("&*.;?(8(>(>DHddD>d38n d % d D> d3-d#3-d!sJ!78d&d$E#/?*?$@Ad d'dr%rM)NN)+rmtypingrrrnumpyrrimage_processing_utilsrrr image_transformsr r r r image_utilsrrrrrrrrutilsrrrrrrrur4 get_loggerrjloggerr$rsrqrKrM__all__rVr%r#rs,((UUee      H %'+@D $# ::$# 3-$#sm$# c+;&; <= $#NE1EP & &r%