Vào thẳng nội dung

Ước lượng tư thế cơ thể

Tổng quan

Sử dụng ước lượng tư thế cơ thể để phát hiện tay chạm vào hộp.

Cài đặt yêu cầu

clone repo easy ViTPose (third_party) Đi tới /projects/body_pose_estimation

git clone https://github.com/JunkyByte/easy_ViTPose.git
cd easy_ViTPose
pip install -e .

Install ECOS_CORE before and run

pip install -r requirements.txt

Pre-trained(Các mô hình đã được huấn luyện sẵn)

ViTPose: https://huggingface.co/JunkyByte/easy_ViTPose/resolve/main/torch/vitpose-25-s.pth

YOLOv5: https://huggingface.co/JunkyByte/easy_ViTPose/resolve/main/torch/vitpose-25-s.pth

Sử dụng

Ví dụ:

python inference.py --vitpose-path ./vitpose-25-s.pth --yolo-path yolov5s.onnx --input-path "<PATH_TO_INPUT_VIDEO>" --output-path "<PATH_TO_OUTPUT_VIDEO>"
  • PATH_TO_INPUT_VIDEO: đường dẫn tới video đầu vào
  • PATH_TO_OUTPUT_VIDEO: đường dẫn tới video đầu ra

Ghi

Chỉnh sửa bbox_a, bbox_b, bbox_c, bbox_d để định nghĩa hộp (xyxy)