a d@srddlZddlZddlmZddgddgdd gd d gd d ggZdZGdddeZdddZ ddZ dddZ dS)N) get_similarity_transform_for_cv2gXjK>@gQ` I@g bP@g-I@g״9H@g$Q@gsO@@g_dW@g_m]O@gP W@`pcseZdZfddZZS)FaceWarpExceptioncsdtt|S)Nz In File {}:{})format__file__super__str__)self __class__]/var/www/html/stable-diffusion-webui/repositories/CodeFormer/facelib/detection/align_trans.pyr szFaceWarpException.__str__)__name__ __module__ __qualname__r __classcell__rrr rr srrrFc Cstt}tt}|r8t||}||d7}||7}|r`|d|dkr`|d|dkr`|S|dkr|dkr|dur||Std|d|krdksntd|dks|ddks|ddkr|dur|d|dtj}|t|7}|d|dkr|d|dks"td |dkrZ||d}||d7}|t |tj7}t|t|d}|d|d|d|dkrtd |dtj |d}||}|}|t|} |}| S) a Function: ---------- get reference 5 key points according to crop settings: 0. Set default crop_size: if default_square: crop_size = (112, 112) else: crop_size = (96, 112) 1. Pad the crop_size by inner_padding_factor in each side; 2. Resize crop_size into (output_size - outer_padding*2), pad into output_size with outer_padding; 3. Output reference_5point; Parameters: ---------- @output_size: (w, h) or None size of aligned face image @inner_padding_factor: (w_factor, h_factor) padding factor for inner (w, h) @outer_padding: (w_pad, h_pad) each row is a pair of coordinates (x, y) @default_square: True or False if True: default crop_size = (112, 112) else: default crop_size = (96, 112); !!! make sure, if output_size is not None: (output_size - outer_padding) = some_scale * (default crop_size * (1.0 + inner_padding_factor)) Returns: ---------- @reference_5point: 5x2 np.array each row is a pair of transformed coordinates (x, y) rrrNz1No paddings to do, output_size must be None or {}g?z&Not (0 <= inner_padding_factor <= 1.0)zMNot (outer_padding[0] < output_size[0] and outer_padding[1] < output_size[1])z_Must have (output_size - outer_padding)= some_scale * (crop_size * (1.0 + inner_padding_factor)) nparrayREFERENCE_FACIAL_POINTSDEFAULT_CROP_SIZEmaxrrastypeint32roundfloat32) output_sizeinner_padding_factor outer_paddingdefault_squareZtmp_5ptsZ tmp_crop_sizeZ size_diffZsize_bf_outer_pad scale_factorZreference_5pointrrrget_reference_facial_pointssD%    $($   "r&c Cstgdgdg}|jd}t|df|j}t||g}t||g}tj||\}}} } | dkrt|d|d|dg|d |d |d gg}n2| d krt|d|ddg|d |d dgg}|S) a Function: ---------- get affine transform matrix 'tfm' from src_pts to dst_pts Parameters: ---------- @src_pts: Kx2 np.array source points matrix, each row is a pair of coordinates (x, y) @dst_pts: Kx2 np.array destination points matrix, each row is a pair of coordinates (x, y) Returns: ---------- @tfm: 2x3 np.array transform matrix from src_pts to dst_pts )rrr)rrrrrr)rr)rr)rr)rr)rrr)rr shapeonesdtypehstacklinalglstsq) src_ptsZdst_ptstfmn_ptsr)Zsrc_pts_Zdst_pts_Aresranksrrrget_affine_transform_matrixps 4*r5 smilaritycCsB|durD|ddkr&|ddkr&t}nd}d}d}|}t||||}t|} | j} t| dkslt| d krttd | dd kr| j} t|} | j} t| dkst| d krtd | dd kr| j} | j| jkrtd |d krt | dd| dd} n |dkrt | | } n t | | } t || |d|df}|S)a4 Function: ---------- apply affine transform 'trans' to uv Parameters: ---------- @src_img: 3x3 np.array input image @facial_pts: could be 1)a list of K coordinates (x,y) or 2) Kx2 or 2xK np.array each row or col is a pair of coordinates (x, y) @reference_pts: could be 1) a list of K coordinates (x,y) or 2) Kx2 or 2xK np.array each row or col is a pair of coordinates (x, y) or 3) None if None, use default reference facial points @crop_size: (w, h) output face image size @align_type: transform type, could be one of 1) 'similarity': use similarity transform 2) 'cv2_affine': use the first 3 points to do affine transform, by calling cv2.getAffineTransform() 3) 'affine': use all points to do affine transform Returns: ---------- @face_img: output face image with size (w, h) = @crop_size NrrrrFrr'rz2reference_pts.shape must be (K,2) or (2,K) and K>2z/facial_pts.shape must be (K,2) or (2,K) and K>2z5facial_pts and reference_pts must have the same shapeZ cv2_affineaffine)rr&rr r(rminrTcv2getAffineTransformr5r warpAffine)src_imgZ facial_ptsZ reference_pts crop_sizeZ align_typer$r"r#r!Zref_ptsZ ref_pts_shpr.Z src_pts_shpr/Zface_imgrrrwarp_and_crop_faces>"         r?)NrrF)Nrr6) r:numpyrZmatlab_cp2tformrrr Exceptionrr&r5r?rrrrs   ]!