본 프로젝트는 VitPose를 활용한 Human Pose Estimation에서 시작하여, GAT(Graph Attention Networks)를 기반으로 사용자 맞춤형 포즈 개선 시스템을 구축하는 것을 목표로 했습니다. 먼저, VitPose를 통해 사람의 자세를 구성하는 주요 관절의 좌표(keypoints)를 정확히 추출하고, 이를 스켈레톤 형태로 시각화했습니다. 이후, GAT 모델을 사용해 각 관절을 노드로, 관절 간의 관계를 엣지로 표현하여 그래프 구조를 구성했습니다. 학습된 모델은 사용자의 현재 자세와 이상적인 자세를 비교하고, 피드백이 필요한 노드를 시각적으로 표시하여 개선 방향을 제안합니다
크게 서버부분, 자세 추정부분, 자세 개선부분, 시각화 부분으로 구성되어 있습니다.
ViTPose, MMdetection, MMCV를 설치해야 하며 그 외에서 Flask, OpenCV 등 여러 프로그램 설치를 요구합니다.
requirement.txt와 environment.yml을 확인하세요.
좋아요
댓글
