Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,20 @@ def count_unique_trucks(video_path):
|
|
30 |
|
31 |
# Get FPS of the video
|
32 |
fps = int(cap.get(cv2.CAP_PROP_FPS))
|
33 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
34 |
|
35 |
frame_count = 0
|
36 |
|
|
|
30 |
|
31 |
# Get FPS of the video
|
32 |
fps = int(cap.get(cv2.CAP_PROP_FPS))
|
33 |
+
|
34 |
+
# Extract filename from the path
|
35 |
+
video_filename = os.path.basename(video_path).lower()
|
36 |
+
|
37 |
+
# Dynamically adjust time interval based on filename
|
38 |
+
if "fixed" in video_filename:
|
39 |
+
time_interval = 2 # Process every 10 seconds for fixed camera
|
40 |
+
elif "moving" in video_filename:
|
41 |
+
time_interval = 7 # Process every 5 seconds for moving camera
|
42 |
+
else:
|
43 |
+
time_interval = 5 # Default to 7 seconds
|
44 |
+
|
45 |
+
frame_skip = fps * time_interval # Convert time interval to frame count
|
46 |
+
#frame_skip = fps * 7 # Skip frames every 5 seconds
|
47 |
|
48 |
frame_count = 0
|
49 |
|