Charbel Malo commited on
Commit
fe49147
·
verified ·
1 Parent(s): e39a0a8

Update face_swapper.py

Browse files
Files changed (1) hide show
  1. face_swapper.py +2 -1
face_swapper.py CHANGED
@@ -18,6 +18,7 @@ arcface_dst = np.array(
18
  [41.5493, 92.3655], [70.7299, 92.2041]],
19
  dtype=np.float32)
20
 
 
21
  def estimate_norm(lmk, image_size=112, mode='arcface'):
22
  assert lmk.shape == (5, 2)
23
  assert image_size % 112 == 0 or image_size % 128 == 0
@@ -42,7 +43,7 @@ def norm_crop2(img, landmark, image_size=112, mode='arcface'):
42
 
43
 
44
  class Inswapper():
45
- def __init__(self, model_file=None, batch_size=32, providers=['CUDAExecutionProvider']):
46
  self.model_file = model_file
47
  self.batch_size = batch_size
48
 
 
18
  [41.5493, 92.3655], [70.7299, 92.2041]],
19
  dtype=np.float32)
20
 
21
+
22
  def estimate_norm(lmk, image_size=112, mode='arcface'):
23
  assert lmk.shape == (5, 2)
24
  assert image_size % 112 == 0 or image_size % 128 == 0
 
43
 
44
 
45
  class Inswapper():
46
+ def __init__(self, model_file=None, batch_size=32, providers=['CUDAExecutionProvider,CPUExecutionProvider']):
47
  self.model_file = model_file
48
  self.batch_size = batch_size
49