Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -22,6 +22,7 @@ def count_unique_trucks(video_path):
|
|
22 |
|
23 |
unique_truck_ids = set()
|
24 |
frame_skip = 5 # Process every 5th frame for efficiency
|
|
|
25 |
|
26 |
frame_count = 0
|
27 |
while True:
|
@@ -53,7 +54,15 @@ def count_unique_trucks(video_path):
|
|
53 |
|
54 |
for obj in tracked_objects:
|
55 |
truck_id = int(obj[4]) # Unique ID assigned by SORT
|
56 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
57 |
|
58 |
cap.release()
|
59 |
|
|
|
22 |
|
23 |
unique_truck_ids = set()
|
24 |
frame_skip = 5 # Process every 5th frame for efficiency
|
25 |
+
truck_tracker_history = {} # To store truck ID with frame count
|
26 |
|
27 |
frame_count = 0
|
28 |
while True:
|
|
|
54 |
|
55 |
for obj in tracked_objects:
|
56 |
truck_id = int(obj[4]) # Unique ID assigned by SORT
|
57 |
+
|
58 |
+
# Only add truck ID if it appears in multiple frames (avoid one-frame detections)
|
59 |
+
if truck_id not in truck_tracker_history:
|
60 |
+
truck_tracker_history[truck_id] = frame_count
|
61 |
+
else:
|
62 |
+
# If truck ID appears in multiple frames, count as unique
|
63 |
+
if frame_count - truck_tracker_history[truck_id] > frame_skip:
|
64 |
+
unique_truck_ids.add(truck_id)
|
65 |
+
truck_tracker_history[truck_id] = frame_count
|
66 |
|
67 |
cap.release()
|
68 |
|