intuitive262 commited on
Commit
7071546
·
1 Parent(s): 5ab6bb5

Updated utils

Browse files
Files changed (1) hide show
  1. utils.py +7 -4
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
- r".\opencv_models\age_net.caffemodel",
10
- r".\opencv_models\age_deploy.prototxt"
11
  )
12
  gender_net = cv2.dnn.readNet(
13
- r".\opencv_models\gender_net.caffemodel",
14
- r".\opencv_models\gender_deploy.prototxt"
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