Update process_frame.py

process_frame.py

@@ -22,7 +22,7 @@ class ProcessFrame:
  # set radius to draw arc
  self.radius = 20
 
- # Colors in BGR format.
+ # Colors in RGB format.
  self.COLORS = {
  'blue' : (0, 127, 255),
  'red' : (255, 50, 50),
@@ -140,6 +140,7 @@ class ProcessFrame:
  pos=(30, 80),
  text_color=(0, 0, 0),
  font_scale=0.6,
+ font_thickness=2,
  text_color_bg=(255, 255, 0)
  ) 
 
@@ -150,6 +151,7 @@ class ProcessFrame:
  pos=(30, dict_maps[idx][1]),
  text_color=(255, 255, 230),
  font_scale=0.6,
+ font_thickness=2,
  text_color_bg=dict_maps[idx][2]
  )
 
@@ -198,39 +200,43 @@ class ProcessFrame:
  frame = cv2.flip(frame, 1)
 
  if display_inactivity:
- # cv2.putText(frame, 'Resetting SQUAT_COUNT due to inactivity!!!', (10, frame_height - 90), 
- # self.font, 0.5, self.COLORS['blue'], 2, lineType=self.linetype)
  play_sound = 'reset_counters'
  self.state_tracker['INACTIVE_TIME_FRONT'] = 0.0
  self.state_tracker['start_inactive_time_front'] = time.perf_counter()
 
  draw_text(
  frame, 
- "CORRECT: " + str(self.state_tracker['SQUAT_COUNT']), 
- pos=(int(frame_width*0.68), 30),
+ 'CORRECT: '+str(self.state_tracker['SQUAT_COUNT']), 
+ pos=(int(frame_width*0.70), 30),
  text_color=(255, 255, 230),
- font_scale=0.7,
- text_color_bg=(18, 185, 0)
+ font_scale=0.6,
+ text_color_bg=(18, 185, 0),
+ font_thickness=2,
+ overlay_image=True,
+ overlay_type='correct' 
  ) 
  
 
  draw_text(
  frame, 
- "INCORRECT: " + str(self.state_tracker['IMPROPER_SQUAT']), 
- pos=(int(frame_width*0.68), 80),
+ 'INCORRECT: '+str(self.state_tracker['IMPROPER_SQUAT']), 
+ pos=(int(frame_width*0.70), 100),
  text_color=(255, 255, 230),
- font_scale=0.7,
+ font_scale=0.6,
  text_color_bg=(221, 0, 0),
- 
+ font_thickness=2,
+ overlay_image=True,
+ overlay_type='incorrect' 
  ) 
  
  
  draw_text(
  frame, 
- 'CAMERA NOT ALIGNED PROPERLY!!!', 
+ 'CAMERA NOT ALIGNED PROPERLY', 
  pos=(30, frame_height-60),
  text_color=(255, 255, 230),
- font_scale=0.65,
+ font_scale=0.6,
+ font_thickness=2,
  text_color_bg=(255, 153, 0),
  ) 
  
@@ -240,7 +246,8 @@ class ProcessFrame:
  'OFFSET ANGLE: '+str(offset_angle), 
  pos=(30, frame_height-30),
  text_color=(255, 255, 230),
- font_scale=0.65,
+ font_scale=0.6,
+ font_thickness=2,
  text_color_bg=(255, 153, 0),
  ) 
 
@@ -324,12 +331,12 @@ class ProcessFrame:
  
  
  # Join landmarks.
- cv2.line(frame, shldr_coord, elbow_coord, self.COLORS['light_blue'], 4, lineType=self.linetype)
- cv2.line(frame, wrist_coord, elbow_coord, self.COLORS['light_blue'], 4, lineType=self.linetype)
- cv2.line(frame, shldr_coord, hip_coord, self.COLORS['light_blue'], 4, lineType=self.linetype)
- cv2.line(frame, knee_coord, hip_coord, self.COLORS['light_blue'], 4, lineType=self.linetype)
- cv2.line(frame, ankle_coord, knee_coord,self.COLORS['light_blue'], 4, lineType=self.linetype)
- cv2.line(frame, ankle_coord, foot_coord, self.COLORS['light_blue'], 4, lineType=self.linetype)
+ cv2.line(frame, shldr_coord, elbow_coord, self.COLORS['light_blue'], 5, lineType=self.linetype)
+ cv2.line(frame, wrist_coord, elbow_coord, self.COLORS['light_blue'], 5, lineType=self.linetype)
+ cv2.line(frame, shldr_coord, hip_coord, self.COLORS['light_blue'], 5, lineType=self.linetype)
+ cv2.line(frame, knee_coord, hip_coord, self.COLORS['light_blue'], 5, lineType=self.linetype)
+ cv2.line(frame, ankle_coord, knee_coord,self.COLORS['light_blue'], 5, lineType=self.linetype)
+ cv2.line(frame, ankle_coord, foot_coord, self.COLORS['light_blue'], 5, lineType=self.linetype)
  
  # Plot landmark points
  cv2.circle(frame, shldr_coord, 7, self.COLORS['yellow'], -1, lineType=self.linetype)
