Skip to content

mfkiwl/movenet

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Workable MoveNet on PyTorch

Movenet is Google's next generation fast pose detection model. However, there seems not PyTorch version.

I ported original tensorflow weights to pytorch, using CenterNet code base to inference. The result is fully right.

Hopefully, you can even train it on pytorch if you have time.

If you are interested on Movenet, open an issue and I will guide you have to contribute.

Demo

Run:

python demo.py single_pose --dataset active --arch movenet --demo data/input_image.jpeg --load_model ./weights/movenet.pth --K 1 --gpus -1 --debug 2

About

Google's Next Gen Pose Estimation in PyTorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 100.0%