a d@sz ddlZWney"dZYn0ddlZddlmZddlZddlmZmZddl m Z ddl m Z ddl mZddlmZGdd d eZeeed d d Zeed ddZeed ddZdeeeedddZdS)N)Enum)rgb_to_grayscale rgba_to_rgb)grayscale_to_rgb) rgb_to_rgba)Tensor) KORNIA_CHECKc@s(eZdZdZdZdZdZdZdZdZ dS) ImageLoadTypez'Enum to specify the desired image type.rN) __name__ __module__ __qualname____doc__ UNCHANGEDGRAY8RGB8RGBA8GRAY32RGB32rrU/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/io/io.pyr sr ) path_filedevicereturncCs6t|}tjj|}|t|ddd S)Nr rr ) kornia_rsZ read_image_rstorchutilsZdlpackZ from_dlpacktorZpermuteclone)rrZ cv_tensorZ th_tensorrrrload_image_to_tensors r#)imagercCst|jtjk|dSNgo@)rdtyperuint8floatr$rrr to_float32+sr*cCst|jtjk|dSr%)rr&rfloat32mulbyter)rrrto_uint80sr.cpu)r desired_typerrc Csltdurtdttj|d|t||}|tjkr@|S|tj kr|j ddkrh|j t j krh|S|j ddkr|j t j krt|}|S|j ddkr|j t j krttt|}t|Sn|tjkr|j ddkr|j t j kr|S|j ddkrd|j t j krdt|}|SnH|tjkr`|j ddkrd|j t j krdtt|d}t|Sn|tjkr|j ddkr|j t j krt|S|j ddkr|j t j krtt|}|S|j ddkrd|j t j krdttt|}|Snp|tjkrV|j ddkr&|j t j kr&t|S|j ddkrd|j t j krdtt|}|Sntd |tgS) aMRead an image file and decode using the Kornia Rust backend. Args: path_file: Path to a valid image file. desired_type: the desired image type, defined by color space and dtype. device: the device where you want to get your image placed. Return: Image tensor with shape :math:`(3,H,W)`. NzGThe io API is not available: `pip install kornia_rs` in a Linux system.zInvalid file: rr r r gzUnknown type: )rModuleNotFoundErrorrospathisfiler#r rrshaper&rr'rrr*r.rrrrrrNotImplementedErrorr) rr0rr$Zgray8Zgray32Zrgb8Zrgba32Zrgb32rrr load_image5sT            r7)r/)r ImportErrorr2enumrrZ kornia.colorrrZkornia.color.grayrZkornia.color.rgbrZ kornia.corerZkornia.testingrr strr#r*r.r7rrrrs