hb-setosys commited on
Commit
4e5fb8f
·
verified ·
1 Parent(s): c72637a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
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
- unique_truck_ids.add(truck_id)
 
 
 
 
 
 
 
 
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