From 62ef148f72a4db23f2eb793f98eeaf23de2fc31c Mon Sep 17 00:00:00 2001 From: xsl Date: Sun, 27 Jul 2025 17:31:05 +0800 Subject: [PATCH] save code --- src/custom_controlnet_aux/dwpose/__init__.py | 42 +++++++++++++----- .../__pycache__/__init__.cpython-312.pyc | Bin 26959 -> 28004 bytes 2 files changed, 32 insertions(+), 10 deletions(-) diff --git a/src/custom_controlnet_aux/dwpose/__init__.py b/src/custom_controlnet_aux/dwpose/__init__.py index bb3ac06..ee90929 100644 --- a/src/custom_controlnet_aux/dwpose/__init__.py +++ b/src/custom_controlnet_aux/dwpose/__init__.py @@ -344,8 +344,19 @@ def draw_poses_mask(poses: List[PoseResult], H, W, cloth_len, line_len, cloth_sh mask = torch.zeros((H, W), dtype=torch.float32) - for pose in poses: - mask = draw_pose_mask(mask, pose.body.keypoints, H, W, cloth_len, line_len, cloth_short) + def findMaxPose(poses): + len = 0 + mp = None + for p in poses: + if p.body.keypoints[0] != None and p.body.keypoints[1] != None: + l = distance(p.body.keypoints[0], p.body.keypoints[1]) + if l > len: + mp = p + len = l + return mp + + max_pose = findMaxPose(poses) + mask = draw_pose_mask(mask, max_pose.body.keypoints, H, W, cloth_len, line_len, cloth_short) return mask @@ -366,16 +377,27 @@ def draw_poses(poses: List[PoseResult], H, W, draw_body=True, draw_hand=True, dr """ canvas = np.zeros(shape=(H, W, 3), dtype=np.uint8) - for pose in poses: - if draw_body: - canvas = util.draw_bodypose(canvas, pose.body.keypoints, xinsr_stick_scaling) + def findMaxPose(poses): + len = 0 + mp = None + for p in poses: + if p.body.keypoints[0] != None and p.body.keypoints[1] != None: + l = distance(p.body.keypoints[0], p.body.keypoints[1]) + if l > len: + mp = p + len = l + return mp + pose = findMaxPose(poses) + # for pose in poses: + if draw_body: + canvas = util.draw_bodypose(canvas, pose.body.keypoints, xinsr_stick_scaling) - if draw_hand: - canvas = util.draw_handpose(canvas, pose.left_hand) - canvas = util.draw_handpose(canvas, pose.right_hand) + if draw_hand: + canvas = util.draw_handpose(canvas, pose.left_hand) + canvas = util.draw_handpose(canvas, pose.right_hand) - if draw_face: - canvas = util.draw_facepose(canvas, pose.face) + if draw_face: + canvas = util.draw_facepose(canvas, pose.face) return canvas diff --git a/src/custom_controlnet_aux/dwpose/__pycache__/__init__.cpython-312.pyc b/src/custom_controlnet_aux/dwpose/__pycache__/__init__.cpython-312.pyc index 3afe575f8d124d27c04a37a318543c3a2c148988..c6db69af00328e6515e28cb434181c7b5c0cae2f 100644 GIT binary patch delta 2935 zcmdT`|5IGm6@T}={r(`ZunQruuzU$&350+&NP?k3#C#}~R!A+b-lUjdhOCnBWYO6ixZ3>#^ zALyIi&z}99bI-l++|T`T?gF`*AS>QUO|@|F#OIIs_jR9H@uo;z>`$UXderSHD|O@` z^QNCC5w+I(DxV&JKstyJL@>=`gajF;&$7k*A`(?|3w}h(Jm7E{mOG_UYDN6x5w3PZ%uhi*T6t7tt@)y1OsiRlc+8X)%llDo;YTDba#NPfmvR=-d?fF2W7 zv5A&@8ZC^cF9Ia=)D!IU(UV9-0HDW%Y`(foJd2F>?muTmHTC2*wY%mWg9S$=z<1$Q zR$bpwBJtnmB1WDI8^hcy1bBENCdHVvI4*1mo7lzMQADbyf;I&FauyF zd>2*%`+ZI#5h8IBKFl?9J)@9L2n)=aX=5K-Y#9>Cfk=rPsN*=9j60WIc@qCM*IkMp zBDH5Z$dqiqOpGL3y}~CWBf79IOa_{@i*_cf8r$i9}srDu$pX%!3w6ZPlp6Eg3e}}wj(DP8T z9w;>@yJus8Wo!&D2@bAmcce>-^q0#2cV%q2qZP?x=X+ln%8BNSj5*F5VIE<12ayEgSO<$3- zIV{k%ijh6g?huS&GqW0P>|wW^Jt2zhdz)4Z=5M95_MMsR51Y!t&d}`C29c*ume)`S zRMt((qBp`K(5bx*et!M`=MtbZP%`>kgx3LX(Y7{&j}$CR`nGBg!%WgQF@>fhc5ZXO z+hw^`Df=*BKf(coE`+rRs1n)@Amgc^Ei4U^tIM+7IgNvV0A=$m>uW3=Ny^5?rDRVW ziHnZW<`0bNm#p?#YjMn4Jk@fl`SiBvw%LltSViOcEuC@eePf3IiD-?s#;u#i3}0`z zJGwh=ZMaQ1B#c`(j2SMA)=`!JrblP^C64|i>u=sgPN`>`uMpU$f6!7(~B*S+tF=a4v$aDi2CwQuZ26BkWF=<_Kw1^=1=Am1O6 zHTZq~eSRgOf3ROt{PYU?v5t^*bZfc(Z(u&Hp5D5S7~Vy8R=u)qhIjlKY?ts?K<%OP z@S}lCpuKEnr=5Jj-rdmv+qrJ;t4rLXRAy937F^9)>fV;Bdooxh}>XGTMb!MUvB#`PY$!rjvCU< zzSFTsB&a$b;R|-PV>NkWj27Tnvx+j;eky(jqeTlLBg@2@wcVSjP<{Qi6sj?`s^MsMx^F6m-> z4h)f1>Kg~j39+(&bTu_EVjr+6dSrRfr}#q>HDhC~NZM3txr&i{3CDzMN0U7L6fK17 zOzCbWpQ^v;-bU)RS0aflPq2T`6VP^4w@+5!kJ29s!ZqkKEeqt6@l3mUI?+XrrARq! zs?9*|RJ#sV^4Xtb+`kb14FGouJKBfYbBFRB91cgTY&MdbnzR}V;9%FxHezP;ht`ly zYPM$|pSL`PR`5eA8@9r;pH{Ks-hw;rNQ=p#Wnu1YgggWXLOwzPLLorHkemxz#6I+v zkzyuFPO_U-NbkA-i8)Ik#-*dHF&;Z?p{2-}=o)0o5L^i52v~o*7A>vrw<~lddJ(Fb zd4(#h?Ye>d~G`TUBRl2}|eXWzQxBbWd0}1B$ AHvj+t delta 2159 zcmZvde^69a6vyA)-^=o2`LW1vb~k>lzd~dr9N9{UP|yl7P=RIHU3M3jeROxRoKUDJ zvBAmdI;J&-O*y8cQ}~#KQJJzC%@|8$ig;>HT9!EqMk)1Ao%7aka(V-wJ@0$ZJ@?*o z&wcktk$iogL=HzpgmLg(KiuWnyQVMll#U!=-{^v*URWNUpddEp(4Qllh5YD#nZ5-a zsUN{3TG@1LKIszX#TkjF6BLV#cCpn459twx3_WtvDU>Ikl&{=|&gF;-fGBU}Tj^`) zuBBU0eH*bIn<4@{4Z57RR^ICm21I2bXm_j^2~T&R%TB~@wk|b;>=Qau*OQ1&P!4bw zxkxt9UN)U(CI^M&^n)a)5gIJ2`-9Sh*w=x;k*OBj!|4`w(>RxGXGx|4azv;!@kHgp zW-mK!&L#cAwD}Uz96?_d_+H8=CWC@0^Nx;u!V(s|pYOvK4eiG+QtUyr(4X!Dh)Qbr zyIph-3Y!5iqK!=~$k7d;BCY)>Gv^hOA)z*JSjAR8q$P*gg2Gh^?OX|0m(6jT)Il}) zN>970UCm66ICi`+r=!QB;M!HIAmuA9gl2&QOwkEHGzTE6n|!=IxG;|np;V2?1;Avh z&`v|7BaS0Z07RX`#|J&OfCsjK+EA_o1T+BSy@$?2$0#SYv;9jV!cM|lH36RrHnb!) z_p08|t+*JKy34t@WdE}FmW^c;jARrHL=QYZuO>X7h_yM@s94&EBMGm(w}Ss(Y4A6=ljb&3e{rO<@IA6WB3erz({J($mIX zw3>*9?XjkX_3VU34=7?WOuU-#ikBvvg5kA zi-yE8L-vRv`$N^oilf>-?O1N{NN(}DCu&9wYr0E+c+exVFLKoIXm{y<`$hFdjT#>7 zF1_-gI|LjxEZ$l=8H=lq=fYz;SI~z!PDn1PRj7`^oai35scZ##Qy4D0N=PsJX!+cn z5zt{*v2I{jN6~8>F>{9K7`~oo^71MW{X33pd~DqHw1*cSQEKz6XHY1++thprED6C`j|g0~(4u zQ@o*-%`DWS4At^XfBFmd!IcD*0I6m}6&uIODhlgKiG72CV2&_=t=UE&{T7cEb0XP-dPjXe%|J$b|<}w z&hH|k0b)cGPh0Fk8+8TRd_h?0E%x%-RkFW{Ft+v$nSKpAu0^av)UpkAg>zLcVMEMpwrS^>!mtcv8Xcy{UC$+4pzLhA1vHIBr&Va08X~_io!$GzO0xhnf zhj&sAqe>YK$AHoxNvQ3|SExizN)f$*7UBZyZYUu)gtHCH2;^y2qcPzYwq@bjk%472 zhI&kJ%;-r}d;Q*^&1M$(#(A=*enZbG#P0z3+8EHrPotRN_VJB&pUq)+c;E|f_j_CH zzE(cqqIc0pO8G3bq*9cU{}-e^UPd`4cV&}IVUDXtmU4f$H83F-F#FVumX!7XOLGRcB9hq)-i0KC9r6~1t{f+u@Gc@}SXT3LqGiuD zvyOTQC#tGkez+8+>w|XT06i#co7()2AkX^(bQ_9z-GuHq>GtWz*FnS(0zcGJX^9G_ z*Ab+6#6@`^NTq_Jcd%J1fSiA)2jyV`ai%^*&5{=0+2(Uu=~D24KQ3@o_+WhkF{_Dw jSN`6xk>oj39N4#GQb^2p=dQ&g@mW(G*ppUv+4t%{6qY|(