Merge pull request #11 from 5sControl/dev

Dev

Author: eugenos-programos

main.py

@@ -39,8 +39,11 @@
         continue
     if preds.size != 0 and not np.any(preds == 1.):
         logger.info("Telephone is detected")
-        if utils.are_bboxes_equal(prev_preds, preds, configs["threshold"]):
+        if utils.bboxes_not_equal(prev_preds, preds, configs["threshold"]):
+            utils.save_cropped_bbox(img, preds[:, :4])
             img = utils.put_rectangle(img, preds[:, :4], preds[:, 4])
             reporter.send_report(reporter.create_report(img, str(start_tracking)))
+        else:
+            logger.debug("Equal bboxes")
         prev_preds = preds
     time.sleep(2)

utils/__init__.py

@@ -1,2 +1,2 @@
-from .functional import put_rectangle, are_bboxes_equal
+from .functional import put_rectangle, bboxes_not_equal, save_cropped_bbox
 from .logger import create_logger

utils/functional.py

@@ -1,6 +1,8 @@
 import cv2
 import numpy as np
 import numba
+import os
+import uuid
 
 
 def put_rectangle(img: cv2.Mat, boxes: list, scores: list) -> np.array:
@@ -12,7 +14,14 @@ def put_rectangle(img: cv2.Mat, boxes: list, scores: list) -> np.array:
 
 
 @numba.njit(numba.boolean(numba.float32[:, :], numba.float32[:, :], numba.float32), parallel=True)
-def are_bboxes_equal(coords_1: np.array, coords_2: np.array, threshold: float) -> bool:
+def bboxes_not_equal(coords_1: np.array, coords_2: np.array, threshold: float) -> bool:
     if coords_1 is None or coords_1.shape != coords_2.shape:
         return True
     return np.abs(coords_1 - coords_2).sum() > threshold
+
+def save_cropped_bbox(img: np.array, bboxes: np.array):
+    os.makedirs("images/debug/", exist_ok=True)
+    for bbox in bboxes:
+        x1, y1, x2, y2 = bbox.astype(int)
+        c = 15
+        cv2.imwrite(f"images/debug/{uuid.uuid4()}.png", img[x1 - c:x2 + c, y1 - c:y2 + c])