search

MVIG-SJTU / AlphaPose

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

Single Hand error #1026

Open Jaykob opened 2 years ago

Jaykob commented 2 years ago

Hi! I'm trying to get the single hand running as described in the model zoo doc. When executing the following as suggested by you, I get an error.

python3 scripts/demo_inference.py --cfg configs/single_hand/resnet/256x192_res50_lr1e-3_2x-dcn-regression.yaml --checkpoint pretrained_models/singlehand_fast50_dcn_regression_256x192.pth --indir examples/demo/ --save_img

Error:

Loading YOLO model..
Loading pose model from pretrained_models/singlehand_fast50_dcn_regression_256x192.pth...
  0% 0/4 [00:00<?, ?it/s]error in deformable_im2col: no kernel image is available for execution on the device

Any idea what's going wrong here?

BTW: Do you have any hints on where to start for foot only training/detection? Should I take the hand cfg as a starter?

HaoyiZhu commented 2 years ago

Hi, what's your CUDA version? And for foot only, you can refer to the single hand cfg. You should also prepare your own annotation data.