Skip to content

Commit 6577e06

Browse files
authored
Update postprocessing_gfpgan.py
Fix gfpgan extension , Inconsistent overlay layer types when visibility value is less than 1
1 parent 7953c57 commit 6577e06

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

scripts/postprocessing_gfpgan.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,15 @@ def process(self, pp: scripts_postprocessing.PostprocessedImage, enable, gfpgan_
2626
res = Image.fromarray(restored_img)
2727

2828
if gfpgan_visibility < 1.0:
29+
30+
# Ensure consistent size
31+
if pp.image.size != res.size:
32+
res = res.resize(pp.image.size)
33+
34+
# Ensure consistent mode
35+
if pp.image.mode != res.mode:
36+
res = res.convert(pp.image.mode)
37+
2938
res = Image.blend(pp.image, res, gfpgan_visibility)
3039

3140
pp.image = res

0 commit comments

Comments
 (0)