From 865b52beec57d8b9d8b60716a83724026cb77b7a Mon Sep 17 00:00:00 2001
From: kev <vimagick@gmail.com>
Date: Mon, 14 Feb 2022 17:52:54 +0800
Subject: [PATCH] update deepstack

---
 deepstack/README.md          | 28 +++++++++++++++++++++++++++-
 deepstack/docker-compose.yml |  6 +++++-
 2 files changed, 32 insertions(+), 2 deletions(-)

diff --git a/deepstack/README.md b/deepstack/README.md
index 8e46644..8a97831 100644
--- a/deepstack/README.md
+++ b/deepstack/README.md
@@ -9,8 +9,34 @@ custom models on multiple edge devices locally or on your private cloud.
 ## up and running
 
 ```bash
+$ mkdir -p data/{datastore,modelstore}
+$ wget -P data/modelstore \
+          https://github.com/odd86/deepstack_licenceplate_model/raw/main/models/licence-plate.pt \
+          https://github.com/OlafenwaMoses/DeepStack_OpenLogo/releases/download/v1/openlogo.pt \
+          https://github.com/OlafenwaMoses/DeepStack_ExDark/releases/download/v1/dark.pt \
+          https://github.com/OlafenwaMoses/DeepStack_ActionNET/releases/download/v2/actionnetv2.pt \
+          https://github.com/MikeLud/DeepStack-Security-Camera-Models/raw/main/Models/animal.pt \
+          https://github.com/avatar42/RMRR.fire/raw/main/fire.pt \
+          https://github.com/avatar42/RMRR.birds/raw/main/birds.pt
 $ docker-compose up -d
-$ http --form http://127.0.0.1:5000/v1/vision/face/ image@photo.jpg
+
+# Face Detection
+$ http --form http://127.0.0.1:5000/v1/vision/face image@photo.jpg
+{"success": true, "predictions": [...]}
+
+# Face Recognition
+$ http --form http://127.0.0.1:5000/v1/vision/face/register userid=kevin image1@photo1.jpg image2@photo2.jpg
+{"success": true, "message": "face added"}
+$ http --form http://127.0.0.1:5000/v1/vision/face/recognize image@photo.jpg
+{"success": true, "predictions": [...]}
+
+# Object Detection
+$ http --form http://127.0.0.1:5000/v1/vision/detection image@photo.jpg
+{"success": true, "predictions": [...]}
+
+# Custom Models
+$ http --form http://127.0.0.1:5000/v1/vision/custom/animal image@photo.jpg
+{"success": true, "predictions": [...]}
 ```
 
 [1]: https://deepstack.cc/
diff --git a/deepstack/docker-compose.yml b/deepstack/docker-compose.yml
index ffed210..1b60ba9 100644
--- a/deepstack/docker-compose.yml
+++ b/deepstack/docker-compose.yml
@@ -10,6 +10,8 @@ services:
       - DEEPSTACK_IP=deepstack
       - DEEPSTACK_PORT=5000
       - DEEPSTACK_TIMEOUT=20
+      # API-KEY=******
+      # DEEPSTACK_CUSTOM_MODEL=xxx
     depends_on:
       - deepstack
     restart: unless-stopped
@@ -19,10 +21,12 @@ services:
     ports:
       - "5000:5000"
     volumes:
-      - ./data:/datastore
+      - ./data/datastore:/datastore
+      - ./data/modelstore:/modelstore/detection
     environment:
       - MODE=Medium
       - VISION-DETECTION=True
       - VISION-FACE=True
       - VISION-SCENE=True
+      # DEEPSTACK_API_KEY=******
     restart: unless-stopped