add DAT stage to compound upscale stage
This commit is contained in:
parent
4a9ca4c4a8
commit
8c54de7a52
|
@ -6,6 +6,7 @@ from . import ChainPipeline, PipelineStage
|
||||||
from .correct_codeformer import CorrectCodeformerStage
|
from .correct_codeformer import CorrectCodeformerStage
|
||||||
from .correct_gfpgan import CorrectGFPGANStage
|
from .correct_gfpgan import CorrectGFPGANStage
|
||||||
from .upscale_bsrgan import UpscaleBSRGANStage
|
from .upscale_bsrgan import UpscaleBSRGANStage
|
||||||
|
from .upscale_dat import UpscaleDATStage
|
||||||
from .upscale_resrgan import UpscaleRealESRGANStage
|
from .upscale_resrgan import UpscaleRealESRGANStage
|
||||||
from .upscale_stable_diffusion import UpscaleStableDiffusionStage
|
from .upscale_stable_diffusion import UpscaleStableDiffusionStage
|
||||||
from .upscale_swinir import UpscaleSwinIRStage
|
from .upscale_swinir import UpscaleSwinIRStage
|
||||||
|
@ -77,6 +78,12 @@ def stage_upscale_correction(
|
||||||
outscale=upscale.outscale,
|
outscale=upscale.outscale,
|
||||||
)
|
)
|
||||||
upscale_stage = (UpscaleBSRGANStage(), bsrgan_params, upscale_opts)
|
upscale_stage = (UpscaleBSRGANStage(), bsrgan_params, upscale_opts)
|
||||||
|
elif "dat" in upscale.upscale_model:
|
||||||
|
dat_params = StageParams(
|
||||||
|
tile_size=stage.tile_size,
|
||||||
|
outscale=upscale.outscale,
|
||||||
|
)
|
||||||
|
upscale_stage = (UpscaleDATStage(), dat_params, upscale_opts)
|
||||||
elif "esrgan" in upscale.upscale_model:
|
elif "esrgan" in upscale.upscale_model:
|
||||||
esrgan_params = StageParams(
|
esrgan_params = StageParams(
|
||||||
tile_size=stage.tile_size,
|
tile_size=stage.tile_size,
|
||||||
|
|
Loading…
Reference in New Issue