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 |
|