search

princeton-vl / pose-hg-train

Training and experimentation code used for "Stacked Hourglass Networks for Human Pose Estimation"
Other
575 stars 185 forks source link

usage of visualize_results #52

Closed Jugaris89 closed 6 years ago

Jugaris89 commented 6 years ago

Hi,

I have a short question regarding the usage of the lua-file for visualizing results "misc/visualize_results.lua"

This file is expecting you to store your predictions in exp/mpii/best

is this "best" file and "best_preds" automatically generated when validating your net? When is the group "pred_heatmap" generated in the preds.h5? I have not been able to find this on the code

Thank you very much in advance

anewell commented 6 years ago

Hi,

I can't guarantee that the code in visualize_results is compatible with the rest of the code as it was written at a different time. It seemed better to leave it in as a reference than to remove it altogether.

I don't really remember when best/best_preds are generated, but you can probably plug in paths to any other prediction file. As for the heatmaps, there's an additional command line option to include heatmaps when saving predictions "-saveHeatmaps", but you might need to tweak the code in visualize_results.lua a bit for it all to work.

Jugaris89 commented 6 years ago

thanks for your reply!

what is exactly saved when enabling "saveHeatmaps" options, because I am training with this option as true and no maps are being stored apparently.

I have tried running the file and pointint to the preds.h5 file generated when "th main.lua -branch your-exp -expID final-preds -finalPredictions -nEpochs 0" but the file looks for groups that I don't know where are they created.

is it correct that I use it with the generated preds.h5 or shall I generate the h5 file with the prediction somehow else? what is being stored with the option "saveHeatmaps" activated?

thanks again