去掉了一些调试代码

pull/21/head
SpenserCai 2023-08-03 16:24:14 +08:00
parent 2d616fb151
commit 94b2b0bb60
2 changed files with 6 additions and 11 deletions

4
app.py
View File

@ -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

View File

@ -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}"