hb-setosys commited on
Commit
b0a9e04
·
verified ·
1 Parent(s): a7d7e62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -1
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
- frame_skip = fps * 7 # Skip frames every 5 seconds
 
 
 
 
 
 
 
 
 
 
 
 
 
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