Recording

.gitignore

@@ -2,3 +2,4 @@ __pycache__
 env
 tmp/*
 !tmp/.gitkeep
+*.mp4

run_opencv.py

@@ -83,12 +83,23 @@ SKIP_FRAMES = 4
 num_skips = 0
 
 # define a video capture object
-vid = cv2.VideoCapture(0)
+camera = cv2.VideoCapture(0)
 
-while vid.isOpened():
+frame_width = int(camera.get(3))
+frame_height = int(camera.get(4))
+size = (frame_width, frame_height)
+
+video_output = cv2.VideoWriter(
+    "activities.mp4", cv2.VideoWriter_fourcc(*"MJPG"), 10, size
+)
+
+if camera.isOpened() == False:
+    print("Error reading video file")
+
+while camera.isOpened():
     # Capture the video frame
     # by frame
-    ret, frame = vid.read()
+    ret, frame = camera.read()
 
     num_skips = (num_skips + 1) % SKIP_FRAMES
 
@@ -97,19 +108,27 @@ while vid.isOpened():
 
     if num_skips == 0:
         img_container.add_frame(frame)
-        inference()
-    rs = img_container.frame_rate.show_fps(frame)
+        # inference()
+    rs = img_container.frame_rate.show_fps(frame, img_container.is_recording)
 
     # Display the resulting frame
     cv2.imshow("ActivityTracking", rs)
 
+    if img_container.is_recording:
+        video_output.write(rs)
+
     # the 'q' button is set as the
     # quitting button you may use any
     # desired button of your choice
-    if cv2.waitKey(1) & 0xFF == ord("q"):
+    k = cv2.waitKey(1)
+
+    if k == ord("q"):
         break
+    elif k == ord("r"):
+        img_container.toggle_recording()
 
 # After the loop release the cap object
-vid.release()
+camera.release()
+video_output.release()
 # Destroy all the windows
 cv2.destroyAllWindows()

utils/frame_rate.py

@@ -27,9 +27,9 @@ class FrameRate:
             self.fps = self.NO_FRAMES / (end_time - self.start_time)
             self.start_time = end_time
 
-    def show_fps(self, image: np.ndarray) -> np.ndarray:
+    def show_fps(self, image: np.ndarray, is_recording=False) -> np.ndarray:
         if self.fps != -1:
-            return cv2.putText(
+            image = cv2.putText(
                 image,
                 f"FPS {self.fps:.0f} _ {self.label}",
                 (50, 50),
@@ -38,5 +38,10 @@ class FrameRate:
                 color=(255, 0, 0),
                 thickness=2,
             )
+            if is_recording:
+                image = cv2.circle(
+                    image, (50, 100), radius=10, color=(0, 0, 255), thickness=-1
+                )
+            return image
         else:
             return image

utils/img_container.py

@@ -20,6 +20,9 @@ class ImgContainer:
             self.imgs.pop(0)
         self.imgs.append(frame)
 
+    def toggle_recording(self) -> None:
+        self.is_recording = not self.is_recording
+
     @property
     def ready(self):
         return len(self.imgs) == self.frames_per_video