Update mask.py
Browse files
mask.py
CHANGED
|
@@ -37,7 +37,7 @@ class FaceSwapper:
|
|
| 37 |
if GFPGANer and os.path.exists(gfpgan_model_path):
|
| 38 |
self.gfpganer = GFPGANer(
|
| 39 |
model_path=gfpgan_model_path,
|
| 40 |
-
upscale=
|
| 41 |
arch="clean",
|
| 42 |
channel_multiplier=2
|
| 43 |
)
|
|
@@ -48,7 +48,7 @@ class FaceSwapper:
|
|
| 48 |
model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64,
|
| 49 |
num_block=23, num_grow_ch=32, scale=2)
|
| 50 |
self.realesrganer = RealESRGANer(
|
| 51 |
-
scale=
|
| 52 |
model_path=realesrgan_model_path,
|
| 53 |
model=model,
|
| 54 |
tile=0,
|
|
@@ -107,7 +107,7 @@ class FaceSwapper:
|
|
| 107 |
# تكبير Real-ESRGAN لو موجود
|
| 108 |
if self.realesrganer:
|
| 109 |
try:
|
| 110 |
-
blended, _ = self.realesrganer.enhance(blended, outscale=
|
| 111 |
except Exception as e:
|
| 112 |
print(f"⚠️ RealESRGAN فشل: {e}")
|
| 113 |
|
|
|
|
| 37 |
if GFPGANer and os.path.exists(gfpgan_model_path):
|
| 38 |
self.gfpganer = GFPGANer(
|
| 39 |
model_path=gfpgan_model_path,
|
| 40 |
+
upscale=1,
|
| 41 |
arch="clean",
|
| 42 |
channel_multiplier=2
|
| 43 |
)
|
|
|
|
| 48 |
model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64,
|
| 49 |
num_block=23, num_grow_ch=32, scale=2)
|
| 50 |
self.realesrganer = RealESRGANer(
|
| 51 |
+
scale=2,
|
| 52 |
model_path=realesrgan_model_path,
|
| 53 |
model=model,
|
| 54 |
tile=0,
|
|
|
|
| 107 |
# تكبير Real-ESRGAN لو موجود
|
| 108 |
if self.realesrganer:
|
| 109 |
try:
|
| 110 |
+
blended, _ = self.realesrganer.enhance(blended, outscale=2)
|
| 111 |
except Exception as e:
|
| 112 |
print(f"⚠️ RealESRGAN فشل: {e}")
|
| 113 |
|