add min_confidence

Author: robmarkcole

deepstack/core.py

@@ -10,6 +10,7 @@
 DEFAULT_TIMEOUT = 10  # seconds
 DEFAULT_IP = "localhost"
 DEFAULT_PORT = 80
+DEFAULT_MIN_CONFIDENCE = 0.45
 
 ## HTTP codes
 HTTP_OK = 200
@@ -136,17 +137,18 @@ def __init__(
         port: int = DEFAULT_PORT,
         api_key: str = DEFAULT_API_KEY,
         timeout: int = DEFAULT_TIMEOUT,
+        min_confidence: float = DEFAULT_MIN_CONFIDENCE,
         url_detect: str = "",
         url_recognize: str = "",
         url_register: str = "",
     ):
-
+        self._api_key = api_key
+        self._timeout = timeout
+        self._min_confidence = min_confidence
         self._url_base = URL_BASE_VISION.format(ip=ip, port=port)
         self._url_detect = self._url_base + url_detect
         self._url_recognize = self._url_base + url_recognize
         self._url_register = self._url_base + url_register
-        self._api_key = api_key
-        self._timeout = timeout
 
     def detect(self):
         """Process image_bytes and detect."""
@@ -170,14 +172,20 @@ def __init__(
         port: int = DEFAULT_PORT,
         api_key: str = DEFAULT_API_KEY,
         timeout: int = DEFAULT_TIMEOUT,
+        min_confidence: float = DEFAULT_MIN_CONFIDENCE,
         custom_model: str = "",
     ):
         if custom_model:
             url_detect = URL_CUSTOM.format(custom_model=custom_model)
         else:
             url_detect = URL_OBJECT_DETECTION
         super().__init__(
-            ip=ip, port=port, api_key=api_key, timeout=timeout, url_detect=url_detect,
+            ip=ip,
+            port=port,
+            api_key=api_key,
+            timeout=timeout,
+            min_confidence=min_confidence,
+            url_detect=url_detect,
         )
 
     def detect(self, image_bytes: bytes):
@@ -200,12 +208,14 @@ def __init__(
         port: int = DEFAULT_PORT,
         api_key: str = DEFAULT_API_KEY,
         timeout: int = DEFAULT_TIMEOUT,
+        min_confidence: float = DEFAULT_MIN_CONFIDENCE,
     ):
         super().__init__(
             ip=ip,
             port=port,
             api_key=api_key,
             timeout=timeout,
+            min_confidence=min_confidence,
             url_recognize=URL_SCENE_RECOGNIZE,
         )
 
@@ -230,12 +240,14 @@ def __init__(
         port: int = DEFAULT_PORT,
         api_key: str = DEFAULT_API_KEY,
         timeout: int = DEFAULT_TIMEOUT,
+        min_confidence: float = DEFAULT_MIN_CONFIDENCE,
     ):
         super().__init__(
             ip=ip,
             port=port,
             api_key=api_key,
             timeout=timeout,
+            min_confidence=min_confidence,
             url_detect=URL_FACE_DETECTION,
             url_register=URL_FACE_REGISTER,
             url_recognize=URL_FACE_RECOGNIZE,

usage-object-detection.ipynb

@@ -355,6 +355,27 @@
     "predictions"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 16,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'a': 'a', 'b': 'b'}"
+      ]
+     },
+     "execution_count": 16,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "data = {'a':'a', 'b':'b'}\n",
+    "data"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,