pythonwebcoder
diff --git a/‎face_recognition/api.py
Lines changed: 3 additions & 3 deletions b/‎face_recognition/api.py
Lines changed: 3 additions & 3 deletions
diff --git a/‎face_recognition/models/dlib_face_recognition_resnet_model_v1.dat
-21.4 MB b/‎face_recognition/models/dlib_face_recognition_resnet_model_v1.dat
-21.4 MB
diff --git a/‎face_recognition/models/shape_predictor_68_face_landmarks.dat
-95.1 MB b/‎face_recognition/models/shape_predictor_68_face_landmarks.dat
-95.1 MB
diff --git a/‎setup.py
Lines changed: 2 additions & 1 deletion b/‎setup.py
Lines changed: 2 additions & 1 deletion
diff --git a/‎tests/test_face_recognition.py
Lines changed: 1 addition & 1 deletion b/‎tests/test_face_recognition.py
Lines changed: 1 addition & 1 deletion
@@ -3,12 +3,12 @@
 import scipy.misc
 import dlib
 import numpy as np
-from pkg_resources import resource_filename
+import face_recognition_models
 
-predictor_model = resource_filename(__name__, "models/shape_predictor_68_face_landmarks.dat")
+predictor_model = face_recognition_models.pose_predictor_model_location()
 pose_predictor = dlib.shape_predictor(predictor_model)
 
-face_recognition_model = resource_filename(__name__, "models/dlib_face_recognition_resnet_model_v1.dat")
+face_recognition_model = face_recognition_models.face_recognition_model_location()
 face_encoder = dlib.face_recognition_model_v1(face_recognition_model)
 
 
 
@@ -10,6 +10,7 @@
     history = history_file.read()
 
 requirements = [
+    'face_recognition_models',
     'Click>=6.0',
     'dlib>=19.3.0',
     'numpy',
@@ -22,7 +23,7 @@
 
 setup(
     name='face_recognition',
-    version='0.1.2',
+    version='0.1.3',
     description="Recognize faces from Python or from the command line",
     long_description=readme + '\n\n' + history,
     author="Adam Geitgey",
 
@@ -52,7 +52,7 @@ def test_face_landmarks(self):
         img = api.load_image_file(os.path.join(os.path.dirname(__file__), "test_images", "obama.jpg"))
         face_landmarks = api.face_landmarks(img)
 
-        assert list(face_landmarks[0].keys()) == ['chin', 'left_eyebrow', 'right_eyebrow', 'nose_bridge', 'nose_tip', 'left_eye', 'right_eye', 'top_lip', 'bottom_lip']
+        assert set(face_landmarks[0].keys()) == set(['chin', 'left_eyebrow', 'right_eyebrow', 'nose_bridge', 'nose_tip', 'left_eye', 'right_eye', 'top_lip', 'bottom_lip'])
         assert face_landmarks[0]['chin'] == [(369, 220), (372, 254), (378, 289), (384, 322), (395, 353), (414, 382), (437, 407), (464, 424), (495, 428), (527, 420), (552, 399), (576, 372), (594, 344), (604, 314)]
 
     def test_face_encodings(self):