feat(idle): remove code for model inference. Add support for sending requests to flask server

Author: eugenos-programos

.dockerignore

@@ -1,2 +1,3 @@
 .cache
-__pycache__
+__pycache__
+idle_models

confs/configs.json

@@ -2,9 +2,6 @@
     "classes": [
         67
     ],
-    "iou_thres": 0.5,
-    "conf_thres": 0.3,
-    "model_path": "idle_models/idle_v0.3.11.pt",
     "wait_time": 10,
     "threshold": 100
 }

confs/settings.env

@@ -1,5 +1,5 @@
 username = 'admin'
 password = 'just4Taqtile'
 camera_url = 'http://192.168.1.162/onvif-http/snapshot?Profile_1'
-server_url = 'http://192.168.1.999'
+server_url = 'http://127.0.0.1'
 folder = 'images/192.168.1.163'

functional.py

@@ -9,9 +9,7 @@
 import requests
 import httplib2
 import numpy as np
-from idle_models.IdleObjectDetectionModel import IdleObjectDetectionModel
 import time
-from dotenv import load_dotenv
 
 
 def create_logger():
@@ -34,9 +32,6 @@ def create_logger():
 
 def init_connection():
     password = os.environ.get("password")
-    if password is None:
-        load_dotenv("confs/settings.env")
-        password = os.environ.get("password")
     username = os.environ.get("username")
     try:
         h = httplib2.Http(".cache")
@@ -47,16 +42,6 @@ def init_connection():
     return None
 
 
-def init_model():
-    model = IdleObjectDetectionModel(
-        MODEL_PATH,
-        CONF_THRES,
-        IOU_THRES,
-        CLASSES
-    )
-    return model
-
-
 def get_frame(h):
     try:
         time_ = datetime.datetime.now()

main.py

@@ -1,16 +1,20 @@
 import time
 from functional import *
+from send_request import predict
+from dotenv import load_dotenv
 
 
+password = os.environ.get("password")
+if password is None:
+    load_dotenv("confs/settings.env")
 logger = create_logger()
 
 
 def run():
+    server_url = os.environ.get("server_url")
     while (h := init_connection()) is None:
         logger.warning("Cannot create connection")
         time.sleep(1)
-    model = init_model()
-    logger.info("Model is initialized")
     prev_preds = None
     iter_idx = 0
     while True:
@@ -24,8 +28,8 @@ def run():
             time.sleep(1)
             continue
         time.sleep(2)
-        preds, scores = model(img)
-        img = put_rectangle(img, preds.numpy(), scores.numpy())
+        preds, scores = predict(img, server_url)
+        img = put_rectangle(img, preds, scores)
         if len(scores) > 0:
             logger.info("Telephone is detected")
             if check_coordinates_diffs(prev_preds, preds):

requirements.txt

@@ -1,17 +1,10 @@
 colorlog==4.8.0
 httplib2==0.22.0
-matplotlib==3.6.3
 numpy==1.22.3
-onnx==1.13.1
 opencv_python==4.7.0.72
 pafy==0.5.5
-pandas==1.5.3
 Pillow==9.5.0
 pydantic==1.10.2
 python-dotenv==1.0.0
 PyYAML==6.0
 requests==2.27.1
-scipy==1.10.1
-seaborn==0.12.2
-tqdm==4.64.1
-ultralytics==8.0.112

send_request.py

@@ -0,0 +1,17 @@
+import requests
+import numpy as np
+
+
+PORT = 5000
+
+
+def predict(img: np.array, server_url: str):
+    response = requests.post(
+        f"{server_url}:{PORT}/predict",
+        json={
+            "image": img.tolist()
+        }
+    )
+    coordinates = np.array(response.json().get("coordinates"))
+    confidences = np.array(response.json().get("confidences"))
+    return [coordinates, confidences]