fThSSKJrJr SSKrSSKJrJrJrJ r J r SSK J r J r \"5(a SSKJr SSKJr \"5(aSS KJr \ R("\5r\"\ "S S 95"S S \ 55rg))ListUnionN)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^\rSrSrSrU4SjrSrS\\\ \S\ S4S\S\ S44U4Sjjr S r S S jr S r S rU=r$)ImageToImagePipeline'a 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). ch>[TU]"U0UD6 [US5 UR[5 g)Nvision)super__init__r check_model_typer)selfargskwargs __class__s ]/var/www/auris/envauris/lib/python3.13/site-packages/transformers/pipelines/image_to_image.pyrImageToImagePipeline.__init__Fs. $)&)$) DEc N0n0n0nSU;aUSUS'SU;aUSUS'X$U4$)Ntimeout head_mask)rrpreprocess_paramspostprocess_paramsforward_paramss r_sanitize_parameters)ImageToImagePipeline._sanitize_parametersKsN  +1)+< i ( & *0*=N; ' 2DDDr imagesz Image.Imagereturnc &>[TU]"U40UD6$)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. )r__call__)rr*rrs rr-ImageToImagePipeline.__call__Ws4w1&11r c*UR"S0UD6nU$)Nr$)model)r model_inputs model_outputss r_forwardImageToImagePipeline._forwardss 2\2 r c[XS9nURU/SS9nURS:XaURUR5nU$)N)r"pt)r*return_tensors)rimage_processor frameworkto torch_dtype)rimager"inputss r preprocessImageToImagePipeline.preprocesswsH52%%eWT%J >>T !YYt//0F r c/nSUR5;a URnWHnURR5R 5R 5R SS5R5n[R"USSS9nUS-R5R[R5nUR[R"U55 M [!U5S:aU$US$)Nreconstructionrr )source destinationgo@)keysrAdatasqueezefloatcpuclamp_numpynpmoveaxisroundastypeuint8appendr fromarraylen)rr2r*outputsoutputs r postprocess ImageToImagePipeline.postprocess~s }113 3#22GF[[((*002668??1EKKMF[[rBFun++-44RXX>F MM%//&1 2  Vqv7fQi7r r$)N)__name__ __module__ __qualname____firstlineno____doc__rr(rstrrr-r3r>rV__static_attributes__ __classcell__)rs@rrr'sc:F E2CcM4 ;NNO2 }d=11 228 8 8r r)typingrrrKrLutilsrrrr r baser r PILr image_utilsrmodels.auto.modeling_autor get_loggerrXloggerrr$r rrhss5(R   H %,FG`88`8H`8r