giahan commited on
Commit
ce95383
1 Parent(s): db86fb5
Files changed (3) hide show
  1. .gitignore +2 -1
  2. run_opencv.py +1 -1
  3. utils/frame_rate.py +1 -1
.gitignore CHANGED
@@ -2,4 +2,5 @@ __pycache__
2
  env
3
  tmp/*
4
  !tmp/.gitkeep
5
- *.mp4
 
 
2
  env
3
  tmp/*
4
  !tmp/.gitkeep
5
+ *.mp4
6
+ .DS_Store
run_opencv.py CHANGED
@@ -29,7 +29,7 @@ class ArgParser(Tap):
29
  # "fcakyon/timesformer-large-finetuned-k600",
30
  model_name: Optional[str] = "facebook/timesformer-base-finetuned-k400"
31
 
32
- num_skip_frames: Optional[int] = 4
33
 
34
  top_k: Optional[int] = 5
35
 
 
29
  # "fcakyon/timesformer-large-finetuned-k600",
30
  model_name: Optional[str] = "facebook/timesformer-base-finetuned-k400"
31
 
32
+ num_skip_frames: Optional[int] = 2
33
 
34
  top_k: Optional[int] = 5
35
 
utils/frame_rate.py CHANGED
@@ -43,7 +43,7 @@ class FrameRate:
43
  # )
44
  pil_image = Image.fromarray(image)
45
  draw = ImageDraw.Draw(pil_image)
46
- draw.text((50, 50), text, font=self.font)
47
  image = np.asarray(pil_image)
48
 
49
  if is_recording:
 
43
  # )
44
  pil_image = Image.fromarray(image)
45
  draw = ImageDraw.Draw(pil_image)
46
+ draw.text((50, 50), text, font=self.font, fill=(0, 0, 204))
47
  image = np.asarray(pil_image)
48
 
49
  if is_recording: