[Mediapipe] 전신 포즈 인식

""" 230514_mediapipe_예림예시.py"""

import cv2
import mediapipe as mp

mp_drawing = mp.solutions.drawing_utils
mp_pose = mp.solutions.pose

# 이미지 파일 경로 설정
image_path = "C:\\Users\\songs\\01_ICT_study\\example_001_jpg.rf.8b91a22192839c7862f9d4233a0f9429.jpg"

# 이미지 파일 읽기
image = cv2.imread(image_path)

# 이미지 크기 조절 (선택 사항)
image = cv2.resize(image, (640, 480))

# BGR 색상 공간에서 RGB 색상 공간으로 변환
image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

# 전신 포즈 인식 모델 초기화
with mp_pose.Pose(
    static_image_mode=True,
    model_complexity=2,
    min_detection_confidence=0.5,
    min_tracking_confidence=0.5) as pose:

    # 이미지를 처리하고 결과를 얻음
    results = pose.process(image_rgb)

    # 처리된 이미지를 BGR로 변환
    image_bgr = cv2.cvtColor(image_rgb, cv2.COLOR_RGB2BGR)

    # 포즈 랜드마크를 이미지에 그림
    if results.pose_landmarks:
        mp_drawing.draw_landmarks(
            image_bgr, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)

    # 결과를 표시
    cv2.imshow('image', image_bgr)
    cv2.waitKey(0)
    cv2.destroyAllWindows()