Spaces:
Running
Running
Commit
·
7071546
1
Parent(s):
5ab6bb5
Updated utils
Browse files
utils.py
CHANGED
@@ -2,16 +2,19 @@ import cv2
|
|
2 |
from ultralight import UltraLightDetector
|
3 |
from ultralight.utils import draw_faces
|
4 |
import time
|
|
|
5 |
import asyncio
|
6 |
|
7 |
def load_caffe_models():
|
|
|
|
|
8 |
age_net = cv2.dnn.readNet(
|
9 |
-
|
10 |
-
|
11 |
)
|
12 |
gender_net = cv2.dnn.readNet(
|
13 |
-
|
14 |
-
|
15 |
)
|
16 |
|
17 |
return age_net, gender_net
|
|
|
2 |
from ultralight import UltraLightDetector
|
3 |
from ultralight.utils import draw_faces
|
4 |
import time
|
5 |
+
import os
|
6 |
import asyncio
|
7 |
|
8 |
def load_caffe_models():
|
9 |
+
models_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "opencv_models")
|
10 |
+
|
11 |
age_net = cv2.dnn.readNet(
|
12 |
+
os.path.join(models_dir, "age_net.caffemodel"),
|
13 |
+
os.path.join(models_dir, "age_deploy.prototxt")
|
14 |
)
|
15 |
gender_net = cv2.dnn.readNet(
|
16 |
+
os.path.join(models_dir, "gender_net.caffemodel"),
|
17 |
+
os.path.join(models_dir, "gender_deploy.prototxt")
|
18 |
)
|
19 |
|
20 |
return age_net, gender_net
|