felixrosberg
commited on
Commit
β’
ea104c0
1
Parent(s):
e921287
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,6 @@ import cv2
|
|
10 |
from scipy.ndimage import gaussian_filter
|
11 |
|
12 |
from tensorflow.keras.models import load_model
|
13 |
-
from retinaface.models import *
|
14 |
from options.swap_options import SwapOptions
|
15 |
|
16 |
# Invalidated!
|
@@ -21,6 +20,8 @@ opt = SwapOptions().parse()
|
|
21 |
|
22 |
retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/retinaface_resnet50",
|
23 |
private=True, use_auth_token=token, git_user="felixrosberg")
|
|
|
|
|
24 |
RetinaFace = load_model("retina_model/retinaface_res50.h5",
|
25 |
custom_objects={"FPN": FPN,
|
26 |
"SSH": SSH,
|
|
|
10 |
from scipy.ndimage import gaussian_filter
|
11 |
|
12 |
from tensorflow.keras.models import load_model
|
|
|
13 |
from options.swap_options import SwapOptions
|
14 |
|
15 |
# Invalidated!
|
|
|
20 |
|
21 |
retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/retinaface_resnet50",
|
22 |
private=True, use_auth_token=token, git_user="felixrosberg")
|
23 |
+
|
24 |
+
from retina_model.models import *
|
25 |
RetinaFace = load_model("retina_model/retinaface_res50.h5",
|
26 |
custom_objects={"FPN": FPN,
|
27 |
"SSH": SSH,
|