apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

lint(api): remove unused upsampler arg

This commit is contained in:
Sean Sube 2023-02-06 08:56:34 -06:00
parent 24ab52bf0d
commit 564cfc1279
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/onnx_web/chain/correct_gfpgan.py
View File

@ -1,16 +1,13 @@
from logging import getLogger from logging import getLogger
from os import path from os import path
from typing import Optional
import numpy as np import numpy as np
from gfpgan import GFPGANer from gfpgan import GFPGANer
from PIL import Image from PIL import Image
from realesrgan import RealESRGANer
from ..device_pool import JobContext from ..device_pool import JobContext
from ..params import DeviceParams, ImageParams, StageParams, UpscaleParams from ..params import DeviceParams, ImageParams, StageParams, UpscaleParams
from ..utils import ServerContext, run_gc from ..utils import ServerContext, run_gc
from .upscale_resrgan import load_resrgan
logger = getLogger(__name__) logger = getLogger(__name__)
@ -19,9 +16,7 @@ last_pipeline_instance = None
last_pipeline_params = None last_pipeline_params = None
def load_gfpgan( def load_gfpgan(ctx: ServerContext, upscale: UpscaleParams, _device: DeviceParams):
ctx: ServerContext, upscale: UpscaleParams, _device: DeviceParams
):
global last_pipeline_instance global last_pipeline_instance
global last_pipeline_params global last_pipeline_params
@ -56,7 +51,6 @@ def correct_gfpgan(
source_image: Image.Image, source_image: Image.Image,
*, *,
upscale: UpscaleParams, upscale: UpscaleParams,
upsampler: Optional[RealESRGANer] = None,
**kwargs, **kwargs,
) -> Image.Image: ) -> Image.Image:
if upscale.correction_model is None: if upscale.correction_model is None:
@ -65,7 +59,7 @@ def correct_gfpgan(
logger.info("correcting faces with GFPGAN model: %s", upscale.correction_model) logger.info("correcting faces with GFPGAN model: %s", upscale.correction_model)
device = job.get_device() device = job.get_device()
gfpgan = load_gfpgan(server, upscale, device, upsampler=upsampler) gfpgan = load_gfpgan(server, upscale, device)
output = np.array(source_image) output = np.array(source_image)
_, _, output = gfpgan.enhance( _, _, output = gfpgan.enhance(