Update app.py

app.py

@@ -22,7 +22,8 @@ def count_unique_trucks(video_path):
 
     unique_truck_ids = set()
     frame_skip = 5  # Process every 5th frame for efficiency
-    truck_tracker_history = {}  # To store truck ID with frame count
+    truck_tracker_history = {}  # To store truck ID with frame count and position
+    min_distance_threshold = 50  # Minimum distance between truck positions to be counted as different
 
     frame_count = 0
     while True:
@@ -54,15 +55,32 @@ def count_unique_trucks(video_path):
 
             for obj in tracked_objects:
                 truck_id = int(obj[4])  # Unique ID assigned by SORT
+                x1, y1, x2, y2 = obj[:4]  # Get the bounding box coordinates
 
-                # Only add truck ID if it appears in multiple frames (avoid one-frame detections)
+                # Compute the center of the truck for better tracking
+                truck_center = ((x1 + x2) / 2, (y1 + y2) / 2)
+
+                # Only add truck ID if it's sufficiently far from previously tracked trucks
                 if truck_id not in truck_tracker_history:
-                    truck_tracker_history[truck_id] = frame_count
+                    truck_tracker_history[truck_id] = {
+                        "frame_count": frame_count,
+                        "position": truck_center
+                    }
+                    unique_truck_ids.add(truck_id)  # Add the truck as a unique truck
                 else:
-                    # If truck ID appears in multiple frames, count as unique
-                    if frame_count - truck_tracker_history[truck_id] > frame_skip:
+                    last_truck = truck_tracker_history[truck_id]
+                    last_position = last_truck["position"]
+
+                    # Calculate the distance between the current and last positions
+                    distance = np.linalg.norm(np.array(truck_center) - np.array(last_position))
+
+                    # If the distance between truck positions is greater than the threshold, it's a new truck detection
+                    if distance > min_distance_threshold:
                         unique_truck_ids.add(truck_id)
-                        truck_tracker_history[truck_id] = frame_count
+                        truck_tracker_history[truck_id] = {
+                            "frame_count": frame_count,
+                            "position": truck_center
+                        }
 
     cap.release()