Commit
Β·
c46bfbb
1
Parent(s):
9adeced
Update app.py
Browse files
app.py
CHANGED
@@ -18,8 +18,7 @@ token = os.environ['model_fetch']
|
|
18 |
|
19 |
opt = SwapOptions().parse()
|
20 |
|
21 |
-
retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/retinaface_resnet50",
|
22 |
-
private=True, use_auth_token=token)
|
23 |
|
24 |
from retina_model.models import *
|
25 |
|
@@ -30,25 +29,21 @@ RetinaFace = load_model("retina_model/retinaface_res50.h5",
|
|
30 |
"LandmarkHead": LandmarkHead,
|
31 |
"ClassHead": ClassHead})
|
32 |
|
33 |
-
arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/arcface_tf",
|
34 |
-
private=True, use_auth_token=token)
|
35 |
ArcFace = load_model("arcface_model/arc_res50.h5")
|
36 |
ArcFaceE = load_model("arcface_model/arc_res50e.h5")
|
37 |
|
38 |
-
g_repo = Repository(local_dir="g_model_c_hq", clone_from="felixrosberg/affa_config_c_hq",
|
39 |
-
private=True, use_auth_token=token)
|
40 |
G = load_model("g_model_c_hq/generator_t_28.h5", custom_objects={"AdaIN": AdaIN,
|
41 |
"AdaptiveAttention": AdaptiveAttention,
|
42 |
"InstanceNormalization": InstanceNormalization})
|
43 |
|
44 |
-
r_repo = Repository(local_dir="reconstruction_attack", clone_from="felixrosberg/reconstruction_attack",
|
45 |
-
private=True, use_auth_token=token)
|
46 |
R = load_model("reconstruction_attack/reconstructor_42.h5", custom_objects={"AdaIN": AdaIN,
|
47 |
"AdaptiveAttention": AdaptiveAttention,
|
48 |
"InstanceNormalization": InstanceNormalization})
|
49 |
|
50 |
-
permuter_repo = Repository(local_dir="identity_permuter", clone_from="felixrosberg/identitypermuter",
|
51 |
-
private=True, use_auth_token=token, git_user="felixrosberg")
|
52 |
|
53 |
from identity_permuter.id_permuter import identity_permuter
|
54 |
|
|
|
18 |
|
19 |
opt = SwapOptions().parse()
|
20 |
|
21 |
+
retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/retinaface_resnet50", use_auth_token=token)
|
|
|
22 |
|
23 |
from retina_model.models import *
|
24 |
|
|
|
29 |
"LandmarkHead": LandmarkHead,
|
30 |
"ClassHead": ClassHead})
|
31 |
|
32 |
+
arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/arcface_tf", use_auth_token=token)
|
|
|
33 |
ArcFace = load_model("arcface_model/arc_res50.h5")
|
34 |
ArcFaceE = load_model("arcface_model/arc_res50e.h5")
|
35 |
|
36 |
+
g_repo = Repository(local_dir="g_model_c_hq", clone_from="felixrosberg/affa_config_c_hq", use_auth_token=token)
|
|
|
37 |
G = load_model("g_model_c_hq/generator_t_28.h5", custom_objects={"AdaIN": AdaIN,
|
38 |
"AdaptiveAttention": AdaptiveAttention,
|
39 |
"InstanceNormalization": InstanceNormalization})
|
40 |
|
41 |
+
r_repo = Repository(local_dir="reconstruction_attack", clone_from="felixrosberg/reconstruction_attack", use_auth_token=token)
|
|
|
42 |
R = load_model("reconstruction_attack/reconstructor_42.h5", custom_objects={"AdaIN": AdaIN,
|
43 |
"AdaptiveAttention": AdaptiveAttention,
|
44 |
"InstanceNormalization": InstanceNormalization})
|
45 |
|
46 |
+
permuter_repo = Repository(local_dir="identity_permuter", clone_from="felixrosberg/identitypermuter", use_auth_token=token, git_user="felixrosberg")
|
|
|
47 |
|
48 |
from identity_permuter.id_permuter import identity_permuter
|
49 |
|