去掉了一些调试代码
parent
2d616fb151
commit
94b2b0bb60
4
app.py
4
app.py
|
|
@ -3,7 +3,7 @@ Author: SpenserCai
|
|||
Date: 2023-07-28 15:49:52
|
||||
version:
|
||||
LastEditors: SpenserCai
|
||||
LastEditTime: 2023-08-03 15:04:13
|
||||
LastEditTime: 2023-08-03 16:22:46
|
||||
Description: file content
|
||||
'''
|
||||
from deoldify import device
|
||||
|
|
@ -26,14 +26,12 @@ def image_to_base64(image_path):
|
|||
with open(image_path, 'rb') as f:
|
||||
image = f.read()
|
||||
image_b64 = base64.b64encode(image).decode()
|
||||
print("image to base64")
|
||||
return image_b64
|
||||
|
||||
def ColorizeImage(base64str, render_factor=50, artistic=False):
|
||||
vis = get_image_colorizer(root_folder=Path("models"),render_factor=render_factor, artistic=artistic)
|
||||
# 把base64转换成图片 PIL.Image
|
||||
img = Image.open(BytesIO(base64.b64decode(base64str)))
|
||||
print("loaded image")
|
||||
outImg = vis.get_transformed_image_from_image(img, render_factor=render_factor)
|
||||
return outImg
|
||||
|
||||
|
|
|
|||
|
|
@ -3,7 +3,7 @@ Author: SpenserCai
|
|||
Date: 2023-07-28 14:41:28
|
||||
version:
|
||||
LastEditors: SpenserCai
|
||||
LastEditTime: 2023-08-03 15:11:07
|
||||
LastEditTime: 2023-08-03 16:23:53
|
||||
Description: file content
|
||||
'''
|
||||
# DeOldify UI & Processing
|
||||
|
|
@ -24,17 +24,16 @@ warnings.filterwarnings("ignore", category=FutureWarning, message="Arguments oth
|
|||
|
||||
class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):
|
||||
name = "Deoldify"
|
||||
order = 20999
|
||||
model = None
|
||||
order = 20001
|
||||
|
||||
def ui(self):
|
||||
with FormRow():
|
||||
is_enabled = gr.Checkbox(label="启用")
|
||||
is_enabled = gr.Checkbox(label="Deoldify")
|
||||
is_enabled.value = False
|
||||
render_factor = gr.Slider(minimum=1, maximum=50, step=1, label="渲染因子")
|
||||
render_factor = gr.Slider(minimum=1, maximum=50, step=1, label="Render Factor")
|
||||
render_factor.value = 35
|
||||
# 一个名为artistic的复选框,初始值是False
|
||||
artistic = gr.Checkbox(label="艺术化")
|
||||
artistic = gr.Checkbox(label="Artistic")
|
||||
artistic.value = False
|
||||
|
||||
return {
|
||||
|
|
@ -51,8 +50,6 @@ class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):
|
|||
def process(self, pp: scripts_postprocessing.PostprocessedImage, is_enabled, render_factor, artistic):
|
||||
if not is_enabled or is_enabled is False:
|
||||
return
|
||||
|
||||
print(type(pp.image))
|
||||
|
||||
pp.image = self.process_image(pp.image, render_factor, artistic)
|
||||
pp.info["deoldify"] = f"render_factor={render_factor}, artistic={artistic}"
|
||||
Loading…
Reference in New Issue