search

MVIG-SJTU / AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System
http://mvig.org/research/alphapose.html
Other
7.96k stars 1.97k forks source link

Pull Request - CPU only Docker image #604

Open nickgrealy opened 4 years ago

nickgrealy commented 4 years ago

Hi all,

There seems to be some demand for a CPU-only build:

There are a set of instructions for getting this to work (based on the pytorch branch) here.

I've applied the instructions, to a pytorch-cpu branch (on a fork) - https://github.com/nickgrealy/AlphaPose/tree/pytorch-cpu

Would you like this code merged back into this main code base (e.g. pytorch-cpu branch)?

Additionally, I've created a Docker image - which allows users to run AlphaPose in a reproducible manner on any on host architecture.

Fang-Haoshu commented 4 years ago

Oh Hi! Yes, it would be great if you could help!

Is it possible to add a flag to the current pytorch branch like --gpu, and --gpu -1 means using CPU? Just like the current master branch did: https://github.com/MVIG-SJTU/AlphaPose/blob/master/scripts/demo_inference.py#L177

It would also be great if you could share the docker.

Cheers!

nickgrealy commented 4 years ago

Hi @Fang-Haoshu ,

Dockerfile is in the branch, here -> https://github.com/nickgrealy/AlphaPose/tree/pytorch-cpu/docker .

Regarding adding a flag - sounds great, but I'm not a python expert... I'll leave that to someone who knows what they're doing! 😄 I basically just copied the instructions, to change .cuda() to .cpu().

Kind regards, Nick

Fang-Haoshu commented 4 years ago

Oh I see. Haha, thanks a lot!

liguoyu666 commented 4 years ago

Hi @Fang-Haoshu ,

Dockerfile is in the branch, here -> https://github.com/nickgrealy/AlphaPose/tree/pytorch-cpu/docker .

Regarding adding a flag - sounds great, but I'm not a python expert... I'll leave that to someone who knows what they're doing! I basically just copied the instructions, to change .cuda() to .cpu().

Kind regards, Nick Hi，Did you build the dockerfile for alphapose gpu?

athulvingt commented 3 years ago

@nickgrealy is it possible to control fps in webcam_demo.py

nickgrealy commented 3 years ago

@athulvingt - no idea. Try raising a ticket?