not use cuda

ebf9d350 · 赵威 · b9d17bb4 · ebf9d350
Commit ebf9d350 authored Sep 09, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

images.py src/utils/images.py +2 -0

No files found.
--- a/src/utils/images.py
+++ b/src/utils/images.py
@@ -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")
 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_detector = dlib.get_frontal_face_detector()
 shape_predictor = dlib.shape_predictor(SHAPE_PATH)