From 7db5472531498a1af3709bbed02c592bf7053377 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 6 Feb 2023 17:59:34 -0600 Subject: [PATCH] fix(api): include face correction outscale in final image size --- api/onnx_web/params.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py index e8a3b677..f81e61c9 100644 --- a/api/onnx_web/params.py +++ b/api/onnx_web/params.py @@ -177,7 +177,10 @@ class UpscaleParams: ) def resize(self, size: Size) -> Size: - return Size(size.width * self.outscale, size.height * self.outscale) + return Size( + size.width * self.outscale * self.face_outscale, + size.height * self.outscale * self.face_outscale, + ) def tojson(self): return {