KenjieDec commited on
Commit
30fe1c8
·
1 Parent(s): f14d73f

Update face_enhancement.py

Browse files
Files changed (1) hide show
  1. 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