Charbel Malo
commited on
Update face_swapper.py
Browse files- 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 |
|