a ■dРу@sBddlZddlZddlmZmZdddДZd ddДZdd dДZdS)щN)┌cos┌sinщdcCsЬ|tjd}|tjd}|tjd}|dus<|dur^|jddЕ\}}|d}|d}|t|Гt|Г|} |t|Гt|Гt|Гt|Гt|Г|} |t|Гt|Г|}|t|Гt|Гt|Гt|Гt|Г|}|t|Г|} |t|Гt|Г|}tа|t|Гt|Гft| Гt| Гfddбtа|t|Гt|Гft|Гt|Гfddбtа|t|Гt|Гft| Гt|Гfddб|S)zdraw head pose axis.щ┤Nщйrrщ щйrrrйrrrй┌np┌pi┌shaperr┌cv2┌line┌int)┌img┌yaw┌pitch┌roll┌tdx┌tdy┌size┌height┌width┌x1┌y1┌x2┌y2┌x3┌y3йr"·l/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/facexlib/visualization/vis_headpose.py┌ draw_axiss"00***r$ч└b@c Cs0|tjd}|tjd}|tjd} |durV|durV|d|} |d|}n2|jddЕ\}} | dd|} |dd|}|t|Гt| Г| }|t|Гt| Гt| Гt|Гt|Г|}|t|Гt| Г| }|t|Гt| Гt|Гt|Гt| Г|}|t|Г| }|t|Гt|Г|}tа|t| Гt|Гft|Гt|Гfddбtа|t| Гt|Гft|Гt|Гfddбtа|t|Гt|Гft||| Гt|||Гfddбtа|t|Гt|Гft||| Гt|||Гfddбtа|t| Гt|Гft|Гt|Гfddбtа|t|Гt|Гft||| Гt|||Гfddбtа|t|Гt|Гft||| Гt|||Гfddбtа|t||| Гt|||Гft|||d| Гt|||d|Гfddбtа|t||| Гt|||Гft|||d| Гt|||d|Гfddбtа|t||| Гt|||Гft|||d| Гt|||d|Гfddбtа|t|Гt|Гft||| Гt|||Гfddбtа|t|Гt|Гft||| Гt|||Гfddб|S) zНdraw head pose cube. Where (tdx, tdy) is the translation of the face. For pose we have [pitch yaw roll tdx tdy tdz scale_factor] rNgр?rrr rr r)rrrrrrr┌p┌y┌rZface_xZface_yrrrrrrr r!r"r"r#┌draw_pose_cube#sD00**::*::$2 $2 $2 ::r)Fc Cs╘tа|б}|rtа|tjб}d|dабdЫd|dабdЫdЭd|dабdЫЭ}tj||d|jddfddd d dНt||d|d|ddd Нt ||d|d|dddddН|dur╨tа ||бdS)Nzy rz.2fz, p z, zr щщrr)ZfontFaceZ fontScale┌colorZ thicknessr)rщ2)rrr)r ┌copyrZcvtColorZ COLOR_RGB2BGR┌itemZputTextrr)r$Zimwrite)rrrrZ save_pathZto_bgrZshow_stringr"r"r#┌visualize_headposeQs :&"r0)NNr)NNr%)NF) r┌numpyr ┌mathrrr$r)r0r"r"r"r#┌s .