a  žd2ć@s>ddlZddlZddlZdd„Zdd„Ze ”d dd„ƒZdS) éNcCsnt |”}|d}|d}d|d|d|dk |”d|d|d|d|dk|dk |”S)z2cubic function used for calculate_weights_indices.éégų?g@égąæé)ŚtorchŚabsZtype_as)ŚxZabsxZabsx2Zabsx3©r śg/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/basicsr/utils/matlab_functions.pyŚcubics  ’ ’’’r cCsĄ|dkr|r||}t d||”}||ddd|}t ||d”}t |”d} | |d” || ”t d| d| ” d| ” || ”} | |d” || ”| } |dkrĘ|rĘ|t| |ƒ} nt| ƒ} t | d” |d”} | |  || ”} t | dkd”}tj |dddds<|   dd| d”} |   dd| d”} tj |ddddsv|   dd| d”} |   dd| d”} |   ”} |   ”} |   ” d}|   ”|}| |d} | | t|ƒt|ƒfS)a8Calculate weights and indices, used for imresize function. Args: in_length (int): Input length. out_length (int): Output length. scale (float): Scale factor. kernel_width (int): Kernel width. antialisaing (bool): Whether to apply anti-aliasing when downsampling. rgą?rrgķµ ÷Ę°>)Zrel_tolé’’’’)rZlinspaceŚfloorŚmathŚceilŚviewŚexpandr ŚsumŚiscloseŚnarrowŚ contiguousŚminŚmaxŚint)Z in_lengthZ out_lengthŚscaleŚkernelŚ kernel_widthŚ antialiasingrŚuŚleftŚpŚindicesZdistance_to_centerŚweightsZ weights_sumZweights_zero_tmpZ sym_len_sZ sym_len_er r r Ścalculate_weights_indicess8 ,’   r"TcCsÄd}t|ƒjtjkrVd}|jdkr<|dd…dd…df}d}t | ddd”” ”}nd}|jdkrr|  d”}d}|  ”\}}}t   ||”t   ||”}} d} d} t |||| | |ƒ\} } }}t || || | |ƒ\}}}}t |||||”}| d||” |”|dd…d|…dd…f}t |  d”dd d ” ”}| d|”}| dd|” |”|dd…| d…dd…f}t |  d”dd d ” ”}| d|”}| d|||” |”t |||”}|   d”} t|ƒD]`}t| |dƒ}t|ƒD]@}||||| …dd…f dd” | |”|||dd…f<qņqÖt |||||”}| d||” |”|dd…dd…d|…f}t |  d”dd d ” ”}| d|”}| dd|” |”|dd…dd…| d…f}t |  d”dd d ” ”}| d|”}| d|||” |”t ||| ”}|  d”} t| ƒD]X}t||dƒ}t|ƒD]8}||dd…||| …f ||”||dd…|f<qPq4|rž| d”}|rĄ| ”}|sĄ| ddd”}|S) a\imresize function same as MATLAB. It now only supports bicubic. The same scale applies for both height and width. Args: img (Tensor | Numpy array): Tensor: Input image with shape (c, h, w), [0, 1] range. Numpy: Input image with shape (h, w, c), [0, 1] range. scale (float): Scale factor. The same scale applies for both height and width. antialisaing (bool): Whether to apply anti-aliasing when downsampling. Default: True. Returns: Tensor: Output image with shape (c, h, w), [0, 1] range, w/o round. FTrNrrrr r )ŚtypeŚ __module__ŚnpŚ__name__ŚndimrZ from_numpyZ transposeŚfloatZ unsqueezeŚsizerrr"Z FloatTensorrZcopy_ZarangeŚlongZ index_selectŚrangerŚmvZsqueezeŚnumpy)ŚimgrrZ squeeze_flagZ numpy_typeZin_cZin_hZin_wZout_hZout_wrrZ weights_hZ indices_hZ sym_len_hsZ sym_len_heZ weights_wZ indices_wZ sym_len_wsZ sym_len_weZimg_augZ sym_patchZinv_idxZ sym_patch_invZout_1ŚiŚidxŚjZ out_1_augZout_2r r r ŚimresizeUsx    ’  ’      B     : r2)T)rr-r%rr r"Zno_gradr2r r r r Śs  E