+jZ%SrSSKrSSKrSSKJrJrJr SSKJ r SSK J r SSK r SSKr SSKr SSKrSSKJrJr SSKJr SSKJr \R0"S 5r\R0"S 5rS rS r"S S5rS\R<S\R>4Sjr S\!S\"\#\#\#44Sjr$S\%S\%4Sjr&S\%S\%S\'S\%4Sjr(S\)\%S\)\%S\)\"\#\#44Sjr*S\)\)\%S\!SS4Sjr+S\,\"\#\#4S\)\)\#4Sjr-"S S!\R\5r/S"r0"S#S$\R\5r1/\QS%PS&P7r2\"\!S'4\3S('S)r4\"\"\!\54S'4\3S*'"S+S,\R\5r6S=S-\R>S.\)S/\"\#\#\#4S0\#S1\#S2\)\)\#S-S\R>4S3jjr7S4r8\"\!S'4\3S5'S6r9\"\"\!\54S'4\3S7'"S8S9\R\5r:"S:S;\5r;S\;4S<jru,ComfyUI nodes for the pure-PyTorch MediaPipe Face Landmarker port. Custom IO types: FACE_LANDMARKER — FaceLandmarkerModel wrapper (ModelPatcher inside) FACE_LANDMARKS — {"frames": List[List[face_dict]], "image_size": (H, W), "connection_sets": dict[str, frozenset[(int, int)]]} face_dict: bbox_xyxy, blendshapes, landmarks_xy, landmarks_3d, presence, score, transformation_matrix MediaPipeFaceLandmarker also emits the core BOUNDING_BOX type — pair with DrawBBoxes. N)Image ImageColor ImageDraw)tqdm)override)ComfyExtensionio)FaceLandmarker)$transformation_matrix_from_detectionFACE_DETECTION_MODELFACE_LANDMARKS)canonical_verticesprocrustes_indicesprocrustes_weights) face_ovalleft_eye right_eye left_eyebrow right_eyebrowlipsc<\rSrSrSrS\4SjrS\S\S\ 4Sjr S r g ) FaceLandmarkerModel%a Loaded FaceLandmarker variants + ModelPatcher per variant. Safetensors layout: `detector_short.*` / `detector_full.*` plus shared `mesh.*`, `blendshapes.*`, `canonical_*`, and `topology.*`. PReLU forces plain-nn / fp32 (manual_cast strands buffers across devices). state_dictc r^ [RR5Ul[RR 5n[ R Ul0m UVs/sHo3RS5(dMUPM snHHn[[[URU5R555T U[S5S'MJ [5R"U 4Sj[ 56T S'T ST S-T S-T S'T Ul[$Vs0sH"o3URU5R'5_M$ snUlUR+5VVs0sHup4URS5(dMX4_M! nnn0Ul0UlSGHnS US 3n[1U5nUR3UR+5VVs0sH/up4URU5(dMS U[U5S3U_M1 snn5 [5X RSUS 9R75n U R9US S9 XR,U'[R:R=XRU[RR?U 5S9UR.U'GM gs snfs snfs snnfs snnf)Nz topology.c3.># UH nTUv M g7fN).0pbases /FaceLandmarkerModel.__init__..6s.O1tAwcontoursirisesnoseall)zmesh.z blendshapes.)shortfull detector_.z detector.)devicedtype operationsdetector_variantF)strict) load_deviceoffload_devicesize) comfymodel_managementtext_encoder_devicer3text_encoder_offload_devicetorchfloat32r/ startswith frozensetmaptuplepoptolistlenunion_CONTOUR_PARTSconnection_sets_CANONICAL_KEYSnumpycanonical_dataitemsmodelspatchersdictupdater evalload_state_dict model_patcherCoreModelPatcher module_size) selfrr4kvsharedvariantprefixsubflr!s @r"__init__FaceLandmarkerModel.__init__-sR 11EEG//KKM]] &('EZ<< +D!ZEA)23ujnnQ>O>V>V>X3Y)ZD3{#$% &F$;,,.O.OPZ:&h7$v,FU 59\k5l\kWX9J9P9P9R6R\k5l#-#3#3#5a#541F_9`$!$#5a13 EG (G  +Fv,C JJZEUEUEWpEWTQ[\[g[ghn[o8)Ac&klO#45q8EWp q~ZZTXkrsxxzB   s5  1#%KK %*%8%8%I%I 0 0++77;&J&DMM' ")F 6maqs*J#;J#)J(J-?J- J3 +J3 num_faces score_threshrWc[RRURU5 URUR XUS9$)N)r]r^)r6r7load_model_gpurKrJ detect_batch)rSimagesr]r^rWs r"ra FaceLandmarkerModel.detect_batchMs? --dmmG.DE{{7#00[g0hh)rHrEr/r3rJrKN) __name__ __module__ __qualname____firstlineno____doc__rLr[intfloatstrra__static_attributes__rrdr"rr%s34@iciiQTirdrimagereturncUSSS24RS5RS5RSS5R[R 5R 5R5$)N.o@?r)muladd_clamp_tor:uint8cpurG)rns r"_image_to_uint8r{RsX bqb>  e $ ) )# . 5 5a = @ @ M Q Q S Y Y [[rdcolorcV[R"U5SS$![a gf=f)Nrq)rrtr)rgetrgb ValueError)r|s r" _parse_colorrVs1  '++ s  ((facecUSR5[US5USR5USR5USUSS.$)zKShallow copy of a face_dict with array-fields cloned so callers can mutate. bbox_xyxy blendshapes landmarks_xy landmarks_3dpresencescorerrrrrr)copyrL)rs r" _copy_facer]s][)..0T-01^,113^,113Z(W  rdabtc SU- US-X!S--USVs0sHo3SU- USU-X!SU--_M snSU- US-X!S--SU- US-X!S--SU- US-X!S--SU- US-X!S--S.$s snf) Nrrrrrrrr)rrrrTs r" _lerp_facerisQ!K.0aK.6HH]^_l]mn]mXYQUa &6q&99A-@PQR@S# UH)n[[T55H nTX4X4v M M+ g7fr)rangerB)riaibrdistss r"r#_match_faces..s3`-BRWX[\]X^R_Brv/R_/-s14) nparraylinalgnormsetsortedrrBappendadd) rrf centers_a centers_bpairsused_aused_b candidates_rrrs ` @r" _match_facesrts A STVSTa!K."3a nQ6G"GH!K."3a nQ6G"GHJSTVWISTVSTa!K."3a nQ6G"GH!K."3a nQ6G"GHJSTVWI IINN9QW- $?bN IE#%EuFuF`%A-``J r <2<  bX 2 2  LVVs 2E$42E)framesmodec ^US:Xag[U5VVs/sHup#U(dMUPM nnnU(dgUS:XaK/n[U5H9umnU(aUnMU(dMUVs/sHn[U5PM snUT'M; g[[U55HmUT(aM[ U4SjU5SS9n[ U4SjU5SS9nUc#XVs/sHn[U5PM snUT'M]Uc#XVs/sHn[U5PM snUT'MTU- X- - n [ XX5n U V V s/sHup[XU XU U 5PM sn n UT'M gs snnfs snfs snfs snfs sn n f)zIn-place fill empty frame slots from neighbouring detections. Multi-face aware: pairs faces across bracketing frames by greedy bbox-centre NN. When counts differ, unmatched faces are dropped from the synthesised frame.emptyNpreviousc36># UHoT:dM Uv M g7frrrrUis r"r#'_fill_missing_frames..0Aa%aa  defaultc36># UHoT:dM Uv M g7frrrs r"r#rrr) enumeraterrrBmaxminrr) rrrfrvalidlastrprev_inext_irrrrs ` r"_fill_missing_framesrsw w%f- 4-51Q-E 4  zv&EAr489DqZ]D9q '  3v;  !9 00$?00$? >06?1A?F1I ^06?1A?F1IV0A @EY^_Y^QUQRFN1$5v~a7H!LY^_F1I  5:@?`s" E/E/*E5E:E?#Fedgesc^ 0nUHWup#URU[55RU5 URU[55RU5 MY [5n/nUH~nXd;aM U/nURU5 SUsm n[U 4SjX5S5n U bX:XaO(UR U 5 URU 5 Xsm nMIUR U5 M U$)zWalk an unordered edge set into one or more closed-loop vertex rings (handles multi-loop sets like FACEMESH_LIPS: outer + inner).rc36># UHoT:wdM Uv M g7frr)rrUprevs r"r#!_ordered_rings..s98aDy8rN) setdefaultrrnextr) radjrrvisitedringsstartringcurnxtrs @r"_ordered_ringsrs "C q#% $$Q' q#% $$Q'GE   w E c93894@C{cl KK  KK ID#   T LrdcT\rSrSrSr\S5r\S\R4Sj5r Sr g)LoadMediaPipeFaceLandmarkerzLoad MediaPipe Face Landmarker v2 weights. Contains both detector variants (short / full), shared mesh, blendshapes, and canonical geometry.c [R"S/SQSS[RRS[R "S5SS9/[ R5/S 9$) Nrrfacial mediapipe face landmark face mesh blazefaceface detectionz%Load Face Detection Model (MediaPipe)z model/loaders model_name detectionz,Face detection model from models/detection/.)optionstooltip)node_idsearch_aliases display_namecategoryinputsoutputs)r SchemaComboInput folder_pathsget_filename_listFaceDetectionTypeOutputclss r" define_schema)LoadMediaPipeFaceLandmarker.define_schemas^yy1w@$|\5S5ST_5`'UW'--/0  rdroc[RR[R"SU5SS9n[ U5n[ R"U5$)NrT) safe_load)r6utilsload_torch_filerget_full_path_or_raiserr NodeOutput)rrsdwrappers r"execute#LoadMediaPipeFaceLandmarker.executesC [[ ( ()L)L[Zd)equ ( v%b)}}W%%rdrN rerfrgrhri classmethodrr rrrmrrdr"rrs;I    &BMM&&rdr)rr interpolatecT\rSrSrSr\S5r\S\R4Sj5r Sr g)MediaPipeFaceLandmarkeruBlazeFace → FaceMesh v2 → ARKit-52 blendshapes, batched across the input. Also emits a BOUNDING_BOX list (landmark-extent bbox per face) — pair with DrawBBoxes for detector-only viz or MediaPipeFaceMeshVisualize for the mesh overlay.c [R"S/SQSSS[RS5[RRS5[R RS/S QS S S 9[R RS SSSSSS9[RRSSSSSSSS9[R RS[[5SSSS9/[RSS 9[RRS!5/S"9$)#Nrrz!Detect Face Landmarks (MediaPipe)image/detectionz/Detects facial landmarks using MediaPipe model.face_detection_modelrnr1)r*r+bothr*uFace detector range. 'short' is tuned for close-up faces (within ~2 m of the camera); 'full' covers farther / smaller faces (up to ~5 m) but is slower. 'both' runs both detectors and keeps whichever found more faces per frame (~2× detection cost).)rrrr]rrzDMaximum faces to return per frame. 0 = no cap (return all detected).rrrsteprmin_confidencersgg?g{Gz?Tz?BlazeFace score threshold. Lower to catch small/occluded faces.)rrrr advancedrmissing_frame_fallbackra$Per-frame behaviour when detection fails in a batch. 'empty' leaves the frame faceless. 'previous' copies the most recent successful detection. 'interpolate' lerps landmarks/bbox/blendshapes between bracketing successful frames. Multi-face: pairs faces across frames by greedy bbox-centre NN.)rrr rface_landmarks)rbboxesrrrr descriptionrr) r rrrrrIntFloatlist_FALLBACK_MODESFaceLandmarksTyper BoundingBoxrs r"r%MediaPipeFaceLandmarker.define_schemasyy-w<&I!''(>?w'1;T^e(kl  [!%km/#3UYdh'hj7oAV`grv(|}&"((6F(G%%h/1  rdroc (^^^^^^^^TRn[U5mTRSSumpSmUS:Hn U (aST-OTm[RR T5mS[ S[[[4UUUUUUUU4Sjjn U (aNU "S5n U "S 5n [T5Vs/sH'n[X5[X5:aXOXPM) nnOU "U5n[X5 /nUHjn/nUHNn[UXU5US 'S US 5unnnnURUUUU- UU- S [US5S.5 MP URU5 Ml [R "XU 4TR"S.U5$s snf)NrqrrrrWroc >/n[TSUS3S9n[STT5Hn[UT-T5nURTR [X45Vs/sHnT UPM sn[ T 5[ T 5US95 T R[T RXC- -T 55 URXC- 5 M SSS5 U$s snf!,(df  U$=f)NzMediaPipe Face Landmarker ())totaldescr)r]r^rW) rrrextendrarjrkupdate_absolutecurrentrM)rWrestqrendbiBchunkrimg_npr r]pbar total_works r"_run-MediaPipeFaceLandmarker.execute.._runs$&CA&A'!$LMQSq!U+Aa%i+CJJ3@@.3Am%: ' A  ((T\\SW-Ez)RSIIcg&,NJ= NMJsACC $AC C C!r*r+transformation_matrixc38# UHn[U5v M g7fr)rk)rrUs r"r#2MediaPipeFaceLandmarker.execute..3s!CNq%((Nsrrr)xywidthheightlabelr)r image_sizerE)rHr{shaper6r ProgressBarrlrrLrrBrr rrkr rrE)rrrnr1r]r r  canonicalHWis_bothr) short_resfull_resr#rr per_frameper_bbrx1y1x2y2r$r%r&r'r(s ` `` @@@@@r"rMediaPipeFaceLandmarker.execute s)77  ',,r"1"f,%QU1 {{&&z2 # $tDz"2   W IF|H (("B"%Y]!3s8<7H!H hlZ" (F *+FV<IF-QRSUV[d-e)*!CAkN!CBB BR"r'RRTW_epuvwxwAqBCD MM& !  }}!f1E1U1UWX^` `(s3.FrNrrrdr"rrs>   >-`+-==-`-`rdrr'r(._ALL_CONNECTION_PARTS) rT)rT)rT)rT)rT)rT)r'T)r(T) tesselationF_CUSTOM_FEATUREScX\rSrSrSr\S5r\SS\R4Sjj5r Sr g) MediaPipeFaceMeshVisualizeiIzqDraw a FACEMESH_* subset over an image. Topology travels with the FACE_LANDMARKS payload (set at detection time).c [R"S/SQSSS[RS5[RRSSS S 9[R RS S [R R S /5[R R S/5[R R S[VVs/sH'up[RRXSUS3S9PM) snn5/S9[RRSSS9[RRSSSSSSS9[RRSSSSSS S9/[RR5/S!9$s snnf)"NrH)rrrrrrr visualizez$Visualize Face Landmarks (MediaPipe)rz-Draws face landmarks mesh on the input image.r rnTz.If not connected, a black canvas will be used.)optionalr connectionsz'all' = oval+eyes+brows+lips+irises+nose. 'fill' = solid face_oval polygon (silhouette mask). 'custom' = toggle each feature individually (including 'tesselation', the full 2547-edge wireframe).r)fillcustomz Draw the 'z' connection set.rrrrr|z#00ff00r thicknessrrz7Edge line thickness in pixels. 0 disables edge drawing.r point_sizerrz8Landmark dot radius in pixels. 0 disables point drawing.r) r rrrr DynamicComboOptionrFBooleanColorrr)rfeatrs r"r(MediaPipeFaceMeshVisualize.define_schemaMslyy0E?&G!''(89w?op%%!a..ub9..vr:..x2B:2B JJ,,T7A$GX5Y-[1A:& w : [!%^` \1!!%_a',XX__&'9  :s.E?*Nroc j^USmUSnSnUS:Xa[TS5n[5n OUS:XaU[V V s/sH upURU S5(dMU PM" n n n [5R"U4SjU 56n O%[5R"U4Sj[ 56n [ U5[U5[U5pn US nUc9US unn[R"[U5UUS 4[RS 9nO [U5nURS n[U5n[RR!U5n[R""U5n[%U5H9nUU:aUUO/n['UUUXXU5UU'UR)US-5 M; [*R,"[.R0"U5R3[R4R75[R4R95S9R;S55$s sn n f)NrErLrMrrNFc3.># UH nTUv M g7frrrr setss r"r#5MediaPipeFaceMeshVisualize.execute..xs'?AQr%c3.># UH nTUv M g7frrr\s r"r#r^zs'O9NAQ9Nr%rr3rqr/rrr.r/rr)rr=rFgetrCrCrrjrzerosrBryr{r4r6rr5 empty_liker _draw_meshrr rr: from_numpyrxr7intermediate_deviceintermediate_dtypediv_)rr rLr|rQrSrnsel fill_ringsrrXrpartsrgbthickpsizerr7r8r&r$n_framesr'outr#facesr]s @r"r"MediaPipeFaceMeshVisualize.executens/0-(-1 &='[(9:JKE H_)9Z)9gd[__TSX=YT)9EZK%%'?'?@EK%%'O9N'OPE(/YZE) =!,/DAqXXs6{Aq!4BHHEF$U+F LLOv;{{&&q)mmF#(B"$x-F2JRE UEjYCG  a (}}U--c255))==?((;;=6  $u+ '[s H/H/rrrrrdr"rHrHIs>7  @_a_l_lrdrH image_rgbrrrmrQrSrkc US:=(a UnU(aUcU(dUS::aUR5$[R"U5n[R"U5n US-n UbjUHNn U Sn UH@n U R U Vs/sH!n[ XS45[ XS454PM# snUS9 MB MP [R"U5$UGHn U Sn U RSnU(ajUHdunnUU:dMUU:dMU R[ U US45[ U US454[ U US45[ U US454/X4S9 Mf US:Xa.U RU R5R5US9 MUS:dMU HJunnU R[ U5U - [ U5U - [ U5U -[ U5U -4US9 ML GM [R"U5$s snf)NrrsrrrM)rMr0)rr fromarrayrDrawpolygonrkrasarrayr4linepointflattenrAellipse)rtrrrrmrQrSrk draw_edgespildrawrrlmksrrnrrr.r/s r"reresQ(5J Z' zQ~~ //) $C >># DSAA^$D" dSduTQ$Z0%T 2CDdSZ] ^#zz#   JJqM 1q5QUIId1a4j 15ad3DE %d1a4j 15ad3DEGMPc ? JJt||~,,.SJ 9 !^1 eAhlE!HqL%(Q,aST U\_ ` ::c?Ts(H )rrrrr' _MASK_REGIONS)rD)rF)rF)rF)r'F_MASK_CUSTOM_FEATUREScT\rSrSrSr\S5r\S\R4Sj5r Sr g)MediaPipeFaceMaskizBinary mask from face landmarks, filled polygon per face. One mask per frame in the batch; faces in the same frame composite (union).c[R"S/SQSSS[RS5[RRSS[RR S /5[RR S [ VVs/sH'up[RRXS US 3S 9PM) snn5/S9/[RR5/S9$s snnf)Nr)rrrz face maskrrrJzDraw Face Mask (MediaPipe)rz!Draws a mask from face landmarks.r regionsz'all' = union of face_oval+lips+eyes+irises (which collapses to face_oval since it encloses the rest). 'custom' = toggle each region individually for combos like lips+eyes.r)rNz Include the 'z' region in the mask.rOrPr) r rrrrTrUrrVMaskr)rregrs r"rMediaPipeFaceMask.define_schemasyy's5&;!''(89%%K..ub9..x1F:1F JJ,,S7DSEI^5_-a0E:& WW^^%&+  :s .C#&rocvUSnUSnUS:Xa4[VVs/sH upVURUS5(dMUPM" nnnO[[5nUVVs/sHn[ X55HoPM M n nnUSn USup[ R "[U 5X4[ RS9n [RR[U 55n[U 5HunnU(a[R"SX4S 5n[R "U5nUHPnUS nU HBnUR#UVs/sH#n[%UUS 45[%UUS 454PM% snS S 9 MD MR [ R&"U5X'UR)US -5 M [*R,"[.R0"U 5R3[R4R75[R4R95S9R;S55$s snnfs snnfs snf)NrErrNFrr3r`Lrrrrtrvrarr)rrbrrrrrcrBryr6rr5rrnewrrxryrkrzrr rr:rfrxr7rgrhri)rr rr]rjrrpickedrrrr7r8masksr'r#r<rrrrrrs r"rMediaPipeFaceMask.executes/0i  (?(=Y(=fcSRWAXc(=FYF-(F$HfsnTY.G.GfH)l+#f+q,BHH={{&&s6{3&v.MB iiaVQ/ ~~c*"A^,D % VZ%[VZQRuT!Q$Z'8%QT :K&LVZ%[be f!&#JJsO  a (/}}U--e477))==?((;;=8  $u+ %ZI&\sH*H* H0*H6rNrrrdr"rrs;F  2rdrcL\rSrSr\S\\\R4Sj5r Sr g)MediaPipeFaceExtensioniroc6# [[[[/$7fr)rrrHr)rSs r" get_node_list$MediaPipeFaceExtension.get_node_lists+-DF`bsttsrN) rerfrgrhrrtyper ComfyNoderrmrrdr"rrs, uT$r||*<%=uurdrc# [5$7fr)rrrdr"comfy_entrypointrs ! ##s r)=rirGrr:PILrrr tqdm.autortyping_extensionsrcomfy.model_managementr6comfy.model_patcher comfy.utilsrcomfy_api.latestrr &comfy_extras.mediapipe.face_landmarkerr $comfy_extras.mediapipe.face_geometryr CustomrrrFrDrTensorndarrayr{rlr?rjrrLrrkrrrrr=rrrrrrC__annotations__rFboolrHrerrrrrrrdr"rs  ,,&/AUII45II./T`*i*iZ\5<<\BJJ\c3m 4 T d $4EdDJ4:$uS#X2G.`d4j!1```B)E#s(O4d3i4&",,&47T`bllT`p*M>)L8)LV)LuS#XL 2%c4i(#-. DDRFJ"**T#sC-(58,0cOd,BNPjj@"Z uS#XY7uU39-s238 8vu^u $ 6$rd