a d@sddlZddlmZmZddlmZddlmZGdddeZ ddZ d d Z d'd d Z d(d dZ d)ddZddZd*ddZedkrgdZgdZgdZgdZeeefjZeeefjZedeeedeeeee\ZZedeeedeeeded eeeejdd!ffZ e!e eZ"ee"ed"e eeZ"ee"ed#ed$eeeejdd!ffZ#e!e#eZ$ee$ed%e eeZ$ee$e eeZ$ed&ee$dS)+N)invlstsq) matrix_rank)normcseZdZfddZZS)MatlabCp2tormExceptioncsdtt|S)Nz In File {}:{})format__file__super__str__)self __class__a/var/www/html/stable-diffusion-webui/repositories/CodeFormer/facelib/detection/matlab_cp2tform.pyr szMatlabCp2tormException.__str__)__name__ __module__ __qualname__r __classcell__rrr rrsrcCsBt|t|jddff}t||}|ddddf}|S)ap Function: ---------- apply affine transform 'trans' to uv Parameters: ---------- @trans: 3x3 np.array transform matrix @uv: Kx2 np.array each row is a pair of coordinates (x, y) Returns: ---------- @xy: Kx2 np.array each row is a pair of transformed coordinates (x, y) rN)nphstackonesshapedot)transuvxyrrrtformfwd s rcCst|}t||}|S)a Function: ---------- apply the inverse of affine transform 'trans' to uv Parameters: ---------- @trans: 3x3 np.array transform matrix @uv: Kx2 np.array each row is a pair of coordinates (x, y) Returns: ---------- @xy: Kx2 np.array each row is a pair of inverse-transformed coordinates (x, y) )rr)rrTinvrrrrtforminv%s r c Csddi}|d}|jd}|dddfd}|dddfd}t||t|dft|dff}t|| t|dft|dff}t||f} |dddfd} |dddfd} t| | f} t| d|krt| | dd\} }}}t | } nt d| d}| d}| d}| d }t || dg||dg||dgg}t |}t gd |dddf<||fS) NKr)rrrr)rcondzcp2tform:twoUniquePointsReqrrr) rreshaperrrzerosvstackrankrsqueeze Exceptionarrayr)rroptionsr!Mxytmp1tmp2XuvUr_scsstxtyrTrrrfindNonreflectiveSimilarity<s. &( $r>cCsddi}t|||\}}|}d|dddf|dddf<t|||\}}tgdgdgdg}t||} t||} t| |} t| |} t| |} | | kr||fSt| }| |fSdS)Nr!r"rr)rrr)rrrr%)r>rr,rrrr)rrr-Ztrans1Z trans1_invZxyRZtrans2rZ trans2r_invZ TreflectYZtrans2xy1norm1xy2norm2Z trans2_invrrrfindSimilarity^s      rCTcCs*|rt||\}}nt||\}}||fS)ap Function: ---------- Find Similarity Transform Matrix 'trans': u = src_pts[:, 0] v = src_pts[:, 1] x = dst_pts[:, 0] y = dst_pts[:, 1] [x, y, 1] = [u, v, 1] * trans Parameters: ---------- @src_pts: Kx2 np.array source points, each row is a pair of coordinates (x, y) @dst_pts: Kx2 np.array destination points, each row is a pair of transformed coordinates (x, y) @reflective: True or False if True: use reflective similarity transform else: use non-reflective similarity transform Returns: ---------- @trans: 3x3 np.array transform matrix from uv to xy trans_inv: 3x3 np.array inverse of trans, transform matrix from xy to uv )rCr>)src_ptsdst_pts reflectiver trans_invrrrget_similarity_transforms rHcCs|ddddfj}|S)aP Function: ---------- Convert Transform Matrix 'trans' into 'cv2_trans' which could be directly used by cv2.warpAffine(): u = src_pts[:, 0] v = src_pts[:, 1] x = dst_pts[:, 0] y = dst_pts[:, 1] [x, y].T = cv_trans * [u, v, 1].T Parameters: ---------- @trans: 3x3 np.array transform matrix from uv to xy Returns: ---------- @cv2_trans: 2x3 np.array transform matrix from src_pts to dst_pts, could be directly used for cv2.warpAffine() Nrr")r=)r cv2_transrrrcvt_tform_mat_for_cv2srJcCst|||\}}t|}|S)a Function: ---------- Find Similarity Transform Matrix 'cv2_trans' which could be directly used by cv2.warpAffine(): u = src_pts[:, 0] v = src_pts[:, 1] x = dst_pts[:, 0] y = dst_pts[:, 1] [x, y].T = cv_trans * [u, v, 1].T Parameters: ---------- @src_pts: Kx2 np.array source points, each row is a pair of coordinates (x, y) @dst_pts: Kx2 np.array destination points, each row is a pair of transformed coordinates (x, y) reflective: True or False if True: use reflective similarity transform else: use non-reflective similarity transform Returns: ---------- @cv2_trans: 2x3 np.array transform matrix from src_pts to dst_pts, could be directly used for cv2.warpAffine() )rHrJ)rDrErFrrGrIrrr get_similarity_transform_for_cv2srK__main__)r)rr$)rr)rirPz --->uv:z --->xy:z --->trans matrix:z --->trans_inv matrix:z ---> apply transform to uvz xy_m = uv_augmented * transrz xy_m = tformfwd(trans, uv)z# ---> apply inverse transform to xyz uv_m = xy_augmented * trans_invz uv_m = tformfwd(trans_inv, xy)z uv_m = tforminv(trans, xy))N)N)T)T)%numpyr numpy.linalgrrrr)rr+rrr r>rCrHrJrKrr4r5r/r0r,r=rrprintrrGrrrZuv_augrZxy_mZxy_augZuv_mrrrrs\   " $ ( % '