4 Commits

Author SHA1 Message Date
  w-e-w 1a773bf2c8
Merge pull request #16751 from Neokmi/master 1 year ago
  w-e-w f113474a6e lint 1 year ago
  klx 6577e063d1
Update postprocessing_gfpgan.py 1 year ago
  klx 7953c570d9
Update postprocessing_codeformer.py 1 year ago
2 changed files with 8 additions and 0 deletions
Split View
  1. +4
    -0
      scripts/postprocessing_codeformer.py
  2. +4
    -0
      scripts/postprocessing_gfpgan.py

+ 4
- 0
scripts/postprocessing_codeformer.py View File

@@ -29,6 +29,10 @@ class ScriptPostprocessingCodeFormer(scripts_postprocessing.ScriptPostprocessing
res = Image.fromarray(restored_img)
if codeformer_visibility < 1.0:
if pp.image.size != res.size:
res = res.resize(pp.image.size)
if pp.image.mode != res.mode:
res = res.convert(pp.image.mode)
res = Image.blend(pp.image, res, codeformer_visibility)
pp.image = res


+ 4
- 0
scripts/postprocessing_gfpgan.py View File

@@ -26,6 +26,10 @@ class ScriptPostprocessingGfpGan(scripts_postprocessing.ScriptPostprocessing):
res = Image.fromarray(restored_img)
if gfpgan_visibility < 1.0:
if pp.image.size != res.size:
res = res.resize(pp.image.size)
if pp.image.mode != res.mode:
res = res.convert(pp.image.mode)
res = Image.blend(pp.image, res, gfpgan_visibility)
pp.image = res


Loading…
Cancel
Save
Baidu
map