apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): correct output size math for upscale and correction

This commit is contained in:
Sean Sube 2024-03-16 14:26:43 -05:00
parent 3765fb9cbb
commit 7b0d73664d
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
api/onnx_web/params.py
View File

@ -473,13 +473,19 @@ class UpscaleParams:
)
def resize(self, size: Size) -> Size:
face_outscale = self.face_outscale
if self.upscale_order == "correction-both":
face_outscale *= self.face_outscale
face_outscale = 1
if self.faces:
face_outscale = self.face_outscale
if self.upscale_order == "correction-both":
face_outscale *= self.face_outscale
upscale_outscale = 1
if self.upscale:
upscale_outscale = self.outscale
return Size(
size.width * self.outscale * face_outscale,
size.height * self.outscale * face_outscale,
size.width * face_outscale * upscale_outscale,
size.height * face_outscale * upscale_outscale,
)
def tojson(self):