Skip to content

Body Pose Estimation

Overviews

Using body pose estimation to detect hand touching the box

Install requirements

clone repo easy ViTPose (third_party) Go to /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

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

Usages

Example:

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: path to input video
  • PATH_TO_OUTPUT_VIDEO: path to output video

Note

Modify bbox_a, bbox_b, bbox_c, bbox_d to define box(xyxy)