From a18808374ba3de0f19a76f9a8ef8ddb44574de3a Mon Sep 17 00:00:00 2001
From: Andrey Pohilko <apc4@ya.ru>
Date: Wed, 1 Aug 2018 10:04:10 +0300
Subject: [PATCH] Proper example folder

---
 .../{face_tracker.py => tracker/__init__.py}   | 18 ++++++++----------
 examples/tracker/color.json                    |  4 ++++
 2 files changed, 12 insertions(+), 10 deletions(-)
 rename examples/{face_tracker.py => tracker/__init__.py} (91%)
 create mode 100644 examples/tracker/color.json

diff --git a/examples/face_tracker.py b/examples/tracker/__init__.py
similarity index 91%
rename from examples/face_tracker.py
rename to examples/tracker/__init__.py
index df34b2f..32b23a3 100644
--- a/examples/face_tracker.py
+++ b/examples/tracker/__init__.py
@@ -1,5 +1,6 @@
 import json
 import logging
+import os
 import time
 import traceback
 from threading import Thread
@@ -42,7 +43,7 @@ class FaceTracker(MoveHub):
         bodies, rejects, weights = self.face_cascade.detectMultiScale3(self.cur_img, 1.5, 5, outputRejectLevels=True)
         return bodies, weights
 
-    def _find_heli(self):
+    def _find_color(self):
         # from https://www.pyimagesearch.com/2015/09/14/ball-tracking-with-opencv/
         # and https://thecodacus.com/opencv-object-tracking-colour-detection-python/#.W2DHFd_IQsM
 
@@ -50,13 +51,9 @@ class FaceTracker(MoveHub):
         hsv = cv2.cvtColor(blurred, cv2.COLOR_BGR2HSV)
 
         try:
-            with open("/tmp/1.json") as fhd:
+            # having color values in file allows finding values easier
+            with open(os.path.join(os.path.dirname(__file__), "color.json")) as fhd:
                 data = json.loads(fhd.read())
-                level = int(time.time()) % 51
-                level *= 5
-                logging.info("%s", level)
-                # data[0][0] = level
-                # data[1][0] = level + 5
                 lower = tuple(data[0])
                 upper = tuple(data[1])
         except:
@@ -67,8 +64,8 @@ class FaceTracker(MoveHub):
         mask = cv2.erode(mask, None, iterations=2)
         mask = cv2.dilate(mask, None, iterations=2)
 
-        if not (int(time.time()) % 2):
-            self.cur_img = mask
+        # if not (int(time.time()) % 2):
+        #    self.cur_img = mask
 
         cnts = cv2.findContours(mask.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
         cnts = cnts[0] if imutils.is_cv2() else cnts[1]
@@ -109,7 +106,8 @@ class FaceTracker(MoveHub):
 
         cur_face = (0, 0, 0, 0)
         while thr.isAlive():
-            bodies, weights = self._find_faces()
+            bodies, weights = self._find_color()
+            #bodies, weights = self._find_faces()
 
             if len(bodies):
                 logging.debug("Bodies: %s / Weights: %s", bodies, weights)
diff --git a/examples/tracker/color.json b/examples/tracker/color.json
new file mode 100644
index 0000000..3784089
--- /dev/null
+++ b/examples/tracker/color.json
@@ -0,0 +1,4 @@
+[
+  [95, 64, 128],
+  [100, 255, 255]
+]
\ No newline at end of file