a PSic$ @sddlmZddlmZddlZddlmZddlmZz edWn6e e fyzZ zede WYdZ [ n dZ [ 00Gd d d eZ e ejd d d Ze ejddddZe jfeje ejdddZd-ejeejdddZd.eje edddZe jdfeje e ejdddZd/ejeejd d!d"Zd0eje ed#d$d%Ze jfeje ejdd&d'Ze jfe e ejd(d)d*Ze jfe e ejd(d+d,ZdS)1)Enum)warnN) _load_library)_log_api_usage_onceimagez'Failed to load image Python extension: c@s$eZdZdZdZdZdZdZdZdS) ImageReadModeaV Support for various modes while reading images. Use ``ImageReadMode.UNCHANGED`` for loading the image as-is, ``ImageReadMode.GRAY`` for converting to grayscale, ``ImageReadMode.GRAY_ALPHA`` for grayscale with transparency, ``ImageReadMode.RGB`` for RGB and ``ImageReadMode.RGB_ALPHA`` for RGB with transparency. rrN) __name__ __module__ __qualname____doc__ UNCHANGEDZGRAYZ GRAY_ALPHARGB RGB_ALPHArrP/var/www/html/django/DPS/env/lib/python3.9/site-packages/torchvision/io/image.pyrs  r)pathreturncCs.tjstjstttjj|}|S)z Reads and outputs the bytes contents of a file as a uint8 Tensor with one dimension. Args: path (str): the path to the file to be read Returns: data (Tensor) )torchjit is_scripting is_tracingr read_fileopsr)rdatarrrr"s r)filenamerrcCs0tjstjstttjj||dS)z Writes the contents of a uint8 tensor with one dimension to a file. Args: filename (str): the path to the file to be written data (Tensor): the contents to be written to the output file N)rrrrr write_filerr)rrrrrr3s r)inputmodercCs4tjstjstttjj||jd}|S)aw Decodes a PNG image into a 3 dimensional RGB or grayscale Tensor. Optionally converts the image to the desired format. The values of the output tensor are uint8 in [0, 255]. Args: input (Tensor[1]): a one dimensional uint8 tensor containing the raw bytes of the PNG image. mode (ImageReadMode): the read mode used for optionally converting the image. Default: ``ImageReadMode.UNCHANGED``. See `ImageReadMode` class for more information on various available modes. Returns: output (Tensor[image_channels, image_height, image_width]) F) rrrrr decode_pngrrvaluer r!outputrrrr"Asr")r compression_levelrcCs0tjstjstttjj||}|S)a  Takes an input tensor in CHW layout and returns a buffer with the contents of its corresponding PNG file. Args: input (Tensor[channels, image_height, image_width]): int8 image tensor of ``c`` channels, where ``c`` must 3 or 1. compression_level (int): Compression factor for the resulting file, it must be a number between 0 and 9. Default: 6 Returns: Tensor[1]: A one dimensional int8 tensor that contains the raw bytes of the PNG file. )rrrrr encode_pngrr)r r'r%rrrr(Xsr()r rr'cCs4tjstjsttt||}t||dS)a Takes an input tensor in CHW layout (or HW in the case of grayscale images) and saves it in a PNG file. Args: input (Tensor[channels, image_height, image_width]): int8 image tensor of ``c`` channels, where ``c`` must be 1 or 3. filename (str): Path to save the image. compression_level (int): Compression factor for the resulting file, it must be a number between 0 and 9. Default: 6 N)rrrrr write_pngr(r)r rr'r%rrrr)ms  r)cpu)r r!devicercCs\tjstjsttt|}|jdkrFtjj ||j |}ntjj ||j }|S)a Decodes a JPEG image into a 3 dimensional RGB or grayscale Tensor. Optionally converts the image to the desired format. The values of the output tensor are uint8 between 0 and 255. Args: input (Tensor[1]): a one dimensional uint8 tensor containing the raw bytes of the JPEG image. This tensor must be on CPU, regardless of the ``device`` parameter. mode (ImageReadMode): the read mode used for optionally converting the image. Default: ``ImageReadMode.UNCHANGED``. See ``ImageReadMode`` class for more information on various available modes. device (str or torch.device): The device on which the decoded image will be stored. If a cuda device is specified, the image will be decoded with `nvjpeg `_. This is only supported for CUDA version >= 10.1 .. betastatus:: device parameter .. warning:: There is a memory leak in the nvjpeg library for CUDA versions < 11.6. Make sure to rely on CUDA 11.6 or above before using ``device="cuda"``. Returns: output (Tensor[image_channels, image_height, image_width]) cuda) rrrrr decode_jpegr+typerrZdecode_jpeg_cudar#)r r!r+r%rrrr-s  r-K)r qualityrcCsHtjstjstt|dks,|dkr4tdtjj||}|S)a  Takes an input tensor in CHW layout and returns a buffer with the contents of its corresponding JPEG file. Args: input (Tensor[channels, image_height, image_width])): int8 image tensor of ``c`` channels, where ``c`` must be 1 or 3. quality (int): Quality of the resulting JPEG file, it must be a number between 1 and 100. Default: 75 Returns: output (Tensor[1]): A one dimensional int8 tensor that contains the raw bytes of the JPEG file. r dz;Image quality should be a positive number between 1 and 100) rrrrr encode_jpeg ValueErrorrr)r r0r%rrrr2s r2)r rr0cCs4tjstjsttt||}t||dS)a Takes an input tensor in CHW layout and saves it in a JPEG file. Args: input (Tensor[channels, image_height, image_width]): int8 image tensor of ``c`` channels, where ``c`` must be 1 or 3. filename (str): Path to save the image. quality (int): Quality of the resulting JPEG file, it must be a number between 1 and 100. Default: 75 N)rrrrr write_jpegr2r)r rr0r%rrrr4s  r4cCs2tjstjstttjj||j}|S)a Detects whether an image is a JPEG or PNG and performs the appropriate operation to decode the image into a 3 dimensional RGB or grayscale Tensor. Optionally converts the image to the desired format. The values of the output tensor are uint8 in [0, 255]. Args: input (Tensor): a one dimensional uint8 tensor containing the raw bytes of the PNG or JPEG image. mode (ImageReadMode): the read mode used for optionally converting the image. Default: ``ImageReadMode.UNCHANGED``. See ``ImageReadMode`` class for more information on various available modes. Returns: output (Tensor[image_channels, image_height, image_width]) ) rrrrr decode_imagerrr#r$rrrr5sr5)rr!rcCs.tjstjsttt|}t||S)aA Reads a JPEG or PNG image into a 3 dimensional RGB or grayscale Tensor. Optionally converts the image to the desired format. The values of the output tensor are uint8 in [0, 255]. Args: path (str): path of the JPEG or PNG image. mode (ImageReadMode): the read mode used for optionally converting the image. Default: ``ImageReadMode.UNCHANGED``. See ``ImageReadMode`` class for more information on various available modes. Returns: output (Tensor[image_channels, image_height, image_width]) )rrrrr read_imagerr5rr!rrrrr6sr6cCst|}tjj||jdS)NT)rrrrr"r#r7rrr _read_png_16sr8)r&)r&)r/)r/)enumrwarningsrr extensionrutilsr ImportErrorOSErrorerstrTensorrrrr"intr(r)r-r2r4r5r6r8rrrrs0     $ (