Spaces:
Running
Running
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,
|