From 35bd8a0c855b5daabd7134788d88cb6a94b168ab Mon Sep 17 00:00:00 2001 From: mattyamonaca Date: Wed, 5 Apr 2023 10:29:42 +0900 Subject: [PATCH] fix main --- scripts/launch.py => launch.py | 4 +-- scripts/__pycache__/convertor.cpython-310.pyc | Bin 0 -> 2006 bytes scripts/__pycache__/td_abg.cpython-310.pyc | Bin 0 -> 3524 bytes scripts/main.py | 32 ++++++------------ scripts/td_abg.py | 2 +- 5 files changed, 13 insertions(+), 25 deletions(-) rename scripts/launch.py => launch.py (95%) create mode 100644 scripts/__pycache__/convertor.cpython-310.pyc create mode 100644 scripts/__pycache__/td_abg.cpython-310.pyc diff --git a/scripts/launch.py b/launch.py similarity index 95% rename from scripts/launch.py rename to launch.py index e26a930..f3fef5e 100644 --- a/scripts/launch.py +++ b/launch.py @@ -2,8 +2,8 @@ import gradio as gr import sys import cv2 -from td_abg import get_foreground -from convertor import pil2cv +from scripts.td_abg import get_foreground +from scripts.convertor import pil2cv diff --git a/scripts/__pycache__/convertor.cpython-310.pyc b/scripts/__pycache__/convertor.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..b0e678d677b89d68e881b8232b34c835f625f7b0 GIT binary patch literal 2006 zcma)6%Wfk@6z%Hj>TWv^1vDCw@K7%hq=zAafDj4@!3-KS5{gD)7RVaO>bTlYJMM?7 z+spbRI~)MB$v()R?C$_y0D(~qz~)MhHSz%B>T<^ zS1vJ&br3Qw9;Q(*?l5X!721f8?>@eA0;sLN^@NYv*dFtW_3esHc+IBFSbNGOllD1( zVvn7%sA&WIxqXFBmv3VHl26>=bIHfK&ByN88~c?r5w$aMr&cWtx&}RiKB4q}5>qyA z)D0ZRfA&i=2=3?EC{|SvWx?S$_kx5D7F2l<1z$&1^gu;vyn`_Ozf!$&AC^jz_JZxK z*vVw1RCE^ZzZ^`_RA3I{YN)dLAlVJJ3%N6&7o+C&pp$1+lw?WP3##MzY9A6?p!u|& z^g+9yWHNq=kOM{3UmBo04u$4V+XKzd08DAt)9fhpv?!0GBGz_RXg`h1Sb|V#K zJ$MVL8}kilp`uYd2%DPYaGD)!cB)xhK~|Qci&YH*x@NTk(6;PeSn4%Yf8XCd&eM1| zFOz)tG*>6PkM14D>Ek>fl#A7}qmrU3cX3prJXbr#nRZq0Xis(p<|%IjR+|BPz#Qhz zfBqkir?$|wq&F81`i$OwM1}N5nu2PabKYkp<#Xt6%;r7bwH|xk<6rCH^Pa}F9;f0H zr)ITX^V;b-vN^RUqT+oQx)QaZXR@rTg*zr$Ey(OlGE3an&~Ss5cRy z)&P)V6?IM^zOp)lB*ievs!u~#yVxR*+!CsdT1)h1=4fUjV_exKYs5B#O`6h^-94x$ zdL4~&6|mLZV)T0h`cdz|v7{J!D09U;L!+|#&<4N6k?>&pBlv8o3ZH5dGP5bkCCR&+KJv=$R}ki3&sFCC7d)|3P0mqo!=;@}CaI#sU)Z=j%3iHT7N3GIRIfAU7Rg|*GrEiL@)lr& zHJQyEvHmA0(-5~%F`KL zJLIHvwV)ukGui_spTJwb3h)gd>+CJI4jDCxjgR0)QWV)R#q>tXZV@5xmudrDnm>B@ tty#;YD5E;Vpc%NZo>|$**8e=saSvSlI~>9Ut935`x~Mn+#&ZjqxQ*`sEXPt_qU_Y zcqiJ9Hbwm<_b-Vx(GbmVng5E24w($6{4AYqDc5yf6@1%TKzqY_!F8>ml+Yo{fZv2} zCpBFsKhNvBj{WL5d$9(6>dHH5V2+iTopRxAS@g}lLdp{jNO{`(|Nd3T(J4Lc>I$8* z{8I_(&jXuyw1=o^?w_5*4QJ$^GfIr>XEvedcqll8I-W*7TT(_zxXL;sVC;A@ET&jm zrlr^Gy{2)4oTLm-r=~hcvOs;XXDaz90L_k5Wh&1indQa~pX?zbqda~VnYxmQ*=d$X zrkX}a!P7W5o}3&E`;$yWiLvD@?KLFw*)j~WBAysFGVI9k(GA0o@J&bfj{G{#){oQa zOc`eq~#q zv}iMV4QNPyzj!EGNQ**)$3PTxc+TdOEmjxW2(H;Cunp^kDL%G~tY(6B-^_{bAd}b- zIZiynd1)i}k#(qzws4Vc#eM@0&8z#ej{m#~HWV|5f<(56+yr>VBptlx78aK)MDGBY z$_zpGv1x=^8U`v#@ln`Kqd@g|5mEAI#7t50mPM*X^bOJvK+fq0HFgQsfUn_uNz{_x9f1hoAk4lp4e-66j$u zN=Q(?l-hkvvXyB*?cX=ml2N{^vRBe4QL26e_PhqrLJV{qhs#@-ghgisLJythwZ>Nv z6B-^ywf@H<8c<;bf6L>T!v^rPsKB>W&4|0FDzJs8Q5~NjBiH4Cf?Y)SKZOWX5h-PA2)OIt7LWsEDjF>a`)ZtIl+7p|ziv~@$&QGu7VBT!56 z?~d-Av~*)=iRRytd8dq7x-(t|-xB!h+Cwa_>Q&J~CRPp|V9OBdR> z)S7p(vVxVaAX--}ZBeYRoHEhbpm}`@1|vGHo^-_0DLYv`wMv=Ayj#w9tohnvhc!*F zd+2Og<8??dwQ;T0L^lJS{|DW^psOw2gQP%At{Uj9v*`<|AFr6mHMah&@^o>1;m0vA>pZARuX6Pt(uB53BWj320 z&}=p_l`xy3H9Bz~r=}98B6=#<3i%`pB^4KWlfZ9@=89Ye;bBpPdK+a^Y?TeGkNS3y zM8Z^IooNy)Q+={P>aXpGQzp|p!)qClqC7m+Keb(Vp#vbbqnM_(cfWwHz-^0je?G7xcQ z_)wbV(QG)3)8QZpqyA`i;5X=!jr)+o?b~pvqO}!m>fo@m&{#r9NS^ed5e3z)~!`)wPxerCYNEk literal 0 HcmV?d00001 diff --git a/scripts/main.py b/scripts/main.py index db3ae85..e8463b6 100644 --- a/scripts/main.py +++ b/scripts/main.py @@ -9,22 +9,10 @@ import gradio as gr import modules.scripts as scripts from modules import script_callbacks -from td_abg import get_foreground -from convertor import pil2cv +from scripts.td_abg import get_foreground +from scripts.convertor import pil2cv -""" -body_estimation = None -presets_file = os.path.join(scripts.basedir(), "presets.json") -presets = {} - -try: - with open(presets_file) as file: - presets = json.load(file) -except FileNotFoundError: - pass -""" - def processing(self, input_image, td_abg_enabled, h_split, v_split, n_cluster, alpha, th_rate, cascadePSP_enabled, fast, psp_L): image = pil2cv(input_image) image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) @@ -49,15 +37,15 @@ def on_ui_tabs(): with gr.Row(): with gr.Column(): input_image = gr.Image(type="pil") - with gr.Accordion("tile division ABG", open=True): + with gr.Accordion("tile division BG Remover", open=True): with gr.Box(): - td_abg_enabled = gr.Checkbox(label="enabled", show_label=True) - h_split = gr.Slider(1, 2048, value=256, step=4, label="horizontal split num", show_label=True) - v_split = gr.Slider(1, 2048, value=256, step=4, label="vertical split num", show_label=True) - - n_cluster = gr.Slider(1, 1000, value=500, step=10, label="cluster num", show_label=True) - alpha = gr.Slider(1, 255, value=100, step=1, label="alpha threshold", show_label=True) - th_rate = gr.Slider(0, 1, value=0.1, step=0.01, label="mask content ratio", show_label=True) + td_abg_enabled = gr.Checkbox(label="enabled", show_label=True) + h_split = gr.Slider(1, 2048, value=256, step=4, label="horizontal split num", show_label=True) + v_split = gr.Slider(1, 2048, value=256, step=4, label="vertical split num", show_label=True) + + n_cluster = gr.Slider(1, 1000, value=500, step=10, label="cluster num", show_label=True) + alpha = gr.Slider(1, 255, value=50, step=1, label="alpha threshold", show_label=True) + th_rate = gr.Slider(0, 1, value=0.1, step=0.01, label="mask content ratio", show_label=True) with gr.Accordion("cascadePSP", open=True): with gr.Box(): diff --git a/scripts/td_abg.py b/scripts/td_abg.py index 7c1eba6..257626a 100644 --- a/scripts/td_abg.py +++ b/scripts/td_abg.py @@ -4,7 +4,7 @@ import numpy as np import pandas as pd from sklearn.cluster import KMeans, MiniBatchKMeans -from convertor import rgb2df, df2rgba, cv2pil +from scripts.convertor import rgb2df, df2rgba import gradio as gr import huggingface_hub