Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -44,6 +44,7 @@ def count_unique_trucks(video_path):
|
|
44 |
if not cap.isOpened():
|
45 |
return {"Error": "Unable to open video file."}
|
46 |
|
|
|
47 |
unique_truck_ids = set()
|
48 |
truck_history = {}
|
49 |
|
@@ -64,6 +65,9 @@ def count_unique_trucks(video_path):
|
|
64 |
|
65 |
frame_count = 0
|
66 |
|
|
|
|
|
|
|
67 |
while True:
|
68 |
ret, frame = cap.read()
|
69 |
if not ret:
|
@@ -124,6 +128,7 @@ def count_unique_trucks(video_path):
|
|
124 |
cap.release()
|
125 |
return {"Total Unique Trucks": len(unique_truck_ids)}
|
126 |
|
|
|
127 |
# Gradio UI function
|
128 |
def analyze_video(video_file):
|
129 |
result = count_unique_trucks(video_file)
|
|
|
44 |
if not cap.isOpened():
|
45 |
return {"Error": "Unable to open video file."}
|
46 |
|
47 |
+
# Reset variables at the start of each analysis
|
48 |
unique_truck_ids = set()
|
49 |
truck_history = {}
|
50 |
|
|
|
65 |
|
66 |
frame_count = 0
|
67 |
|
68 |
+
# Reinitialize the tracker to clear any previous state
|
69 |
+
tracker = Sort()
|
70 |
+
|
71 |
while True:
|
72 |
ret, frame = cap.read()
|
73 |
if not ret:
|
|
|
128 |
cap.release()
|
129 |
return {"Total Unique Trucks": len(unique_truck_ids)}
|
130 |
|
131 |
+
|
132 |
# Gradio UI function
|
133 |
def analyze_video(video_file):
|
134 |
result = count_unique_trucks(video_file)
|