Update process_frame.py

process_frame.py

@@ -12,16 +12,12 @@ class ProcessFrame:
 
         # self.thresholds
         self.thresholds = thresholds
-
         # Font type.
         self.font = cv2.FONT_HERSHEY_SIMPLEX
-
         # line type
         self.linetype = cv2.LINE_AA
-
         # set radius to draw arc
-        self.radius = 20
-
+        self.radius = 20 # Radius for drawing arcs
         # Colors in BGR format.
         self.COLORS = {
                         'blue'       : (0, 127, 255),
@@ -36,7 +32,8 @@ class ProcessFrame:
                       }
 
         # Dictionary to maintain the various landmark features.
-        self.dict_features = {}
+        self.dict_features = {} # Dictionary to store various landmark features based on body part names
+        # Define landmark feature names for left and right body sides
         self.left_features = {
                                 'shoulder': 11,
                                 'elbow'   : 13,
@@ -57,18 +54,19 @@ class ProcessFrame:
                                 'foot'    : 32
                               }
 
+        # Assign landmark feature dictionaries for left, right body sides and nose to main dictionary
         self.dict_features['left'] = self.left_features
         self.dict_features['right'] = self.right_features
         self.dict_features['nose'] = 0
 
         # For tracking counters and sharing states in and out of callbacks.
         self.state_tracker = {
-            'state_seq': [],
+            'state_seq': [], # Sequence of detected states
 
-            'start_inactive_time': time.perf_counter(),
-            'start_inactive_time_front': time.perf_counter(),
-            'INACTIVE_TIME': 0.0,
-            'INACTIVE_TIME_FRONT': 0.0,
+            'start_inactive_time': time.perf_counter(), # Timer for front side inactivity
+            'start_inactive_time_front': time.perf_counter(), # Timer for side view inactivity
+            'INACTIVE_TIME': 0.0, # Total front side inactivity time
+            'INACTIVE_TIME_FRONT': 0.0, # Total side view inactivity time
 
             # 0 --> Bend Backwards, 1 --> Bend Forward, 2 --> Keep shin straight, 3 --> Deep squat
             'DISPLAY_TEXT' : np.full((4,), False),