Fix bug

models/.gitignore

@@ -1,2 +1,3 @@
 **.tflite
-!.gitignore
+!.gitignore
+!yolov5n6-fp16.tflite

src/detect.py

@@ -55,7 +55,7 @@ def preprocess(self, img: np.ndarray) -> np.ndarray:
         img = np.expand_dims(img, axis=0)
         return img.astype(np.float32)
 
-    def postprocess(self, output_data, box_type: str):
+    def postprocess(self, output_data, box_type: str) -> Tuple[np.ndarray]:
         """Postprocess."""
         output_data = output_data[0]
         # xywh
@@ -80,6 +80,7 @@ def postprocess(self, output_data, box_type: str):
 
     def to_xyxy(self, boxes: np.ndarray) -> np.ndarray:
         """Covert xywh to xyxy."""
+        # (x, y) is cordinate fo the center of the box.
         x, y, w, h = boxes[..., 0], boxes[..., 1], boxes[..., 2], boxes[..., 3]
         boxes = np.array([x - w / 2, y - h / 2, x + w / 2, y + h / 2])
         # [4, n] -> [n, 4]

src/main.py

@@ -97,6 +97,7 @@ def main(
         end = time.time()
 
         # Update tracker and draw bounding boxes in frame.
+        # dets:  [xmin, ymin, xmax, ymax, score]
         frame = tracker.update(frame, dets)
 
         # Executed only first time.

src/tracker.py

@@ -30,7 +30,6 @@ def __init__(
         self.border = border
         self.count_callback = count_callback
         self.memory = {}
-        self.counter = 0
         self.counter = {key: 0 for key in directions.keys()}
         self.directions = directions
 

src/utils.py

@@ -2,6 +2,7 @@
 #
 # Copyright 2021.
 # ozora-ogino
+
 from typing import List, Tuple
 
 # Direction configuration.
@@ -36,7 +37,7 @@ def check_direction(
     # True if the direction is right.
     direction_x = (current_center[0] - prev_center[0]) > 0
     # True if the direction is top.
-    direction_y = (current_center[1] - prev_center[1]) > 0
+    direction_y = (current_center[1] - prev_center[1]) < 0
 
     x_is_true = direction[0] is None or direction_x is direction[0]
     y_is_true = direction[1] is None or direction_y is direction[1]