@@ -454,35 +461,40 @@ class ProcessFrame:
 
 
  if display_inactivity:
- # cv2.putText(frame, 'Resetting COUNTERS due to inactivity!!!', (10, frame_height - 20), self.font, 0.5, self.COLORS['blue'], 2, lineType=self.linetype)
+ 
  play_sound = 'reset_counters'
  self.state_tracker['start_inactive_time'] = time.perf_counter()
  self.state_tracker['INACTIVE_TIME'] = 0.0
 
  
- cv2.putText(frame, str(int(hip_vertical_angle)), (hip_text_coord_x, hip_coord[1]), self.font, 0.6, self.COLORS['light_green'], 2, lineType=self.linetype)
- cv2.putText(frame, str(int(knee_vertical_angle)), (knee_text_coord_x, knee_coord[1]+10), self.font, 0.6, self.COLORS['light_green'], 2, lineType=self.linetype)
- cv2.putText(frame, str(int(ankle_vertical_angle)), (ankle_text_coord_x, ankle_coord[1]), self.font, 0.6, self.COLORS['light_green'], 2, lineType=self.linetype)
+ cv2.putText(frame, str(int(hip_vertical_angle)), (hip_text_coord_x, hip_coord[1]), self.font, 0.8, (187, 51, 255), 2, lineType=self.linetype)
+ cv2.putText(frame, str(int(knee_vertical_angle)), (knee_text_coord_x, knee_coord[1]+10), self.font, 0.8, (187, 51, 255), 2, lineType=self.linetype)
+ cv2.putText(frame, str(int(ankle_vertical_angle)), (ankle_text_coord_x, ankle_coord[1]), self.font, 0.8, (187, 51, 255), 2, lineType=self.linetype)
 
  
  draw_text(
  frame, 
- "CORRECT: " + str(self.state_tracker['SQUAT_COUNT']), 
- pos=(int(frame_width*0.68), 30),
+ 'CORRECT: '+str(self.state_tracker['SQUAT_COUNT']), 
+ pos=(int(frame_width*0.70), 30),
  text_color=(255, 255, 230),
- font_scale=0.7,
- text_color_bg=(18, 185, 0)
+ font_scale=0.6,
+ text_color_bg=(18, 185, 0),
+ font_thickness=2,
+ overlay_image=True,
+ overlay_type='correct' 
  ) 
  
 
  draw_text(
  frame, 
- "INCORRECT: " + str(self.state_tracker['IMPROPER_SQUAT']), 
- pos=(int(frame_width*0.68), 80),
+ 'INCORRECT: '+str(self.state_tracker['IMPROPER_SQUAT']), 
+ pos=(int(frame_width*0.70), 100),
  text_color=(255, 255, 230),
- font_scale=0.7,
+ font_scale=0.6,
  text_color_bg=(221, 0, 0),
- 
+ font_thickness=2,
+ overlay_image=True,
+ overlay_type='incorrect' 
  ) 
  
  
@@ -506,30 +518,36 @@ class ProcessFrame:
  if self.state_tracker['INACTIVE_TIME'] >= self.thresholds['INACTIVE_THRESH']:
  self.state_tracker['SQUAT_COUNT'] = 0
  self.state_tracker['IMPROPER_SQUAT'] = 0
- # cv2.putText(frame, 'Resetting SQUAT_COUNT due to inactivity!!!', (10, frame_height - 25), self.font, 0.7, self.COLORS['blue'], 2)
+ 
  display_inactivity = True
 
  self.state_tracker['start_inactive_time'] = end_time
 
  draw_text(
  frame, 
- "CORRECT: " + str(self.state_tracker['SQUAT_COUNT']), 
- pos=(int(frame_width*0.68), 30),
+ 'CORRECT: '+str(self.state_tracker['SQUAT_COUNT']), 
+ pos=(int(frame_width*0.70), 30),
  text_color=(255, 255, 230),
- font_scale=0.7,
- text_color_bg=(18, 185, 0)
+ font_scale=0.6,
+ text_color_bg=(18, 185, 0),
+ font_thickness=2,
+ overlay_image=True,
+ overlay_type='correct' 
  ) 
  
 
  draw_text(
- frame, 
- "INCORRECT: " + str(self.state_tracker['IMPROPER_SQUAT']), 
- pos=(int(frame_width*0.68), 80),
- text_color=(255, 255, 230),
- font_scale=0.7,
- text_color_bg=(221, 0, 0),
- 
- ) 
+ frame, 
+ 'INCORRECT: '+str(self.state_tracker['IMPROPER_SQUAT']), 
+ pos=(int(frame_width*0.70), 100),
+ text_color=(255, 255, 230),
+ font_scale=0.6,
+ text_color_bg=(221, 0, 0),
+ font_thickness=2,
+ overlay_image=True,
+ overlay_type='incorrect' 
+ ) 
+
 
  if display_inactivity:
  play_sound = 'reset_counters'