tpfister / caffe-heatmap

Caffe with heatmap regression & spatial fusion layers. Useful for any CNN image position regression task.
http://www.robots.ox.ac.uk/~vgg/software/cnn_heatmap
Other
164 stars 99 forks source link

ChaLearn #8

Open cfunk1210 opened 8 years ago

cfunk1210 commented 8 years ago

I have tried to run the sample images run on the network pretrained on the ChaLearn dataset but the output is terrible (just has high activations around the outside for each joint confidence map). Is there anything I have to change in the demo code to get the model trained on ChaLearn to work? The flic trained network seems to work well though has issues with legs (which makes sense based on the dataset). Thanks!

tpfister commented 8 years ago

The currently released ChaLearn model expects a background-subtracted image of the human -- i.e. set all non-human pixels to 0. The per-frame masks are included in the original ChaLearn dataset. We plan to release another model without this limitation -- meanwhile please background-subtract your input images.