From 2e99c8acf1cc1a464d9aae489ff4804dadf4b000 Mon Sep 17 00:00:00 2001 From: phamhungd Date: Tue, 3 Feb 2026 13:37:37 +0700 Subject: [PATCH] update name --- __init__.py | 12 ++++++------ __pycache__/__init__.cpython-311.pyc | Bin 608 -> 615 bytes __pycache__/node.cpython-311.pyc | Bin 23168 -> 23763 bytes node.py | 8 ++++++-- 4 files changed, 12 insertions(+), 8 deletions(-) diff --git a/__init__.py b/__init__.py index 37557e9..f0b7a7f 100644 --- a/__init__.py +++ b/__init__.py @@ -2,12 +2,12 @@ from .node import * from .install import * NODE_CLASS_MAPPINGS = { - '🧩 SAMModelLoader': SAMModelLoader, - '🧩 DinoModelLoader': GroundingDinoModelLoader, - '🧩 DinoSAMSegment': GroundingDinoSAMSegment, - '🧩 InvertMask': InvertMask, - "🧩 IsMaskEmpty": IsMaskEmptyNode, - "🧩 SegmentAnything": SegmentAnything, + '🧩 SAMModel Loader': SAMModelLoader, + '🧩 DinoModel Loader': GroundingDinoModelLoader, + '🧩 Dino SAMSegment': GroundingDinoSAMSegment, + '🧩 Invert Mask': InvertMask, + "🧩 Is Mask Empty": IsMaskEmptyNode, + "🧩 Segment Anything": SegmentAnything, } __all__ = ['NODE_CLASS_MAPPINGS'] diff --git a/__pycache__/__init__.cpython-311.pyc b/__pycache__/__init__.cpython-311.pyc index 6524ca56410baf5b28ed26a0d5672f400c8359ea..294b960d862cd614d2c303f7872dd8d547c2d4a3 100644 GIT binary patch delta 186 zcmaFB@|=ZtIWI340}w2p-k8}ikvFPd1jzX?fB8yb9QJ92F zW?nvu3|tFH2Ba@IH9a>qucTA}ro}U_EVZaa!8fruyHp4+SPT|WaLp|!se~EN1+*BV hTVdl%H%38O?HjVDH)M=&NGRQq(!3$7GueSj0RUCKK$!pl delta 179 zcmaFP@_>bRIWI340}vdV)R)S^-mn7B)3 zUOut}Tmwh~s3|x#JvTM4q?8|~!85NcwW!25u{gU_04`Px;=AS+lvE;22I*u5YTfw3 djZsiW>xPWU4QZnr;)*vU)NjaWPj+Wg001ANKfV9} diff --git a/__pycache__/node.cpython-311.pyc b/__pycache__/node.cpython-311.pyc index 893e8c55ee5c1905c6572d3819ea7b09f815da7e..617de81aa2c41924f4dd27a6ebc10761bc26aee6 100644 GIT binary patch delta 4459 zcmaJ_eQ;FO6@T|_vLD%OvPpI~+07>*Uz>y^Bz%NKHrWIU!I1A@ePz0A@*c^`+Yj#B zAW$?oWgKklS1$#v)Iw`~t&bjBFdvEqWN8WgmWIgV1WC-w_{MlWRq1Fep3do5EtDl?}RZG+=*@z$u zGEp<64F0xDcK*(k9Pl=&SHzQdwiC>QZ}@nKPFluPyrQqzR6mB^+0h_qBLgMPET2(U$7A(sPfnY=ruO1TQKRjvkHCD#D9$$r3gQCKI)wflr=x$c+039P`cSM%BhHk=vdlzEQe7FZk~}xaQxAY{j%ZsWszUQwVP-QaW__85 z+Dd`p7byzq7B!+J2CUoZPOPqwz7&uagj6 z%v5JHX=1;07L;Cf!a5e#lvp^b&`KPbk6>m0aIPcWtR!a$+}fVwB0cQxoX<&VYIp8Q z(QQVVD;nr(>=v+o_o8Ax+6#T%tn38eG~F5vZ&M%!8ck>Sx(Ds(H*M#TI`8%nQu<8( zTG5_^lsp6{8}NF_DmLM5Anoj+x1W@-v))Ki0GS;Kod`jME&zWz?PlA2h3;0Qv;k=8 z0A={lV)lrygea+(d^U1zY=;*$;9wJN#9`k2{}*RsayUe;B$ z31;r$s>y~uAPVW1t|e(C0R~m!P!iG2iAf4mqnV)3eStT{!TXxgK7Wp$uEuXyDE%Q0 zuqrWGyFHSeq&Eswh<#MuP<$5(+LYK7Om;Y_=q6Q(`JMC@R7_b@O%Z8h8*BP2_aXs> z=`?_bS%nT?c{)4y(&E&qngb%)$u`whkr2D1&NoUy4y>l}Xd>>x*(~VtTzKn9lsRI6l5)m~NpGqdClJsSk+4MdPxX|=xviu3q%2xzE4lREseG&=3 zL-;*H3gKCVrxBh*c!s^#>}kTercWXK0l;sg&tviz#vtZWXvc=-)BQWG+_!ETLcqhel)Jf1Up6`Sf{R_gq0LMi=T~6*w zC_K0Q8T1#(xoRl7p|lFk^LixkDdNi&S0c|P1B^;)M+KTP5mjPI`flpe_Fl*?c3;;f z_Q2|p4_)#T+5)fRL^qAAnnq9yWTP!0fuJ0S*Yom0Xb;Mo^fXOTw^AO9w}EnPu+On? z1FK-zs*YB2meG!?l5-$4ztmuN_A)qG)8Hz0p`(~rEIP~Wn(W+M{60v4T@xtyA;4v; zuz!}l*ZEyLFI<>Y^c-6g+*~(jqyiMEEdN9XxA7>n7TWl?V0Rgg@1vif+`K}F-7h(? zhAzpl>PxWdwEY95;too`Vv+6*`9Fpx#3HFA&`5YrclgcpZT8o0UoD@gUjdE&8Uag- z@saCY$y#$fcb+RCd0eDlr(Rn_Ow+I6I1%9dw*Y$D(7LW5-HAk8E%a4{T>$=EdUf?; zYU*j@Y9ui)nfS^{pNhnit;V8y9LK(a@Gi>376z+VruQS|QG}2nED`A6k!czaC;d4_ zNX?KP6sMgo^pWjYzbKC{A2VN#+FW9?->zS7>;~sjAFV%3{!id9q25dE=!TFH_)9z; zHs5@o!h`=cvcEz24+6%-D0KoFMG!;)*n+r$N0EQ6m9Sf3rvIhHbBeeGOO3n&~n zvm-?=^B2%O1m~4bAM0G|VJ`)o?AH8DHoWoC0$KEeTI0FHCY$!ycEi#9NE!>e+$Pg) z=`2`~pPrF)2m|#G!lFgoVxij{d_GB0*H8z;^P-Dq9YVaGtaF8Se9pz!>Pc8{&~z4{kTN!1fqa6O&;*bAT#4BFgQYtcz0-2tqLm zn_<&}ItP`07A&WHo$}P@(-1^j8^CJa8tm;2;fYbVC_9yLC{sEVl^u#v_B6l4NR1l? zoi+5;ujWNrw?|?-6q*cSUDfSjG;XeNqB$A_XMXX*4UwD0;(aCL9d=(|WIEgZK=>!S z_NTp-?|W$E!QF@AKaC$z9#hYFT4p>g`|a;|N=}zTbJ-csiW$#}H$8#ZJ%JadPHsEn z8Jh77ogN;U@r;6)!~H;P#!-33Q9a|Re$&zNx})W(o6k5pXB?d;T{DjF>=)m1=O0;d zI=dEs)>?i>f>MFgu&`DXfa!1`Eb8wnwWim7CR`8339Epx`@2Tj>-{UMc`oA~dYn)z z4(H|P`pVEWP)P1!y#p1_1vy#A_71cJc+%t0%he85>Lxf8EU0!s-ksgUW4{qX-w+K) zVh7m8fx-&Zd*c!aw=;nv?1uuq$-)h&K^F?xZwBuyti5bD9OlTC9RsvMZ4L~T*&jxJ zJwhpq3>BGqZhg$2&hWCwhV~BmMI&?lX;@~!=&-69#}l}I;*pY9Z}f*>A+DqX5O%wI zH1uLE9K3e*!PS7fbBt-jZJW6}aBBDz4fz+s1%%HK&LVscpl1%P9T^?!3-tvzt{tZT zM*2mB8|_m~=mNq+%r>&6iid&I_-4~(TzyR1E{$K3_`!(|vinCo^?G^?jMJzIRhOcg z6t>V^V|(Xi+BDG9OVilNkpeFsb?<4q7)Hb0cd*c&ul%>5$ya}NRVKV~1SCn9@ z9)Z_cJTVwIAl#O>7f-X8`NMCF@OK<&>%f z=z54Q`~>YQ#x@Ok%mQgXEi9e?drPpV&Tq;!)f5w0qSwPkqz1mZj#M|yq3JkgOCLBq)1Xi5)u;V6<9WVZ<2+(kDPl` z2%RCR;^-h!@s#PH)WPbYv8}dU(b7WAIOvSyI8$r?VD1dlT17-@ZIy8bYpvgRZW171 z@9h5Wx!?Dl?|!fIopXPGk<4EpsozgYNfO|?@{K1$r5jGA=8{)VRlG1q6q{(11c@ki z(a!IQViMdu`kVjdscmHcpl=$#7}iiMnzs1${Z#S%VFQr3h^#WFr7O1W6hmBUL+ zf}C=o0^Pt%NF`bdDJuGdX4fW}lt2iEAa7SS961 zxsv;=N%BZ}{GNZ-BvwlW#{{uvQkWo;_ZSgtr9ybFlZpV>N$aIzpw&wy@YEo!0c?~? z0h^>U!1Yo&V6#*KxIwA}Y%vL21<7|zkk+0zf!XjEggFGUm9wgVRehJmIc=O%1Dx7b zoQ<4Q2b^`QIGZ@99ykrFIPIL%2%M%RPBy*;F0n%*BZ9B_CO!YgX5Ofnh<^bp^lQK9oqbUAZK%3I7N_28k zrn==&fJTfL<)WBXJ`A3#x;-2?D1(J6wXm16cO>JRX&Z;Q!{s9IzBM^pOeIbvWguV( z^aSqO=X^xBsu~RiHQK_CyK6}syXYRY7NAHWJLS$|C7vNmJ3O*!PnN}(KzrCpPhR#0 zRMZNfq8~+gLA2lSc!?Ch=1GLNLKY5o(;DFV%(NE8MTB(-^$13WV5H~r98S{cOhgJr zCgC47B2On6%U=r~ypey@oYoAoss&)_ZQF`20n1+eiJ25LYuT$FOskHigUv~|Q8}gr zf-=}##;%raG$P{z){(6&P*ljgzASdUB8$o8V_@#3@~OrrfC&*vz=*`a>@w6tLb^3J zMIqm+8J@YPaHkJw>Tlqco31A)(L*w&U%=URIU=ctLfRC?Akq$mPJ}Lmeav22+cpnF zJyDL#K#2u3SvMHxsk>_>nLN+7K|kDQNJRd-hg*F#RTx9fV`d|9Li zsSIxVV9W0xj5WRs_70}H<6mn^GL3!{X5E0lAiNvx;#IGcZlCF97f|RGgcAS@ zCOtvYW@9p6I=&=&7CCo4K>u{x1hfsUu>p$XR&!k?PgA1d51kz6M&4hDmr==Owt0gN zA@{bNF=g~1r3qkX?M?3fx7e@SO6C}$a(Y8qhRK zxe_aCS}{;hRG~K+wGcyP>48{Gne{_e%Bo+B(%|Yc*HdB?x>HaCZWFS*aGDJw5T4Ow z9uygNbh@zlYSnApDD+*!*bDaTr2iG&wd2 zBe>q|^jYau_WowiS}a%k9fao)cr)QIdB87L|HwkdJ+6TGf{A_{zx4<)&;0@?n*eTp z3ZPqtxAt^XEL5Xpeu<;Q0KSZS);HFvZV4)(*gb{C*HpqxD5A9(%k4bQeGB0l%ELYi zYg?ixk@76UK0&Av=x>l|4hA$|x)IU@C>4q|U{vNkKC&~t#hH9@S^3&fm#UQgws*ai z$Mr_s-glOKM&R$D-aG8dqkbdscX(~gZ96%~ga0wI|AX)e!haDkIuxTr1++_*zvclLuzl^L?9&|^DtLsk_b(7?!|DI0 zUiS{?kuR~?;nLOdC}A%SH+Jw;;MC~O2rAdruw^+g_+skR;3aBVmcihP}U zNAgNB&JTABzH0=Ea146Ln2jf-5M9V+Z;c$yD_%AmPA_Dosev|nogLg+)NmU4YY=dz zZXFoi-m_&`PaTg&6*&;`!_i|LPAoiE*jR!lHlLEkez$Xeka}PmBy%55GzS#L*nr_{ z4NqD;ndm4#l;a5i#$3;ltUeXp+XBan*#S65b3dPB-yLn+!~KNq?-mNbkATHQ|BitD zj{XBccMNaYIXXPxAL!n`WrTi&^bZjp)?+KsH-x8I>)8Hs9v@EQy|L5ym@uoeDnGvS zlK>rJKOb}D8Al29&e(`}6EP3<^;0t=yK;cI1qtD65n~++2Af$o)twhaYve(t{tBVDaOaZ)HHl5%9&^5a zcnYu@{%zj`@tINN4%z{7@DtSQ?Cssd7P~+k@lB6qm@C}`mg5z=noWi+$L_dI4n G&Hn?NA`?;o diff --git a/node.py b/node.py index cbf6334..bf336c3 100644 --- a/node.py +++ b/node.py @@ -3,7 +3,7 @@ import sys sys.path.append( os.path.dirname(os.path.abspath(__file__)) ) - +from nodes import NODE_CLASS_MAPPINGS as ALL_NODE import copy import torch import numpy as np @@ -383,7 +383,11 @@ class SegmentAnything: sam_load_model = SAMModelLoader().main(sam_model)[0] dino_load_model = GroundingDinoModelLoader().main(dino_model)[0] output = GroundingDinoSAMSegment().main(dino_load_model, sam_load_model, image, prompt, threshold, threshold) - return (output) + mask = output[1] + invert_mask = (1.0 - mask).to(image.device) + alpha_image = ALL_NODE["JoinImageWithAlpha"]().execute(image, invert_mask)[0] + ui = ALL_NODE["PreviewImage"]().save_images(alpha_image)["ui"] + return {"ui":ui, "result": output} class InvertMask: @classmethod