`i  ddlmZmZmZddlZddlmZmZm Z m Z m Z ddl m Z mZe r ddlmZddlmZerdd lmZe jeZeed Gd d e ZdS))AnyUnionoverloadN)add_end_docstringsis_torch_availableis_vision_availableloggingrequires_backends)Pipelinebuild_pipeline_init_args)Image) load_image)&MODEL_FOR_IMAGE_TO_IMAGE_MAPPING_NAMEST)has_image_processorc 8eZdZdZdZdZdZdZfdZdZ e de e dfde d dfd Ze de ee edfde d edfd Zde e ee dedfde d e dedfffd Zd ZddZdZxZS)ImageToImagePipelineao Image to Image pipeline using any `AutoModelForImageToImage`. This pipeline generates an image based on a previous image input. Example: ```python >>> from PIL import Image >>> import requests >>> from transformers import pipeline >>> upscaler = pipeline("image-to-image", model="caidas/swin2SR-classical-sr-x2-64") >>> img = Image.open(requests.get("http://images.cocodataset.org/val2017/000000039769.jpg", stream=True).raw) >>> img = img.resize((64, 64)) >>> upscaled_img = upscaler(img) >>> img.size (64, 64) >>> upscaled_img.size (144, 144) ``` This image to image pipeline can currently be loaded from [`pipeline`] using the following task identifier: `"image-to-image"`. See the list of available models on [huggingface.co/models](https://huggingface.co/models?filter=image-to-image). FTctj|i|t|d|tdS)Nvision)super__init__r check_model_typer)selfargskwargs __class__s y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/transformers/pipelines/image_to_image.pyrzImageToImagePipeline.__init__KsJ$)&)))$))) DEEEEEc Ti}i}i}d|vr |d|d<d|vr |d|d<|||fS)Ntimeout head_mask)rrpreprocess_paramspostprocess_paramsforward_paramss r_sanitize_parametersz)ImageToImagePipeline._sanitize_parametersPsV   +1)+< i ( & *0*=N; ' .2DDDrimagesz Image.Imagerreturnc dSNr#rr(rs r__call__zImageToImagePipeline.__call__\s[^[^rc dSr+r#r,s rr-zImageToImagePipeline.__call___smpmprc 8tj|fi|S)a Transform the image(s) passed as inputs. Args: images (`str`, `list[str]`, `PIL.Image` or `list[PIL.Image]`): The pipeline handles three types of images: - A string containing a http link pointing to an image - A string containing a local path to an image - An image loaded in PIL directly The pipeline accepts either a single image or a batch of images, which must then be passed as a string. Images in a batch must all be in the same format: all as http links, all as local paths, or all as PIL images. timeout (`float`, *optional*, defaults to None): The maximum time in seconds to wait for fetching images from the web. If None, no timeout is used and the call may block forever. Return: An image (Image.Image) or a list of images (list["Image.Image"]) containing result(s). If the input is a single image, the return will be also a single image, if the input is a list of several images, it will return a list of transformed images. )rr-)rr(rrs rr-zImageToImagePipeline.__call__bs$4 uww11&111rc |jdi|}|S)Nr#)model)r model_inputs model_outputss r_forwardzImageToImagePipeline._forward~s" 22\22 rNct||}||gd}|jdkr||j}|S)N)r!pt)r(return_tensors)rimage_processor frameworktodtype)rimager!inputss r preprocesszImageToImagePipeline.preprocesssS5'222%%eWT%JJ >T ! !YYtz**F rc g}d|vr|j}|D]}|jdd}tj|dd}|dz tj }| tj|t|dkr|n|dS)Nreconstructionrr )source destinationgo@)r@datasqueezefloatcpuclamp_numpynpmoveaxisroundastypeuint8appendr fromarraylen)rr3r(outputsoutputs r postprocessz ImageToImagePipeline.postprocesss } , ,#2G 3 3F[((**00226688??1EEKKMMF[rBBBFun++--44RX>>F MM%/&11 2 2 2 2VqvvfQi7rr+)__name__ __module__ __qualname____doc___load_processor_load_image_processor_load_feature_extractor_load_tokenizerrr'rrstrrr-listr4r>rT __classcell__)rs@rrr'sp:O #OFFFFF E E E^uS-%78^C^M^^^X^ puT#Y]0C%CDpPSpX\]jXkpppXp2CcM4 ;NNO2[^2 }d=11 22222228 8 8 8 8 8 8 8rr)typingrrrrIrJutilsrrr r r baser rPILr image_utilsrmodels.auto.modeling_autor get_loggerrUloggerrr#rrrhsd('''''''''54444444)((((((SRRRRRR  H % %,,FFFGGk8k8k8k8k88k8k8HGk8k8k8r