Adds usage

Author: robmarkcole

deepstack/__pycache__/__init__.cpython-36.pyc

@@ -0,0 +1,100 @@
+"""
+Deepstack cpre.
+"""
+import imghdr
+import requests
+from PIL import Image
+
+## Const
+CLASSIFIER = 'deepstack'
+HTTP_OK = 200
+HTTP_BAD_REQUEST = 400
+HTTP_UNAUTHORIZED = 401
+TIMEOUT = 30  # seconds
+
+
+def get_matched_faces(predictions: dict):
+    """
+    Get the predicted faces and their confidence.
+    """
+    try:
+        matched_faces = {face['userid']: round(face['confidence']*100, 1)
+                         for face in predictions if not face['userid'] == 'unknown'}
+        return matched_faces
+    except:
+        return {}
+
+
+def is_valid_image(file_path: str):
+    """
+    Check file_path is valid image, using PIL then imghdr.
+    """
+    try:
+        with Image.open(file_path):
+            pass
+
+        image_extension = imghdr.what(file_path)
+        if image_extension in ['jpeg', '.jpg', '.png']:
+            return True
+        return False
+    except Exception as exc:
+        print(exc)
+        return False
+
+
+def post_image(url: str, image: bytes):
+    """Post an image to the classifier."""
+    try:
+        response = requests.post(
+            url,
+            files={"image": image},
+            timeout=TIMEOUT
+        )
+        return response
+    except requests.exceptions.ConnectionError:
+        print("ConnectionError: Is %s running?", CLASSIFIER)
+        return None
+    except requests.exceptions.Timeout:
+        print("Timeout error from %s", CLASSIFIER)
+        return None
+
+
+class DeepstackFace():
+    """Work with faces."""
+
+    def __init__(self, ip_address: str, port: str):
+
+        self._url_check = "http://{}:{}/v1/vision/face/recognize".format(
+            ip_address, port)
+
+        self._faces = None
+        self._matched = {}
+
+    def process_file(self, file_path: str):
+        """Process an image file."""
+        if is_valid_image(file_path):
+            with open(file_path, "rb") as image_bytes:
+                self.process_image_bytes(image_bytes)
+
+    def process_image_bytes(self, image_bytes: bytes):
+        """Process an image."""
+        response = post_image(
+            self._url_check, image_bytes)
+        if response:
+            if response.status_code == HTTP_OK:
+                predictions_json = response.json()["predictions"]
+                self._faces = len(predictions_json)
+                self._matched = get_matched_faces(predictions_json)
+
+        else:
+            self._faces = None
+            self._matched = {}
+
+    @property
+    def attributes(self):
+        """Return the classifier attributes."""
+        return {
+            'faces': self._faces,
+            'matched_faces': self._matched,
+            'total_matched_faces': len(self._matched),
+        }

deepstack/__pycache__/core.cpython-36.pyc

@@ -113,7 +113,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 13,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -131,7 +131,7 @@
     "    def process_file(self, file_path : str):\n",
     "        \"\"\"Process an image file.\"\"\"\n",
     "        if is_valid_image(file_path):\n",
-    "            with open(image_path, \"rb\") as image_bytes:\n",
+    "            with open(file_path, \"rb\") as image_bytes:\n",
     "                self.process_image_bytes(image_bytes)\n",
     "        \n",
     "    def process_image_bytes(self, image_bytes : bytes):\n",
@@ -167,7 +167,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 14,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -176,7 +176,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 15,
    "metadata": {},
    "outputs": [
     {
@@ -185,7 +185,7 @@
        "{'faces': None, 'matched_faces': {}, 'total_matched_faces': 0}"
       ]
      },
-     "execution_count": 8,
+     "execution_count": 15,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -196,16 +196,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 16,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "{'faces': 2, 'matched_faces': {}, 'total_matched_faces': 0}\n",
-      "CPU times: user 368 ms, sys: 95.9 ms, total: 464 ms\n",
-      "Wall time: 17.3 s\n"
+      "Timeout error from %s deepstack_face\n",
+      "{'faces': None, 'matched_faces': {}, 'total_matched_faces': 0}\n",
+      "CPU times: user 393 ms, sys: 91.4 ms, total: 485 ms\n",
+      "Wall time: 20.6 s\n"
      ]
     },
     {