Update

test_opencv.py → run_opencv.py

@@ -77,6 +77,7 @@ def get_frames_per_video(model_name: str) -> int:
 
 
 model_name = "facebook/timesformer-base-finetuned-k400"
+# "facebook/timesformer-base-finetuned-k400"
 # "facebook/timesformer-base-finetuned-k600",
 # "facebook/timesformer-base-finetuned-ssv2",
 # "facebook/timesformer-hr-finetuned-k600",
@@ -92,6 +93,10 @@ print(f"Frames per video: {frames_per_video}")
 
 img_container = ImgContainer(frames_per_video)
 
+SKIP_FRAMES = 4
+
+num_skips = 0
+
 # define a video capture object
 vid = cv2.VideoCapture(0)
 
@@ -100,10 +105,14 @@ while(True):
     # by frame
     ret, frame = vid.read()
 
+    num_skips = (num_skips + 1) % SKIP_FRAMES
+
     img_container.img = frame
     img_container.frame_rate.count()
-    img_container.add_frame(frame)
-    inference()
+    
+    if num_skips == 0:
+        img_container.add_frame(frame)
+        inference()
     rs = img_container.frame_rate.show_fps(frame)
 
     # Display the resulting frame