thinh-huynh-re commited on
Commit
a8bf6c1
·
1 Parent(s): 1e87f84
Files changed (2) hide show
  1. run_opencv.py +1 -1
  2. utils/img_container.py +1 -0
run_opencv.py CHANGED
@@ -101,7 +101,7 @@ while True:
101
  rs = img_container.frame_rate.show_fps(frame)
102
 
103
  # Display the resulting frame
104
- cv2.imshow("TimeSFormer", rs)
105
 
106
  # the 'q' button is set as the
107
  # quitting button you may use any
 
101
  rs = img_container.frame_rate.show_fps(frame)
102
 
103
  # Display the resulting frame
104
+ cv2.imshow("ActivityTracking", rs)
105
 
106
  # the 'q' button is set as the
107
  # quitting button you may use any
utils/img_container.py CHANGED
@@ -13,6 +13,7 @@ class ImgContainer:
13
  self.imgs: List[np.ndarray] = []
14
  self.frames_per_video = frames_per_video
15
  self.rs: Optional[DataFrame] = None
 
16
 
17
  def add_frame(self, frame: np.ndarray) -> None:
18
  if len(self.imgs) >= self.frames_per_video:
 
13
  self.imgs: List[np.ndarray] = []
14
  self.frames_per_video = frames_per_video
15
  self.rs: Optional[DataFrame] = None
16
+ self.is_recording: bool = False
17
 
18
  def add_frame(self, frame: np.ndarray) -> None:
19
  if len(self.imgs) >= self.frames_per_video: