From 073ff8e02f993ea3348a9564a94bda1a04ad514d Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 16 Jan 2023 21:12:17 -0600 Subject: [PATCH] fix(api): use correct scale for background correction --- api/onnx_web/upscale.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/upscale.py b/api/onnx_web/upscale.py index 439e437a..5993424b 100644 --- a/api/onnx_web/upscale.py +++ b/api/onnx_web/upscale.py @@ -103,6 +103,18 @@ class UpscaleParams(): self.platform = platform self.half = half + def rescale(self, scale: int, outscale: int = 1): + return UpscaleParams( + self.upscale_model, + correction_model=self.correction_model, + scale=scale, + outscale=outscale, + denoise=self.denoise, + faces=self.faces, + platform=self.platform, + half=self.half, + ) + def resize(self, size: Size) -> Size: return Size(size.width * self.scale * self.outscale, size.height * self.scale * self.outscale) @@ -165,7 +177,8 @@ def upscale_gfpgan(ctx: ServerContext, params: UpscaleParams, image, upsampler=N return image if upsampler is None: - upsampler = make_resrgan(ctx, params, tile=512) + bg_params = params.rescale(params.outscale) + upsampler = make_resrgan(ctx, bg_params, tile=512) face_path = path.join(ctx.model_path, '%s.pth' % (params.correction_model))