search

yuanyuanli85 / Stacked_Hourglass_Network_Keras

Keras Implementation for Stacked Hourglass Network
112 stars 44 forks source link

I want to use only 6 key points, not a 16 key point pre-trained model. #27

Open NaJongHo opened 4 years ago

NaJongHo commented 4 years ago

I want to use only 6 key points, not a 16 key point pre-trained model. So I ran the code with scratch. qs

but I keep getting errors like photos.

Any comments would be greatly appreciated.

yuanyuanli85 commented 4 years ago

you need to make sure the training samples provide you the right shape.

chojuahn commented 4 years ago

It seems like the model is still expecting 16 key points. You should try searching hard-coded number '16' all over the code base and replace with 6