+jLFSrSSKrSSKrSSKrSSKJrJrJr SSK J r SSK J r SSKJr SSKJr SSKJrJrJrJrJr SSKrSS KJr \R6"S 5r\R6"S 5rS \R<S \R<4SjrS\R<S \R<4Sjr "SS\RB5r""SS\RB5r#"SS\RB5r$"SS\RB5r%"SS\RB5r&"SS\5r'S \'4Sjr(g)zNComfyUI nodes for the native MoGe (Monocular Geometry Estimation) integration.N)ComfyExtensionTypesio)override)turbo) MoGeModel)triangulate_grid_mesh)get_panorama_camerassplit_panorama_imagemerge_panorama_depthspherical_uv_to_directions_uv_grid)tqdm MOGE_MODEL MOGE_GEOMETRYpointsreturncb[R"U5RSS9n[R"UR S5U[R "U55nUSSS2SS2SS24USSS2SS2SS24- nUSSS2SS2SS24USSS2SS2SS24- n[R RRURSSS S5S 5RSSSS 5n[R RRURSSS S5S 5RSSSS 5n[R"XCSS9n[R RRUSS9n[R"UR S5U[R "U55$) zICamera-space surface normals from a (B, H, W, 3) point map (v1 fallback).dim.Nr)rrrr)rrrr) torchisfiniteallwhere unsqueeze zeros_likenn functionalpadpermutecross normalize)rfiniteptsdxdyns 7/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_moge.py_normals_from_pointsr.sp ^^F # ' 'B ' /F ++f&&r*FE4D4DV4L MC S!QR] c#q#2#q.1 1B S!"a] c#ssAq.1 1B    Aq!Q!7 F N NqRSUVXY ZB    Aq!Q!7 F N NqRSUVXY ZB B#A %%aR%0A ;;v''+Q0@0@0C DDdepthcp[R"U5n[URS5GHnXn[R"U5US:-nUR 5(dM;[R "USURS5- [R"U55nXTn[R"US5n[R"US5nX- RS5n XW- U - RSS5n [R "XJ[R"U 55X'GM U$)zIPer-batch normalize 1/depth to [0, 1] using 0.1/99.9 percentile clipping.r?gư>gMbP?g+?) rr!rangeshaperanyr clamp_minquantileclamp) r0outidvaliddisp disp_validlohiscalenorms r-_normalize_disparityrD-s   5 !C 5;;q> " Hq!QU+yy{{ {{5# D(9"95;K;KA;NO[ ^^J . ^^J .##D)e#**34U%*:*:4*@A# Jr/cP\rSrSr\S5r\S\R4Sj5rSr g) LoadMoGeModel?c [R"SSS[RRS[R "S5S9/[ R5/S9$)NrFzLoad MoGe Modelz model/loaders model_namegeometry_estimation)options)node_id display_namecategoryinputsoutputs)rSchemaComboInput folder_pathsget_filename_list MoGeModelTypeOutputclss r- define_schemaLoadMoGeModel.define_schema@sRyy#*$|\5S5STi5jk#))+,  r/rc[R"SU5n[RR USS9n[ R "[U55$)NrJT) safe_load)rTget_full_path_or_raisecomfyutilsload_torch_filer NodeOutputr)rYrIpathsds r-executeLoadMoGeModel.executeLsA223H*U [[ ( ( ( >}}Yr]++r/N __name__ __module__ __qualname____firstlineno__ classmethodrZrrbre__static_attributes__rgr/r-rFrF?s3    ,BMM,,r/rFcT\rSrSrSr\S5r\S\R4Sj5r Sr g)MoGePanoramaInferenceSzEquirectangular panorama inference: split into 12 perspective views, run MoGe at fov_x=90 on each, merge via multi-scale Poisson + gradient solve. v2's predicted normals and metric scale are ignored (per-view scales would not align across seams). c[R"S/SQSSS[RS5[RRSSS 9[R RS S S S S S9[R RSSSSSS9[R RSSSSSS9[R RSSSSSS9/[ RSS9/S9$) Nrp)mogepanoramar0geometrydepth estimationgeometry estimationzRun MoGe Panorama Inferenceimage/geometry estimationzRun MoGe on an equirectangular panorama by splitting it into 12 perspective views, running inference on each, and merging the results into a single depth map. moge_modelimagez&Equirectangular panorama (any aspect).)tooltipresolution_level rz1Per-view detail (0 = fastest, 9 = most detailed).defaultminmaxr{split_resolutioniiz%Resolution of each perspective split.merge_resolutionii z9Long-side resolution of the merged equirect distance map. batch_sizer z,Views per inference batch (12 splits total). moge_geometryrM)rLsearch_aliasesrMrN descriptionrOrP)rrQrVrSImageInt MoGeGeometryrWrXs r-rZ#MoGePanoramaInference.define_schemaYsyy+o60y##L1w0XY /q%XZ /#4%LN /3D%`b \1!%SU "((o(FG%  r/rc , ^)^*^+^,URSS:wa[SURSS35eUSSS24n[URS5[URS5p[U[ Xx5- S5n [ [Xy-5S 5[ [X-5S 5p[ 5up[ RRUR5 URnUSRS S 5RXRS 9n[XX5nURSn/nXnnURUU45 [ UU5S::aO US-US-nnM0UR!5 [#U5VVs0sH unnUSU-_M snnm+U[%U5-n['T+R)55n[ R*R-UU-U-5m*Sm)/n/n[/USS9m,[1SUU5HnUUUU-nUR3UUSS SSS9nUR5[7USR95R;S S9R=5R?555 UR5[7USR=5R?555 URSnT)U- m)T*RAT)5 T,RCU5 M SSS5 [/UU-SS9m,U)U*U,4SjnU,4SjnU)U*U+U,4Sjn [EXUU[7U 5U UUU S9 un!n"SSS5 [FRH"W!5n![FRH"W"5n"X4Xx4:wa[FRJRLROU!SXx4SSS9RQ5n![FRJRLROU"SR95Xx4S S!9RQ5S:n"U"RS5(aJU"RU5(d5[FRV"U!U"S"5S#-n#[FRX"U"U!U#5n![FRH"[[[]Xx555n$U$U!S$-R_S5n%U!R_S5n&U"R_S5n'U%U&U'UR=5S%.n([`Rb"U(5$s snnf!,(df  GN=f!,(df  GN=f)&Nrrz9MoGePanoramaInference takes a single image (got batch of ).rrr2 rdevicedtypeTrrzMoGe panorama inferencetotaldescgV@F)r|fov_xforce_projection apply_maskapply_metric_scalerrmaskMoGe panorama merge: viewscT>TS- mTRT5 TRS5 g)Nr)update_absoluteupdate)donepbartqsr-_on_merge_view5MoGePanoramaInference.execute.._on_merge_views% $$T* ! r/c4>TRSUSU35 g)NzMoGe panorama merge: solving x)set_description)whrs r-_on_solve_start6MoGePanoramaInference.execute.._on_solve_starts""%B1#Qqc#JKr/c>TX4nTU- mTRT5 TRU5 TRS5 g)Nr)rrr)rrweightrr solve_weightrs r- _on_solve_end4MoGePanoramaInference.execute.._on_solve_endsA%qf-$$T* &!""#?@r/)on_viewon_solve_start on_solve_end)NNbilinear)sizemode align_cornersnearest)rrgffffff?g@).N)rr0rrz)2r5 ValueErrorintrrr r_model_managementload_model_gpupatcher load_devicemovedimtorr appendreverse enumeratelensumvaluesr` ProgressBarrr4inferextendlistfloatrCcpunumpyrrr r from_numpyr"r# interpolatesqueezer6rr8rr rr rrb)-rYryrzr|rrrHWrBmerge_hmerge_w extrinsics intrinsicsrimg_chwsplitsn_views merge_levelsw_h_r;whn_merge_view_unitsn_merge_solve_units distance_mapsmasksbatchresultr,rrr pano_depth pano_maskfar directionsrr0rrrrrrs- @@@@r-reMoGePanoramaInference.executeps ;;q>Q XY^YdYdefYgXhhijk kc2A2g5;;q>"C A$71$s1y0#6s19~r2CAI4K!5!7  --j.@.@A''(""2r*--VCSCS-T%g:X,,q//1 B   R )2r{c!1WbAgB  09,0GH0Guq"AF 0GH $s<'88!,"5"5"78{{&&w1C'CFY'YZ  &? @B1gz2qZ0#))%BR04t5:u*V$$T&*:*@*@*B*G*GB*G*O*S*S*U*[*[*]%^_ T&."4"4"6"<"<">?@KKN $$T* ! 3A*-@@Gc dhj   L A A%9-Z8H*&Ub%d !J %e,%%j1 $$Y/  ! ',,88J9OWXV\cm~C8DLLNJ++77 *8M8S8S8U]^\bir7s{{}ABBI ==??9==??..I!6=CCY C@J%%&@!&PQ z)44??B$$Q'""1%$*E4RWR[R[R]^ }}]++AIA @e ds S,C4S2 4T2 T TrgN rirjrkrl__doc__rmrZrrbrernrgr/r-rprpSs@   ,],momzmz],],r/rpcP\rSrSr\S5r\S\R4Sj5rSr g) MoGeInferencec[R"S/SQSSS[RS5[RRS5[R RSS S S S S 9[R RS SSSSSSS9[R RSSSSSS 9[RRSSSS9[RRSSSSS9/[RSS9/S 9$)!Nr)rsr0rurvrwzRun MoGe Inferencez:Run MoGe on a single image to estimate depth and geometry.rxryrzr|r}rz0 = fastest, 9 = most detail.r~ fov_x_degreesr3g@e@g?TzHorizontal field of view of the source camera. Sets the focal length used to unproject the depth map into 3D. 0 = auto-recover from the predicted points.)rrrstepadvancedr{rrr@zHImages per inference call. Lower if you OOM on a long video / image set.r)rrrzSet masked-out (sky / invalid) pixels to inf in points and depth so meshing culls them. Disable to keep the raw predicted geometry everywhere; the mask is still returned separately.)rrr{rrrLrrMrrNrOrP) rrQrVrSrrFloatBooleanrrWrXs r-rZMoGeInference.define_schemasyy#c-T0##L1w' /q%DF%VYdh(CD \1!%oq   !3TD Q   td*a!b "((o(FG'  r/rc ^USSS24nURSS5R5nURSn US::aSO [U5n [R R U 5n /m[U SS9n [SX5Hdn XX-nTRURXU XgS95 U R[X-U 55 U RURS5 Mf SSS5 U4S jnS UR50nS HnU"U5nUcMUUU'M [R "U5$!,(df  NW=f) N.rrrrzMoGe inferencer)r|rrrc>TVs/sHoU;dM XPM nnU(a[R"USS9$S$s snf)Nrr)rcat)fieldcvalschunkss r-stack$MoGeInference.execute..stacks<&,;f HAHfD;-1599Tq) ;t ;c2A2g}}R$//1 JJqM#q(deM.B{{&&q)  0 1R1a,q~. j..u_b@P/ij$$S%;< %++a.) -2 <!%))+. HEe A}'( e$I}}]++#2 1s 8A5D>> E rgNrhrgr/r-rrs>  .,vxwDwD,,r/rcT\rSrSrSr\S5r\S\R4Sj5r Sr g) MoGeRenderiz;Render a visualization or mask from a MOGE_GEOMETRY packet.c[R"S/SQSSS[RS5[RRS/SQS S S 9/[R R 5/S 9$) Nr )rsrenderrur0rzRender MoGe Geometryz3Render a depth map or normal map from geometry datarxroutput)r0 depth_colored normal_openglnormal_directxrr0zDirectX vs OpenGL controls the normal-map green-channel convention. DirectX: green = -Y down (Unreal). OpenGL: green = +Y up (Blender, Substance, Unity, glTF).)rKrr{r)rrQrrSrRrrWrXs r-rZMoGeRender.define_schema spyy L/M0""?3x1vAH~ XX__&'  r/rc~US;nURS5nUS;aSU;a [S5eUSnOUU(a#SU;aUSnOBSU;aUSnO6[S5eUS :XaS U;a [S 5eUS nO[S U35eURS n[RR U5n/n[ US U3S9n [U5GH^n XZU S-R5n US;ag[U 5n URUS:Xa [U 5O:U RS5R"/U RQSP76R55 OU(a\SU;aU O [U 5n U(aSOSnXR!SUS/5-n URU S-S-R#SS55 OPUS :XaJURU RS5R"/U RQSP76R55 UR%U S-5 U R'S5 GMa SSS5 [(R*"US S9R-[R.R15[R.R35S9n[4R6"U5$!,(df  N=f)N)rr_opengl)r0rr0z"moge_geometry has no depth output.rrzDmoge_geometry has neither normals nor points to derive normals from.rz!moge_geometry has no mask output.zUnknown output mode: rz MoGe render: rrrrrr2g?r3rr)endswithrr5r_r`rrr4rrDr_turbor expandrr. new_tensorr9rrrrrrintermediate_deviceintermediate_dtyperrb)rYrr is_normalopenglsrcrrr:rr;slcr<r,y_signrs r-reMoGeRender.executesAA + / /m+ !EFF(C =(#H-]*#H- !ghh v ]* !DEE'C4VH=> > IIaL{{&&q)"$ -x 8 9R1XAEl((*77,S1AJJFo,Evay$%KKO$:$:$GAGG$GQ$G$R$R$TV'=8>RSV>WA%+TFLL#vt)<==AJJC# 44S#>?v%JJs}}R077FFAFQQST$$QU+ ! :"3A&))1G1G1[1[1]eje{e{fOfOfQ)R}}V$$%: 9s E/J.. J<rgNrrgr/r-r r s6E    ,%r}},%,%r/r cT\rSrSrSr\S5r\S\R4Sj5r Sr g)MoGePointMapToMeshiKzLTriangulate one image of a MoGe point map into a Types.MESH (UVs + texture).c[R"S/SQSSS[RS5[RRSSSS S S 9[RRS S S SSS 9[R RSSSSSSS9[R RSSSS9/[RR5/S9$)Nr$)rsmeshruz point mapzConvert MoGe Point Map to Meshz(Convert a MoGe point map into a 3D mesh.rxr batch_indexrizWhich image of a batched MoGe geometry to mesh. Per-image vertex counts differ, so batches can't be stacked into a single MESH.r~ decimationrz#Vertex stride; 1 = full resolution.discontinuity_thresholdg{Gz?r3r2g{Gz?z@Drop pixels whose 3x3 depth span exceeds this fraction. 0 = off.)rrrrr{textureTz8Carry the source image through as the baseColor texture.)rr{r) rrQrrSrrrMeshrWrXs r-rZ MoGePointMapToMesh.define_schemaNsyy(D9B0""?3 ]A1$&_` \1!%JL8$CUX_c'ik   D)c!e WW^^%&%  r/rcSU;a [S5eUSnURSnX':a[SUSUS35eSU;aUSUOSn[XbUXHS9upn U RSS:XdU RSS:Xa [S 5eS U;aU SS2/S Q4R5n O=U [R "/S QU R S 9-n U SS2/SQ4R5n U(a USX"S-OSn [R"U RS5U RS5U RS5U S9n [R"U 5$)Nrz#moge_geometry has no points output.rz batch_index z, out of range; moge_geometry has batch size .r0)r(r*r0zOMoGe produced an empty mesh; try discontinuity_threshold=0 or apply_mask=False.r)rrr)r2rr)r)rrrrzr)verticesfacesuvsr+) rr5r rrtensorrrMESHr rrb)rYrr'r(r*r+rr edge_depthvertsr1r2texr&s r-reMoGePointMapToMesh.executeesd = (BC Cx( LLO  |K=8defdgghij j=D}Q %++a.A"5no o } ,!Y,'224EELL):%++NNE!Y,'224EELmG$[qARVzz__Q'//!$ a    }}T""r/rgNrrgr/r-r$r$Ks9V  ,!#bdbobo!#!#r/r$cL\rSrSr\S\\\R4Sj5r Sr g) MoGeExtensionirc@# [[[[[/$7fN)rFrrpr r$)selfs r- get_node_listMoGeExtension.get_node_lists}.CZQcddsrgN) rirjrkrlrrtyper ComfyNoder>rnrgr/r-r:r:s, eT$r||*<%=eer/r:c# [5$7fr<)r:rgr/r-comfy_entrypointrCs ?s ))rr comfy.utilsr_rTcomfy_api.latestrrrtyping_extensionsrcomfy.ldm.colormaprrcomfy.ldm.moge.modelrcomfy.ldm.moge.geometryr comfy.ldm.moge.panoramar r r r rcomfy.model_management tqdm.autorCustomrVrTensorr.rDrArFrprr r$r:rCrgr/r-rOsT 66&.*9KK ,' yy)  E E%,, E $,BLL,({,BLL{,|4,BLL4,n@%@%F<#<#~eNe  r/