Spaces:
Running
Running
Update face_enhancement.py
Browse files- face_enhancement.py +1 -0
face_enhancement.py
CHANGED
@@ -19,6 +19,7 @@ class FaceEnhancement(object):
|
|
19 |
self.facedetector = RetinaFaceDetection(base_dir, device)
|
20 |
self.facegan = FaceGAN(base_dir, size, out_size, model, channel_multiplier, narrow, key, device=device)
|
21 |
self.srmodel = RealESRNet(base_dir, 'realesrnet', 2, 0, device=device)
|
|
|
22 |
self.size = size
|
23 |
self.out_size = size if out_size==None else out_size
|
24 |
self.threshold = 0.9
|
|
|
19 |
self.facedetector = RetinaFaceDetection(base_dir, device)
|
20 |
self.facegan = FaceGAN(base_dir, size, out_size, model, channel_multiplier, narrow, key, device=device)
|
21 |
self.srmodel = RealESRNet(base_dir, 'realesrnet', 2, 0, device=device)
|
22 |
+
self.use_sr = True
|
23 |
self.size = size
|
24 |
self.out_size = size if out_size==None else out_size
|
25 |
self.threshold = 0.9
|