search

tensorboy / pytorch_Realtime_Multi-Person_Pose_Estimation

1.36k stars 409 forks source link

Mobilenet not working #87

Open karthee320 opened 5 years ago

karthee320 commented 5 years ago

I am trying to run the demo/picture_demo.py code with mobilenet option. Going through the codes, I found there is an option to provide mobilenet model. But it seems the code is not fixed for mobilenet. Am I right?

Traceback (most recent call last): File "demo/picture_demo.py", line 29, in model = get_model('mobilenet')
File "/home/nvidia/Tracking/pytorch/pytorch_Realtime_Multi-Person_Pose_Estimation/network/rtpose_vgg.py", line 224, in get_model model = rtpose_model(models) File "/home/nvidia/Tracking/pytorch/pytorch_Realtime_Multi-Person_Pose_Estimation/network/rtpose_vgg.py", line 141, in init self.model0 = model_dict['block0'] KeyError: 'block0'

@tensorboy : Have you tried the mobilenet model for pose estimation?

GreenTeaHua commented 5 years ago

the same question for me. I guss, the pth model file just for vgg19,

LIUHAO121 commented 4 years ago

you need write some codes in rtpose.py for mobilenet

cwlinghk commented 3 years ago

Where can I find the pretrained mobilenet models? Thank you very much.