Skip to content

S
strategy_embedding
Commit ebf9d350 authored by 赵威's avatar 赵威
Browse files
Options

not use cuda

parent b9d17bb4
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/utils/images.py
View file @ ebf9d350
...@@ -15,6 +15,8 @@ MODELS_DIR = os.path.join(BASE_DIR, "_models") ...@@ -15,6 +15,8 @@ MODELS_DIR = os.path.join(BASE_DIR, "_models")
FACEREC_PATH = os.path.join(MODELS_DIR, "dlib_face_recognition_resnet_model_v1.dat") FACEREC_PATH = os.path.join(MODELS_DIR, "dlib_face_recognition_resnet_model_v1.dat")
SHAPE_PATH = os.path.join(MODELS_DIR, "shape_predictor_68_face_landmarks.dat") SHAPE_PATH = os.path.join(MODELS_DIR, "shape_predictor_68_face_landmarks.dat")
dlib.DLIB_USE_CUDA = False
face_rec = dlib.face_recognition_model_v1(FACEREC_PATH) face_rec = dlib.face_recognition_model_v1(FACEREC_PATH)
face_detector = dlib.get_frontal_face_detector() face_detector = dlib.get_frontal_face_detector()
shape_predictor = dlib.shape_predictor(SHAPE_PATH) shape_predictor = dlib.shape_predictor(SHAPE_PATH)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment