apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): resolve face correction model relative to model path

This commit is contained in:
Sean Sube 2023-01-16 16:45:17 -06:00
parent 1d90f9b53c
commit 5a01fe4cb0
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/onnx_web/serve.py
View File

@ -99,7 +99,6 @@ mask_filters = {
upscale_models = [ upscale_models = [
'RealESRGAN_x4plus', 'RealESRGAN_x4plus',
'GFPGANv1.3', 'GFPGANv1.3',
# TODO: convert GFPGAN
# 'https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth', # 'https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth',
] ]

4
api/onnx_web/upscale.py
View File

@ -167,8 +167,10 @@ def upscale_gfpgan(ctx: ServerContext, params: UpscaleParams, image, upsampler=N
if upsampler is None: if upsampler is None:
upsampler = make_resrgan(ctx, params, tile=512) upsampler = make_resrgan(ctx, params, tile=512)
face_path = path.join(ctx.model_path, '%s.pth' % (params.face_model)) # TODO: convert to ONNX
face_enhancer = GFPGANer( face_enhancer = GFPGANer(
model_path=params.face_model, model_path=face_path,
upscale=params.outscale, upscale=params.outscale,
arch='clean', arch='clean',
channel_multiplier=2, channel_multiplier=2,