rh_dZddlmZmZddlZddlmZmZm Z ddl m Z m Z m Z mZmZddlmZmZmZmZmZmZmZmZmZmZmZmZddlmZmZm Z dd l!m"Z"e jFe$Z%e"rddl&Z&Gd d eZ'd gZ(y) z!Image processor class for Nougat.)OptionalUnionN)BaseImageProcessor BatchFeature get_size_dict)get_resize_output_image_sizepadresizeto_channel_dimension_format to_pil_image) IMAGENET_DEFAULT_MEANIMAGENET_DEFAULT_STDChannelDimension ImageInputPILImageResamplingget_image_sizeinfer_channel_dimension_formatis_scaled_imagemake_list_of_imagesto_numpy_array valid_imagesvalidate_preprocess_arguments) TensorTypefilter_out_non_signature_kwargslogging)is_vision_availablec%eZdZdZdgZdddej ddddddddf deded ee e e fd ed ed ed edede e e fdedee e ee fdee e ee fddffd Zdej"fdZdZ d#dej"de deedee e efdej"f dZ d$dej,d e e e fdee e efdee e efdej,f dZ d$dej,d e e e fdee e efdee e efdej,f dZej2ddfdej,d e e e fd edee e efdee e efdej,f dZej2ddfdej,d e e e fd edee e efdee e efdej,f dZedddddddddddddej:dfd edeedeed ee e e fd ed eed eed eedeedee e e fdeedee e ee fdee e ee fd!ee e efdeedee e efde jBjBf"d"Z"xZ#S)%NougatImageProcessora Constructs a Nougat image processor. Args: do_crop_margin (`bool`, *optional*, defaults to `True`): Whether to crop the image margins. do_resize (`bool`, *optional*, defaults to `True`): Whether to resize the image's (height, width) dimensions to the specified `size`. Can be overridden by `do_resize` in the `preprocess` method. size (`dict[str, int]` *optional*, defaults to `{"height": 896, "width": 672}`): Size of the image after resizing. Can be overridden by `size` in the `preprocess` method. resample (`PILImageResampling`, *optional*, defaults to `Resampling.BILINEAR`): Resampling filter to use if resizing the image. Can be overridden by `resample` in the `preprocess` method. do_thumbnail (`bool`, *optional*, defaults to `True`): Whether to resize the image using thumbnail method. do_align_long_axis (`bool`, *optional*, defaults to `False`): Whether to align the long axis of the image with the long axis of `size` by rotating by 90 degrees. do_pad (`bool`, *optional*, defaults to `True`): Whether to pad the images to the largest image size in the batch. do_rescale (`bool`, *optional*, defaults to `True`): Whether to rescale the image by the specified scale `rescale_factor`. Can be overridden by the `do_rescale` parameter in the `preprocess` method. rescale_factor (`int` or `float`, *optional*, defaults to `1/255`): Scale factor to use if rescaling the image. Can be overridden by the `rescale_factor` parameter in the `preprocess` method. do_normalize (`bool`, *optional*, defaults to `True`): Whether to normalize the image. Can be overridden by `do_normalize` in the `preprocess` method. image_mean (`float` or `list[float]`, *optional*, defaults to `IMAGENET_DEFAULT_MEAN`): Mean to use if normalizing the image. This is a float or list of floats the length of the number of channels in the image. Can be overridden by the `image_mean` parameter in the `preprocess` method. image_std (`float` or `list[float]`, *optional*, defaults to `IMAGENET_DEFAULT_STD`): Image standard deviation. pixel_valuesTNFgp?do_crop_margin do_resizesizeresample do_thumbnaildo_align_long_axisdo_pad do_rescalerescale_factor do_normalize image_mean image_stdreturnc  t|di| ||nddd}t|}||_||_||_||_||_||_||_ ||_ | |_ | |_ | | nt|_| | |_yt|_y)Nii)heightwidth)super__init__rr!r"r#r$r%r&r'r(r)r*rr+rr,)selfr!r"r#r$r%r&r'r(r)r*r+r,kwargs __class__s /var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/transformers/models/nougat/image_processing_nougat.pyr3zNougatImageProcessor.__init__[s "6"'tc-JT","   ("4 $,((2(>*DY&/&;AUimagectjtj|}|ddddgf}|jddd}|S)zGThis is a reimplementation of a findNonZero function equivalent to cv2.Nr)np column_stacknonzeroreshape)r4r9non_zero_indicesidxvecs r7python_find_non_zeroz)NougatImageProcessor.python_find_non_zero}sC??2::e+<=!!aV),Aq) r8ctj|djt}tj|djt}|d|d}}|d|z dz}|d|z dz}||||fS)zHThis is a reimplementation of a BoundingRect function equivalent to cv2.rr;)axisrr;)r>minastypeintmax)r4 coordinates min_values max_valuesx_miny_minr0r/s r7python_bounding_rectz)NougatImageProcessor.python_bounding_rects~VVKf5<arrayconvertrIuint8rKrHr rLASTrDrQcrop)r4r9rRrSrTdatamax_valmin_valgraycoordsrOrPr0r/s r7 crop_marginz NougatImageProcessor.crop_margins,  $ >u E U6GHxx c*+22288<((*((* g HHUOE/7HJZJ_J_`E*,E;@QR  L  Lw7W#45;n$**40&*&?&?&G#ueV E5%%-HI&&rxx0+E3DFVF[F[\S^Ri '{rot90r ) r4r9r#rSrT input_height input_width output_height output_widthrot_axess r7align_long_axisz$NougatImageProcessor.align_long_axiss.%35FW$X! k&*8nd7m|  $ >u E   0 5 5 5H "2"8"8 8H89J8KLM M = ([<-G = ([<-GHHUAH5E  "/{VghE r8c|d|d}}t||\}}||z } ||z } | dz} | dz} | | z } | | z }| | f| |ff}t||||S)a Pad the image to the specified size at the top, bottom, left and right. Args: image (`np.ndarray`): The image to be padded. size (`dict[str, int]`): The size `{"height": h, "width": w}` to pad the image to. data_format (`str` or `ChannelDimension`, *optional*): The data format of the output image. If unset, the same format as the input image is used. input_data_format (`ChannelDimension` or `str`, *optional*): The channel dimension format of the input image. If not provided, it will be inferred. r/r0rer=)rSrT)rr )r4r9r#rSrTrorprmrn delta_width delta_heightpad_toppad_left pad_bottom pad_rightpaddings r7 pad_imagezNougatImageProcessor.pad_images('+8nd7m| $25FW$X! k"[0 $|3 !#!#!G+ (* Z(8Y*?@5'{N_``r8c t||\}}|d|d} } t|| } t|| } | |k(r| |k(r|S||kDrt|| z|z } n||kDrt|| z|z } t|f| | f|d||d|S)as Resize the image to make a thumbnail. The image is resized so that no dimension is larger than any corresponding dimension of the specified size. Args: image (`np.ndarray`): The image to be resized. size (`dict[str, int]`): The size `{"height": h, "width": w}` to resize the image to. resample (`PILImageResampling`, *optional*, defaults to `PILImageResampling.BICUBIC`): The resampling filter to use. data_format (`Optional[Union[str, ChannelDimension]]`, *optional*): The data format of the output image. If unset, the same format as the input image is used. input_data_format (`ChannelDimension` or `str`, *optional*): The channel dimension format of the input image. If not provided, it will be inferred. rer/r0g@)r#r$ reducing_gaprSrT)rrHrJr ) r4r9r#r$rSrTr5rmrnrorpr/r0s r7 thumbnailzNougatImageProcessor.thumbnails2%35FW$X! k&*8nd7m| \=1K. \ !e{&:L + % f,|;vay I  `fgW\d&&u@Q&RgFg ouvfkd**5tO`*avFv $ %dXYj kF ouvfkdnn5tO`navFv ouvfkdnn5tO`navFv $ 5Rc dF $U^opF ou ej '{N_ `  '>BB_=hw ww   s6;I& I+,I0I53I:I?9JJ >J)NN)NN)$__name__ __module__ __qualname____doc__model_input_namesrBILINEARboolrdictstrrJrfloatlistr3r>rYrDrQrrcndarrayrrr{BICUBICr~r rrjrrPILImager __classcell__)r6s@r7rr6s  D(( $)-'9'B'B!#(,3!:>9= V V VtCH~& V % V  V! V V Vc5j) V VU5$u+#567 VE%e"456 V  VD"((+"26DH 2xx22./ 2 $E#/?*?$@A 2  2r?CDH -zz-38n-eC)9$9:; - $E#/?*?$@A -  -f?CDH !azz!a38n!aeC)9$9:; !a $E#/?*?$@A !a  !aP(:'A'A>BDH 0 zz0 38n0 % 0 eC)9$9:; 0 $E#/?*?$@A 0  0 n(:'A'A>BDH %zz%38n%% % eC)9$9:; % $E#/?*?$@A % %N%&*.$()-'+'+-1!%%)6:'+:>9=;?2B2H2HDH#WCWC!WCD> WC tCH~& WC % WCtnWC%TNWCWCTNWC!sEz!23WCtnWCU5$u+#567WCE%e"456WC!sJ!78WC ./!WC"$E#/?*?$@A#WC$ %WC'WCr8r))rtypingrrnumpyr>image_processing_utilsrrrimage_transformsr r r r r image_utilsrrrrrrrrrrrrutilsrrrutils.import_utilsr get_loggerrrrr__all__r1r8r7rs{("UU    JI5   H %QC-QCh " "r8