o oi @sddlmZddlmZddlmZddlmZeddGdddZGd d d eZ eddGd d d Z Gd ddeZ eddGdddZ dS)) annotations) dataclass)Enum)TensorT)frozenc@s"eZdZUdZded<ded<dS) ImageSizezData class to represent image shape. Args: height: image height. width: image width. Example: >>> size = ImageSize(3, 4) >>> size.height 3 >>> size.width 4 z int | TensorheightwidthN__name__ __module__ __qualname____doc____annotations__rrE/home/ubuntu/.local/lib/python3.10/site-packages/kornia/image/base.pyr  rc@s eZdZdZdZdZdZdZdS) ColorSpacez1Enum that represents the color space of an image.rN)r r r rUNKNOWNGRAYRGBBGRrrrrr/s rc@s"eZdZUdZded<ded<dS) PixelFormataQData class to represent the pixel format of an image. Args: color_space: color space. bit_depth: the number of bits per channel. Example: >>> pixel_format = PixelFormat(ColorSpace.RGB, 8) >>> pixel_format.color_space >>> pixel_format.bit_depth 8 r color_spaceint bit_depthNr rrrrr8rrc@seZdZdZdZdZdS) ChannelsOrderz4Enum that represents the channels order of an image.rrN)r r r rCHANNELS_FIRST CHANNELS_LASTrrrrrMsrc@s*eZdZUdZded<ded<ded<dS) ImageLayoutaData class to represent the layout of an image. Args: image_size: image size. channels: number of channels. channels_order: channels order. Example: >>> layout = ImageLayout(ImageSize(3, 4), 3, ChannelsOrder.CHANNELS_LAST) >>> layout.image_size ImageSize(height=3, width=4) >>> layout.channels 3 >>> layout.channels_order r image_sizerchannelsrchannels_orderNr rrrrr"Ts  r"N) __future__r dataclassesrenumr kornia.corerrrrrr"rrrrs