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__k/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/facexlib/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)Zrcondzcp2tform:twoUniquePointsReqrrr) rZreshaperrrzerosZvstackrankrZsqueeze Exceptionarrayr)rroptionsr!MxyZtmp1Ztmp2XuvUr_ZscssZtxtyrTrrrfindNonreflectiveSimilarity<s. &( $r6cCsddi}t|||\}}|}d|dddf|dddf<t|||\}}tgdgdgdg}t||} t||} t| |} t| |} t| |} | | kr||fSt| }| |fSdS)Nr!r"rr)rrr)rrrr$)r6rr(rrrr)rrr)Ztrans1Z trans1_invZxyRZtrans2rZ trans2r_invZ TreflectYZtrans2Zxy1Znorm1Zxy2Znorm2Z trans2_invrrrfindSimilarity^s      r7TcCs*|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 )r7r6)src_ptsdst_pts reflectiver trans_invrrrget_similarity_transforms r<cCs|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")r5)r cv2_transrrrcvt_tform_mat_for_cv2sr>cCst|||\}}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() )r<r>)r8r9r:rr;r=rrr get_similarity_transform_for_cv2sr?__main__)r)rr#)rr)rirDz --->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)%numpyrZ numpy.linalgrrrr&rr'rrr r6r7r<r>r?rr.r/r+r,r(r5rrprintrr;rrrZuv_augrZxy_mZxy_augZuv_mrrrrs\   " $ ( % '