ViTPose (simple version)

An unofficial implementation of ViTPose [Y. Xu et al., 2022]

Usage

python inference.py --image-path './examples/img1.jpg'

Note

1. Download the trained model (.pth)
   • ViTPose-B-Multi-COCO.pth
   • ViTPose-L-Multi-COCO.pth
   • ViTPose-H-Multi-COCO.pth
2. Set the config. according to the trained model
   • ViTPose-B-COCO-256x192
   • ViTPose-L-COCO-256x192
   • ViTPose-H-COCO-256x192

---

Reference

All codes were written with reference to the official ViTPose repo.