,}hW* \ddlmZddlmZddlZddlmZddlmZ edGd d eZ d e d ejfdZde dejd dfdZe j"dfdejde ded ejfdZd#dejded ejfdZd#dejde defdZe j"ddfdejde de ded ejf dZd$dejded ejfdZd$dejde defdZe j"dfdejde ded ejfd Ze j"dfd e de ded ejfd!Ze j"fd e de d ejfd"Zy#e e f$rZ ede d YdZ [ dZ [ wwxYw)%)Enum)warnN) _load_library)_log_api_usage_onceimagez(Failed to load image Python extension: 'z'If you don't plan on using image functionality from `torchvision.io`, you can ignore this warning. Otherwise, there might be something wrong with your environment. Did you have `libjpeg` or `libpng` installed before building `torchvision` from source?c$eZdZdZdZdZdZdZdZy) 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__ UNCHANGEDGRAY GRAY_ALPHARGB RGB_ALPHAS/var/www/html/test/engine/venv/lib/python3.12/site-packages/torchvision/io/image.pyr r s"I DJ CIrr pathreturnctjjs-tjjst t tj jj t|}|S)z Reads and outputs the bytes contents of a file as a uint8 Tensor with one dimension. Args: path (str or ``pathlib.Path``): the path to the file to be read Returns: data (Tensor) ) torchjit is_scripting is_tracingr read_fileopsrstr)rdatas rr!r!'sL 99 ! ! #EII,@,@,BI& 99?? $ $SY /D Krfilenamer$ctjjs-tjjst t tj jj t||y)z Writes the contents of an uint8 tensor with one dimension to a file. Args: filename (str or ``pathlib.Path``): the path to the file to be written data (Tensor): the contents to be written to the output file N) rrrr r write_filer"rr#)r%r$s rr'r'8sG 99 ! ! #EII,@,@,BJ' IIOOs8}d3rFinputmodeapply_exif_orientationctjjs-tjjst t tj jj ||jd|}|S)a 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. apply_exif_orientation (bool): apply EXIF orientation transformation to the output tensor. Default: False. Returns: output (Tensor[image_channels, image_height, image_width]) F) rrrr r decode_pngr"rvaluer(r)r*outputs rr,r,FsS* 99 ! ! #EII,@,@,BJ' YY__ ' 'tzz5BX YF Mrcompression_levelctjjs-tjjst t tj jj ||}|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. )rrrr r encode_pngr"r)r(r0r/s rr2r2asK 99 ! ! #EII,@,@,BJ' YY__ ' '/@ AF Mrctjjs-tjjst t t ||}t||y)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 or ``pathlib.Path``): 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)rrrr r write_pngr2r')r(r%r0r/s rr4r4vsB 99 ! ! #EII,@,@,BI& 0 1Fx rcpudevicectjjs-tjjst t tj |}|jdk(r7tjjj||j|}|Stjjj ||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. The supported modes are: ``ImageReadMode.UNCHANGED``, ``ImageReadMode.GRAY`` and ``ImageReadMode.RGB`` 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"``. apply_exif_orientation (bool): apply EXIF orientation transformation to the output tensor. Default: False. Only implemented for JPEG format on CPU. Returns: output (Tensor[image_channels, image_height, image_width]) cuda) rrrr r decode_jpegr6typer"rdecode_jpeg_cudar-)r(r)r6r*r/s rr9r9sJ 99 ! ! #EII,@,@,BK( \\& !F {{f11%VL M,,UDJJ@VW Mrqualityctjjs-tjjst t |dks|dkDr t dtjjj ||}|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) rrrr r encode_jpeg ValueErrorr"r)r(r<r/s rr?r?sc 99 ! ! #EII,@,@,BK({gmVWW YY__ ( ( 8F Mrctjjs-tjjst t t ||}t||y)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 or ``pathlib.Path``): 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)rrrr r write_jpegr?r')r(r%r<r/s rrBrBsA 99 ! ! #EII,@,@,BJ'  (Fx rctjjs-tjjst t tj jj ||j|}|S)aS 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. apply_exif_orientation (bool): apply EXIF orientation transformation to the output tensor. Default: False. Returns: output (Tensor[image_channels, image_height, image_width]) ) rrrr r decode_imager"rr-r.s rrDrDsQ. 99 ! ! #EII,@,@,BL) YY__ ) )%=S TF Mrctjjs-tjjst t t |}t|||S)a 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 or ``pathlib.Path``): 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. apply_exif_orientation (bool): apply EXIF orientation transformation to the output tensor. Default: False. Returns: output (Tensor[image_channels, image_height, image_width]) )r*)rrrr r read_imager!rD)rr)r*r$s rrFrFsD( 99 ! ! #EII,@,@,BJ' T?D d;Q RRrct|}tjjj ||j dS)NT)r!rr"rr,r-)rr)r$s r _read_png_16rHs, T?D 99?? % %dDJJ ==r))K)enumrwarningsrr extensionrutilsr ImportErrorOSErrorer r#Tensorr!r'rboolr,intr2r4r9r?rBrDrFrHrrrrUsC %''D$CELL" 4 4ELL 4T 40=/F/Fgl <<,`d \\6ells5<<*!U\\!S!S!((11#( , <<, , ,! ,  \\ ,^u||c5<<0!ell!c!C!$0=/F/Fgl <<,`d \\<&3%<%<]bS S"SVZS \\S43@2I2I>s>->ell>W W 21#6b csF F+ F&&F+