sanket kheni
commited on
Commit
·
1dc64a7
1
Parent(s):
26c05fc
app.py
CHANGED
@@ -22,12 +22,12 @@ retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/Reti
|
|
22 |
|
23 |
from retina_model import *
|
24 |
|
25 |
-
RetinaFace = load_model("retina_model/RetinaFace-Res50.h5"
|
26 |
-
custom_objects={"FPN": FPN,
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
)
|
32 |
|
33 |
arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFace")
|
|
|
22 |
|
23 |
from retina_model import *
|
24 |
|
25 |
+
RetinaFace = load_model("retina_model/RetinaFace-Res50.h5"
|
26 |
+
# custom_objects={"FPN": FPN,
|
27 |
+
# "SSH": SSH,
|
28 |
+
# "BboxHead": BboxHead,
|
29 |
+
# "LandmarkHead": LandmarkHead,
|
30 |
+
# "ClassHead": ClassHead}
|
31 |
)
|
32 |
|
33 |
arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFace")
|