kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

set face restore upscale to 1; realesrgan till to 512

Browse Source
This commit is contained in:
Qing 2023-04-03 13:19:26 +08:00
parent ed798bb47c
commit dd1d45aa79
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lama_cleaner/plugins/gfpgan_plugin.py
View File

@ -25,7 +25,7 @@ class GFPGANPlugin(BasePlugin):
# Use GFPGAN for face enhancement # Use GFPGAN for face enhancement
self.face_enhancer = MyGFPGANer( self.face_enhancer = MyGFPGANer(
model_path=model_path, model_path=model_path,
upscale=2, upscale=1,
arch="clean", arch="clean",
channel_multiplier=2, channel_multiplier=2,
device=device, device=device,

2
lama_cleaner/plugins/realesrgan.py
View File

@ -70,7 +70,7 @@ class RealESRGANUpscaler(BasePlugin):
model_path=model_path, model_path=model_path,
model=model_info["model"](), model=model_info["model"](),
half=True if "cuda" in str(device) else False, half=True if "cuda" in str(device) else False,
tile=640, tile=512,
tile_pad=10, tile_pad=10,
pre_pad=10, pre_pad=10,
device=device, device=device,

2
lama_cleaner/plugins/restoreformer.py
View File

@ -24,7 +24,7 @@ class RestoreFormerPlugin(BasePlugin):
self.face_enhancer = MyGFPGANer( self.face_enhancer = MyGFPGANer(
model_path=model_path, model_path=model_path,
upscale=2, upscale=1,
arch="RestoreFormer", arch="RestoreFormer",
channel_multiplier=2, channel_multiplier=2,
device=device, device=device,