Wpf5NdZddlZddlZddlmZmZmZmZmZddl Z ddl m Z ddl ZddlmZddlmZddlmZdZdZdZd Zd Zd Zd Zd ZejGddZdedededededeeefff dZeeefdej dej!dedefdZ"deeedfdej dej#deeeeefdeeeeeeffd eeeeeefeffd!e$f d"Z%d#d$d%efdej d&ej d'ej d(eeefd)eeefd*ed+efd,Z&d-Z'deed./eed./d0d0fdej#deeeeefded ed1ed2ef d3Z(dS)4z!MediaPipe solution drawing utils.N)ListMappingOptionalTupleUnion) detection_pb2) landmark_pb2)location_data_pb2?)r r rrr)rr)rr)rrrcLeZdZUeZeeeefed<dZeed<dZ eed<dS) DrawingSpeccolor thickness circle_radiusN) __name__ __module__ __qualname__ WHITE_COLORrrint__annotations__rrh/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/mediapipe/python/solutions/drawing_utils.pyrr'sO!,%sC} +++)S-rr normalized_x normalized_y image_width image_heightreturncdtdtfd}||r ||sdSttj||z|dz }ttj||z|dz }||fS)z4Converts normalized value pair to pixel coordinates.valuer$cn|dkstjd|o|dkptjd|S)Nr)mathisclose)r&s ris_valid_normalized_valuezC_normalized_to_pixel_coordinates..is_valid_normalized_value7sA AI /a// Neai7M6:l1e6L6LNrNr()floatboolminr)floor)r r!r"r#r+x_pxy_pxs r _normalized_to_pixel_coordinatesr21s NuNNNNN $ #L 1 1 # #L 1 1 4 TZ {2 3 3[1_ E E$ TZ |3 4 4lQ6F G G$ trrimage detectionkeypoint_drawing_specbbox_drawing_specc|jsdS|jdtkrtd|j\}}}|j}|jt jjkrtd|jD]E}t|j |j ||} tj || |j|j|jF|dsdS|j} t| j| j||} t| j| jz| j| jz||} tj|| | |j|jdS)aDraws the detction bounding box and keypoints on the image. Args: image: A three channel BGR image represented as numpy ndarray. detection: A detection proto message to be annotated on the image. 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 one of the followings: a) If the input image is not three channel BGR. b) If the location data is not relative data. Nr0Input image must contain three channel bgr data.z?LocationData must be relative for this drawing funtion to work.relative_bounding_box) location_datashape _BGR_CHANNELS ValueErrorformatr LocationDataRELATIVE_BOUNDING_BOXrelative_keypointsr2xycv2circlerrrHasFieldr:xminyminwidthheight rectangle) r4r5r6r7 image_rows image_cols_locationkeypoint keypoint_pxr:rect_start_pointrect_end_points rdraw_detectionrUDs(    F [^}$$ G H HH#k*j!  $( _)6LLL I K KK-MMh28:xz3=zKKKJuk#8#F$*,A,KMMMM   2 3 3  F"85 "7"> #8#??.-'!'):)DFFFFFrT landmark_list connectionslandmark_drawing_specconnection_drawing_specis_drawing_landmarkscZ|sdS|jdtkrtd|j\}}}i} t|jD]s\} } | dr| jtks%| dr| jtkrPt| j | j ||} | r| | | <t|rt|j} |D]}|d}|d}d|cxkr| krnnd|cxkr| ksntd|d |d || vrQ|| vrMt|tr||n|}t!j|| || ||j|j|r|r| D]\} } t|tr|| n|}t+|jdzt/|jd z}t!j|| |t2|jt!j|| |j|j|jdSdSdS) aDraws the landmarks and the connections on the image. Args: image: A three channel BGR image represented as numpy ndarray. landmark_list: A normalized landmark list proto message 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. Nrr9 visibilitypresencerr(BLandmark index is out of range. Invalid connection from landmark # to landmark #.g333333?)r<r=r> enumeratelandmarkrGr\_VISIBILITY_THRESHOLDr]_PRESENCE_THRESHOLDr2rCrDlen isinstancerrElinerritemsmaxrrrFr)r4rVrWrXrYrZrMrNrOidx_to_coordinatesidxrb landmark_px num_landmarks connection start_idxend_idx drawing_speccircle_border_radiuss rdraw_landmarksrsws H   F [^}$$ G H HH#k*j! !788 , ,mc8   < ( (  4 4 4   : & & 5  0 0 028:xz3=zKKK, +) .//M! ) ) Q-i1 g9,,,,},,,,,g1M1M1M1M 1M1M1M1MO+4OODKOOOPP P ( ( (W8J-J-J>H #W?.?.K.z::3J  *95#G,l.@' ) ) )  =3 =.4466 = =[3= 4*4*E*3///D!!;a!?!$\%?#%E!F!FHH j %9;'))) j \%?#\%;==== = = = = = =r)?rt)rug?rotation translation focal_lengthprincipal_point axis_lengthaxis_drawing_specc|jdtkrtd|j\}}} tjgdgdgdgdg} tj||| jzj|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|j tj |||t|j tj |||t|j dS)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. rr9r)r(rr)rr(r)rrr().r).r().rgh㈵>grtr(r rr N)r<r=r>npfloat32matmulTclipint32rE arrowedLine RED_COLORr GREEN_COLOR BLUE_COLOR)r4rvrwrxryrzr{rMrNrO axis_worldaxis_camrCrDzfxfypxpyx_ndcy_ndcx_imy_imoriginx_axisy_axisz_axiss r draw_axisrs. [^}$$ G H HH#k*j!z999iiiIIIFGG* Yxz|!; < < > L(v!v!v! &"b &"b '2#'QX&+S" 5 5% '2#'QX&+S" 5 5% 1u9#j0 1 1$ 1u9#j0 1 1$ GT!W & GT!W & GT!W & GT!W &/%4E4OPPP/%#-////%#-/////rc4td|DS)Nc3 K|] }|dz V dS)go@Nr).0vs r z#_normalize_color..s&''Aq4x''''''r)tupler3s r_normalize_colorrs ''''' ' ''r)rr elevationazimuthc R|sdStjdtjd}|||i}t |jD]\}} | dr| jtks%| dr| j tkrP| | j g| j g| j gt|jddd |j | j | j | j f||<|rt%|j} |D]} | d } | d } d | cxkr| krnnd | cxkr| ksnt'd | d| d| |vr| |vr|| || g}||d d |d d g|d d |d d g|d d|d dgt|jddd |j tjdS)aPlot the landmarks and the connections in matplotlib 3d. Args: landmark_list: A normalized landmark list proto message 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\r])xsyszsr linewidthrr(r^r_r`r)pltfigureaxes view_initrarbrGr\rcr]rd scatter3DrrCrDrrrrer>plot3Dshow)rVrWrXrYrraxplotted_landmarksrkrbrmrnrorp landmark_pairs rplot_landmarksrs0   F*X x4   ",,IG,,,, !788 D Dmc8   < ( (  4 4 4   : & & 5  0 0 0LL ZK= J< ZK=4:44R4@AA'1 333 (zk8: {Cc9 .//M!99 Q-i1 g9,,,,},,,,,g1M1M1M1M 1M1M1M1MO+4OODKOOOPP P ' ' 'G7H,H,H i (*;G*D  a #]1%5a%89a #]1%5a%89a #]1%5a%89"#:#@2#FGG-7  9 9 9 (*****r))__doc__ dataclassesr)typingrrrrrrEmatplotlib.pyplotpyplotrnumpyr}mediapipe.framework.formatsrr r rdrcr=r BLACK_COLORrrr dataclassrr,rr2ndarray DetectionrUNormalizedLandmarkListr-rsrrrrrrrs(' 88888888888888 555555444444999999       ',;>eCHo 56,*59)E)E)E%0[]] 0F0F :0F&0F'0F# 0F0F0F0Fl48  )$$$  !%P=P= :P=6P=$uS#X/0P=$ k73 #3445 P=#WU38_k9::P=P=P=P=P=l3=5?#&/:{}} 1/1/RZ1/ 1/:1/"%,/1/ %UE\2 1/ ! 1/ "- 1/1/1/1/h((( CG8C &!959595:E+(A;7;7;7$&"$> > ,"E> (eCHo)> ?> *5> -8 > " >  > > > > > > r