Update

.gitignore

@@ -3,4 +3,5 @@ env
 tmp/*
 !tmp/.gitkeep
 *.mp4
-.DS_Store
+.DS_Store
+diary.json

run_opencv.py

@@ -1,8 +1,8 @@
 import json
+from datetime import datetime
 from typing import List, Optional, Tuple
 
 import cv2
-import numpy as np
 import pandas as pd
 import torch
 from tap import Tap
@@ -50,6 +50,12 @@ class ActivityModel:
 
         self.load_json()
 
+        self.diary: List[Tuple[str, str, float]] = []  # [time, activity, confidence]
+
+    def save_diary(self):
+        df = pd.DataFrame(self.diary, columns=["time", "activity", "confidence"])
+        df.to_csv("diary.csv")
+
     def load_json(self):
         if args.id2label is not None:
             with open(args.id2label, encoding="utf-8") as f:
@@ -93,6 +99,7 @@ class ActivityModel:
             self.args.threshold is not None and confidence >= self.args.threshold
         ):
             img_container.frame_rate.label = f"{predicted_label}_{confidence:.2f}%"
+            self.diary.append((str(datetime.now()), predicted_label, confidence))
 
         # logits = np.squeeze(logits)
         # logits = logits.squeeze().numpy()
@@ -167,6 +174,8 @@ def main(args: ArgParser):
         elif k == ord("r"):
             img_container.toggle_recording()
 
+    activity_model.save_diary()
+
     # After the loop release the cap object
     camera.release()
     video_output.release()