Function main

semaphore.py:271–428

()

  return cv2.waitKey(5) & 0xFF == 27

def main():
  global last_frames, frame_midpoint

  parser = argparse.ArgumentParser()
  parser.add_argument('--input', '-i', help='Input video device or file (number or path), defaults to 0', default='0')
  parser.add_argument('--flip', '-f', help='Set to any value to flip resulting output (selfie view)')
  parser.add_argument('--landmarks', '-l', help='Set to any value to draw body landmarks')
  parser.add_argument('--record', '-r', help='Set to any value to save a timestamped AVI in current directory')
  parser.add_argument('--type', '-t', help='Set to any value to type output rather than only display')
  parser.add_argument('--repeat', '-p', help='Set to any value to allow instant semaphore repetitions')
  args = parser.parse_args()

  INPUT = int(args.input) if (args.input and args.input.isdigit()) else args.input
  FLIP = args.flip is not None
  DRAW_LANDMARKS = args.landmarks is not None
  RECORDING = args.record is not None
  DISPLAY_ONLY = args.type is None
  ALLOW_REPEAT = args.repeat is not None

  cap = cv2.VideoCapture(INPUT)

  frame_size = (int(cap.get(3)), int(cap.get(4)))
  frame_midpoint = (int(frame_size[0]/2), int(frame_size[1]/2))

  recording = cv2.VideoWriter(RECORDING_FILENAME,
    cv2.VideoWriter_fourcc(*'MJPG'), FPS, frame_size) if RECORDING else None

  with mp.solutions.pose.Pose() as pose_model:
    with mp.solutions.hands.Hands(max_num_hands=2) as hands_model:
      while cap.isOpened():
        success, image = cap.read()
        if not success: break

        image.flags.writeable = False
        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
        pose_results = pose_model.process(image)
        hand_results = hands_model.process(image)

        image.flags.writeable = True
        image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)

        # draw pose
        if DRAW_LANDMARKS:
          mp.solutions.drawing_utils.draw_landmarks(
            image,
            pose_results.pose_landmarks,
            mp.solutions.pose.POSE_CONNECTIONS,
            DEFAULT_LANDMARKS_STYLE)

        hands = []
        hand_index = 0
        if hand_results.multi_hand_landmarks:
          for hand_landmarks in hand_results.multi_hand_landmarks:
            # draw hands
            if DRAW_LANDMARKS:
              mp.solutions.drawing_utils.draw_landmarks(
                image,
                hand_landmarks,