a  dQ@sBddlZddlZddlZddZd ddZddd Zdd d ZdS)NcCsp|}|jdvrZ|jd|jd}}||||}}|d||d||df}ntd|jd|S)zMod crop images, used during testing. Args: img (ndarray): Input image. scale (int): Scale factor. Returns: ndarray: Result image. )rN.zWrong img ndim: .)copyndimshape ValueError)imgscalehwZ h_remainderZ w_remainderr`/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/basicsr/data/transforms.pymod_crops   rc st|ts|g}t|ts |g}t|dr2dnd}|dkrp|ddd\}}|ddd\}} n,|djdd\}}|djdd\}} ||||ks| ||krtd|d| d |d d |d|d |ks|kr"td |d|ddd|d td|td||dkrdfdd|D}nfdd|D}t |t ||dkrfdd|D}nfdd|D}t |dkr|d}t |dkr|d}||fS)aePaired random crop. Support Numpy array and Tensor inputs. It crops lists of lq and gt images with corresponding locations. Args: img_gts (list[ndarray] | ndarray | list[Tensor] | Tensor): GT images. Note that all images should have the same shape. If the input is an ndarray, it will be transformed to a list containing itself. img_lqs (list[ndarray] | ndarray): LQ images. Note that all images should have the same shape. If the input is an ndarray, it will be transformed to a list containing itself. gt_patch_size (int): GT patch size. scale (int): Scale factor. gt_path (str): Path to ground-truth. Default: None. Returns: list[ndarray] | ndarray: GT images and LQ images. If returned results only have one element, just return ndarray. rZTensorZNumpyNrzScale mismatches. GT (z, z ) is not zx zmultiplication of LQ (z).zLQ (z) is smaller than patch size (z). Please remove rc s4g|],}|ddddfqSNr.0vleft lq_patch_sizetoprr Mz&paired_random_crop..cs*g|]"}|dfqS.rrrrrrOrc s4g|],}|ddddfqSrrr gt_patch_sizeleft_gttop_gtrrrTrcs*g|]"}|dfqSrrrrrrrVrr) isinstancelisttorchZ is_tensorsizerr randomrandintintlen) Zimg_gtsZimg_lqsrr Zgt_pathZ input_typeZh_lqZw_lqZh_gtZw_gtr)rrrrrr rpaired_random_cropsJ     r)TFcsotdk|otdk|o.tdkfddfddt|ts`|g}fdd|D}t|dkr|d }|d urt|ts|g}fd d|D}t|dkr|d }||fS|r|ffS|Sd S) amAugment: horizontal flips OR rotate (0, 90, 180, 270 degrees). We use vertical flip and transpose for rotation implementation. All the images in the list use the same augmentation. Args: imgs (list[ndarray] | ndarray): Images to be augmented. If the input is an ndarray, it will be transformed to a list. hflip (bool): Horizontal flip. Default: True. rotation (bool): Ratotation. Default: True. flows (list[ndarray]: Flows to be augmented. If the input is an ndarray, it will be transformed to a list. Dimension is (h, w, 2). Default: None. return_status (bool): Return the status of flip and rotation. Default: False. Returns: list[ndarray] | ndarray: Augmented images and flows. If returned results only have one element, just return ndarray. g?cs:rt|d|r$t|d|r6|ddd}|S)Nrrrcv2ZflipZ transpose)r hfliprot90vfliprr_augmentxszaugment.._augmentcsr0t|d||dddddfd9<r`t|d||dddddfd9<r|ddd}|ddddddgf}|S)Nrrrr*)flowr,rr _augment_flowszaugment.._augment_flowcsg|] }|qSrr)rr )r0rrrrzaugment..rrNcsg|] }|qSrr)rr2)r3rrrr)r%r!r"r()Zimgsr-ZrotationZflowsZ return_statusr)r0r3r-r.r/raugment^s(     r4?cCsN|jdd\}}|dur*|d|df}t|||}t||||f}|S)aRotate image. Args: img (ndarray): Image to be rotated. angle (float): Rotation angle in degrees. Positive values mean counter-clockwise rotation. center (tuple[int]): Rotation center. If the center is None, initialize it as the center of the image. Default: None. scale (float): Isotropic scale factor. Default: 1.0. Nr)rr+ZgetRotationMatrix2DZ warpAffine)r Zanglecenterr r r ZmatrixZ rotated_imgrrr img_rotates r7)N)TTNF)Nr5)r+r%r#rr)r4r7rrrrs  D C