dj36dZddlZddlZddlmZmZmZddlZddlm Z ddl Z ddl mZddl mZddlmZdZdZdZej.j0Zd Zd Zd Zd Zd Zej>GddZ de!de!de"de"dede#e"e"fff dZ$e ee fde jJdejLde de fdZ'de ee dfde jJde(ejRdee(edeee ee"e ffd ee ee#e"e"fe ffd!e*f d"Z+d#d$d%e fde jJd&e jJd'e jJd(e#e!e!fd)e#e!e!fd*e!d+e fd,Z,d-Z-de ed./e ed./d0d0fde(ejRdee(e#e"e"fde d e d1e"d2e"f d3Z.y)4z!MediaPipe solution drawing utils.N)MappingOptionalUnion) detections)landmark)face_landmarker?)r r rrr)rr)rr)r rrcDeZdZUeZeeeefed<dZeed<dZ eed<y) DrawingSpeccolor thickness circle_radiusN) __name__ __module__ __qualname__ WHITE_COLORrtupleint__annotations__rrn/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/mediapipe/tasks/python/vision/drawing_utils.pyrr(s-!,%sC} +)S-rr normalized_x normalized_y image_width image_heightreturncdtdtfd}||r||syttj||z|dz }ttj||z|dz }||fS)z4Converts normalized value pair to pixel coordinates.valuer#cz|dkDxstjd|xr|dkxstjd|S)Nr)mathisclose)r%s ris_valid_normalized_valuezC_normalized_to_pixel_coordinates..is_valid_normalized_value;s? AI /a/ Neai7M6:ll1e6LNrNr')floatboolminr(floor)rr r!r"r*x_pxy_pxs r _normalized_to_pixel_coordinatesr12stNuNN $L 1 #L 1  TZZ {2 3[1_ E$ TZZ |3 4lQ6F G$ trrimage detectionkeypoint_drawing_specbbox_drawing_specc|jdtk7r td|j\}}}|jrj|jD][}t |j |j ||}tj|||j|j|j]|jj|jjf} |jj|jjz|jj|jj zf} tj"|| | |j|jy)a'Draws the detction bounding box and keypoints on the image. Args: image: A three channel BGR image represented as numpy ndarray. detection: A detection object with a bounding box. keypoint_drawing_spec: A DrawingSpec object that specifies the keypoints' drawing settings such as color, line thickness, and circle radius. bbox_drawing_spec: A DrawingSpec object that specifies the bounding box's drawing settings such as color and line thickness. Raises: ValueError: If the input image is not three channel BGR. r0Input image must contain three channel bgr data.N)shape _BGR_CHANNELS ValueError keypointsr1xycv2circlerrr bounding_boxorigin_xorigin_ywidthheight rectangle) r3r4r5r6 image_rows image_cols_keypoint keypoint_pxrect_start_pointrect_end_points rdraw_detectionrNHs3& [[^}$ G HH#kk*j!'' 4 **hjj*jk jj    - -  % %  ) )  %%%% %% (>(>(D(DD%% (>(>(E(EE.--'!''):)D)DFrT landmark_list connectionslandmark_drawing_specconnection_drawing_specis_drawing_landmarkscJ|sy|jdtk7r td|j\}}}i} t|D]n\} } | j| jt ks| j | j tkrEt| j| j||} | sj| | | <p|rt|} |D]}|j}|j}d|cxkr| krnn d|cxkr| ksntd|d|d|| vsM|| vsRt|tr|||fn|}t!j"|| || ||j$|j&|r|r| j)D]\} } t|tr|| n|}t+|j,dzt/|j,d z}t!j0|| |t2|j&t!j0|| |j,|j$|j&yyy) aDraws the landmarks and the connections on the image. Args: image: A three channel BGR image represented as numpy ndarray. landmark_list: A list of normalized landmarks to be annotated on the image. connections: A list of landmark index tuples that specifies how landmarks to be connected in the drawing. landmark_drawing_spec: Either a DrawingSpec object or a mapping from hand landmarks to the DrawingSpecs that specifies the landmarks' drawing settings such as color, line thickness, and circle radius. If this argument is explicitly set to None, no landmarks will be drawn. connection_drawing_spec: Either a DrawingSpec object or a mapping from hand connections to the DrawingSpecs that specifies the connections' drawing settings such as color and line thickness. If this argument is explicitly set to None, no landmark connections will be drawn. is_drawing_landmarks: Whether to draw landmarks. If set false, skip drawing landmarks, only contours will be drawed. Raises: ValueError: If one of the followings: a) If the input image is not three channel BGR. b) If any connetions contain invalid landmark index. Nrr8rBLandmark index is out of range. Invalid connection from landmark # to landmark #.r'g333333?)r9r:r; enumerate visibility_VISIBILITY_THRESHOLDpresence_PRESENCE_THRESHOLDr1r=r>lenstartend isinstancerr?linerritemsmaxrrr@r)r3rOrPrQrRrSrGrHrIidx_to_coordinatesidxr landmark_px num_landmarks connection start_idxend_idx drawing_speccircle_border_radiuss rdraw_landmarksrmzsQF   [[^}$ G HH#kk*j! / ,mc8'   "7 7%    3 328::xzz3=zKK + , &M!) ""ig9,},g1M 1M++4+^G9AOP P ( (W8J-J17; $Y$8 9(  *95#G,l.@.@'' ))"3.446 =[3= 4**3//D!!;!;a!?!$\%?%?#%E!FH jj %9;'') jj \%?%?##\%;%;= =4r)?rn)rog?rotation translation focal_lengthprincipal_point axis_lengthaxis_drawing_specc||jdtk7r td|j\}}} tjgdgdgdgdg} tj ||| j zj |z} | d} | d} | d }|\}}|\}}tj| | z|d zz |zd d }tj| | z|d zz |zd d }tjd |zdz|z}tjd |z dz|z}|d|df}|d |d f}|d|df}|d|df}tj|||t|jtj|||t|jtj|||t|jy)aDraws the 3D axis on the image. Args: image: A three channel BGR image represented as numpy ndarray. rotation: Rotation matrix from object to camera coordinate frame. translation: Translation vector from object to camera coordinate frame. focal_length: camera focal length along x and y directions. principal_point: camera principal point in x and y. axis_length: length of the axis in the drawing. axis_drawing_spec: A DrawingSpec object that specifies the xyz axis drawing settings such as line thickness. Raises: ValueError: If one of the followings: a) If the input image is not three channel BGR. rr8r )r'rr)rr'r)rrr').r).r').rgh㈵>grnr'r rr N)r9r:r;npfloat32matmulTclipint32r? arrowedLine RED_COLORr GREEN_COLOR BLUE_COLOR)r3rprqrrrsrtrurGrHrI axis_worldaxis_camr=r>zfxfypxpyx_ndcy_ndcx_imy_imoriginx_axisy_axisz_axiss r draw_axisrs2 [[^}$ G HH#kk*j!zz9iIFG* YYxz||!; < > > L(v!v!v! &"b &"b ''2#'QX&+S" 5% ''2#'QX&+S" 5% 1u9#j0 1$ 1u9#j0 1$ GT!W & GT!W & GT!W & GT!W &//%4E4O4OP//%#--///%#--/rc&td|DS)Nc3&K|] }|dz  yw)go@Nr).0vs r z#_normalize_color.. s'Aq4x's)rr2s r_normalize_colorrs '' ''r)rr elevationazimuthc |sytjdtjd}|j||i}t |D]\}} | j | j t ks| j| jtkrE|j| j xsdg| jxsdg| j xsdgt|jddd|j | j | j| j f||<|rt!|} |D]} | d} | d } d| cxkr| krnn d| cxkr| ksnt#d | d | d | |vs?| |vsD|| || g}|j%|dd|d dg|dd |d d g|dd|d dgt|jddd|j tj&y)aPlot the landmarks and the connections in matplotlib 3d. Args: landmark_list: A list of normalized landmarks to be plotted. connections: A list of landmark index tuples that specifies how landmarks to be connected. landmark_drawing_spec: A DrawingSpec object that specifies the landmarks' drawing settings such as color and line thickness. connection_drawing_spec: A DrawingSpec object that specifies the connections' drawing settings such as color and line thickness. elevation: The elevation from which to view the plot. azimuth: the azimuth angle to rotate the plot. Raises: ValueError: If any connection contains an invalid landmark index. N)rr)figsize3d) projection)elevazimr)xsyszsr linewidthr'rUrVrWr)pltfigureaxes view_initrXrYrZr[r\ scatter3Drr=r>rrrr]r;plot3Dshow)rOrPrQrRrraxplotted_landmarksrerrgrhrirj landmark_pairs rplot_landmarksr s>8  **X xx4 ",,IG,, /Dmc8'   "7 7%    3 3LL ZZK 1  JJO!  ZZK 1 4::4R4@A'11  (zzk8:: {Cc!D" &M!9 Q-i1 g9,},g1M 1M++4+^G9AOP P ' 'G7H,H i (*;G*D  a #]1%5a%89a #]1%5a%89a #]1%5a%89"#:#@#@2#FG-77  99 ((*r)/__doc__ dataclassesr(typingrrrr?matplotlib.pyplotpyplotrnumpyrw,mediapipe.tasks.python.components.containersrdetections_modulerlandmark_modulemediapipe.tasks.python.visionrr\rZr:FaceLandmarksConnections Connection _CONNECTIONr BLACK_COLORr~rr dataclassrr+rrr1ndarray DetectionrNlistNormalizedLandmarkr,rmrrrrrrrs( ++ XT9 66AA          4sCx ! 2*59)E%0] /F ::/F **/F'/F# /Fj04 )$  !%U= ::U=::;U=${+,U=$ k73 #3445 U=#WU38_k9::U=U=x)3+5%0]3/ ::3/jj3/3/u % 3/ 5%<( 3/  3/#3/l( 48)41*,7Q,F ::;F $uS#X/0F 'F ) F F F r